YOLOvME: 강아지 배설물 미스터리 사건

다음에 누군가의 마당에 반려견이 배변하게 둔다면, 다시 한번 생각해보는 것이 좋을 것입니다. Ben Farahmand는 딸이 마당에 나타나기 시작한 수많은 반려견 배변에 대해 알려주기 시작했을 때 이 문제를 정확히 깨달았습니다. 누가 이런 일을 저질렀는지 확신할 수 없었던 Ben은 범인을 현장에서 잡을 방법을 고민하기 시작했습니다.

저희는 이야기가 어떻게 끝났는지 궁금하여 Ben과 마주 앉아 YOLOv5 및 비전 AI 경험에 대해 몇 가지 질문을 던졌습니다.

Link to this section객체 탐지 도구를 만드는 과정은 어떠했나요?#

Link to this section도전 과제 확인#

“문제를 발견했습니다. 딸아이가 마당에서 노는데 마당에 있는 반려견 배변을 알아차린 것이죠. 저는 반려견이 배변하도록 두고 치우지 않는 사람과 직접 이야기하고 싶었습니다.”라고 Ben은 말합니다. Ben은 24시간 내내 창밖만 내다보는 것보다 더 나은 해결책을 찾아야 했습니다.

Link to this section적절한 ML 모델 선택#

그래서 간단한 구글 검색 후, Ben은 YOLOv5를 발견하고 모델 학습을 시작했습니다. “사용 가능한 몇 가지 모델을 발견했습니다. YOLOv5는 문서화가 잘 되어 있는 것 같아 바로 그것으로 시작했죠.” 그 후 Ben은 트위터를 통해 잠복근무 내용을 기록하며 저희 모두를 긴장하게 했습니다.

Link to this sectionML 모델 학습#

처음에 Ben은 몇 가지 장애물에 부딪혔습니다. Ben의 모델은 초기에는 사건의 용의자였던 작은 개들을 탐지하는 데 어려움을 겪었습니다. 때로는 작은 개들을 새(bird)로 분류하기도 했고, 때로는 아예 탐지하지 못하기도 했습니다. 특히 크고 복슬복슬한 흰색 개가 양으로 분류된 적도 있었습니다. 이 문제를 해결하기 위해 Ben은 사용하던 작은 신경망을 더 큰 것으로 교체했고, 여기서 성공을 거두었습니다. 작은 개들은 더 이상 새로 식별되지 않았고, 이전의 양은 마침내 개로 올바르게 식별되었습니다.

Link to this section범인을 잡았나요?#

“네, 잡았습니다. 결론적으로 견주들은 종종 제 마당을 지나가곤 했습니다. 때로는 아버지가, 때로는 딸이 지나갔죠. 데이터를 살펴보니 아버지는 항상 반려견 배변을 치웠지만, 딸은 그렇지 않다는 사실을 발견했습니다.” Ben의 동네 상황은 나아진 것 같습니다. “우리는 편지를 쓰고 그들과 이야기를 나눴고, 지금까지는 더 이상의 반려견 배변은 없습니다. 하지만 가끔 밖으로 나갈 때 지나가는 개들을 알아볼 때면 기분이 조금 묘합니다. 그들이 제가 자신들을 지켜보고 있다는 사실을 알까 하는 생각이 듭니다.”

Link to this sectionUltralytics와 YOLOv5를 어떻게 시작하게 되었나요?#

“글쎄요, 우연이었습니다. 대학원 시절 OpenCV에 대해 알고 조금 다뤄본 적이 있었거든요. 하지만 제 대학원 과정은 공학적 초점보다는 사회적 영향력을 위한 디자인에 더 가까웠습니다. 그저 조금 만져본 수준이었죠. 최근에 제가 원하는 것을 구현할 수 있다는 것을 알았고, 다만 제가 할 수 있을지 100% 확신이 없었을 뿐입니다. 또한 이렇게 쉬울 줄은 몰랐습니다. 훌륭한 문서를 제공해주신 여러분께 감사드립니다. 문서를 작성하신 분은 꼭 승진해야 합니다.”

Link to this sectionYOLOv5로 해결하고 싶은 다른 과제가 있나요?#

“현재 가장 먼저 떠오르는 것은 저희 동네에 사람들이 항상 무시하고 지나가는 정지 표지판이 있다는 것입니다. 아마 누군가 이미 이런 애플리케이션을 만들었을지 모르지만, 저는 누가 계속해서 정지 표지판을 무시하는지 알고 싶습니다. 하지만 이런 것을 설치하는 것이 법적으로 어떤 문제가 있을지는 말씀드리기 어렵네요.”

Ben Farahmand는 데이터 문해력 교육을 촉진하고 교사들이 실제 세계의 학습을 수학 및 과학 수업에 도입할 수 있도록 지원하는 Tuva Labs라는 소규모 교육 회사의 프로덕트 매니저입니다. Tuva Labs는 통계 수업을 커리큘럼에 통합할 수 있도록 데이터셋, 수업 자료 및 그래프 소프트웨어를 제공합니다. Ben은 또한 고전적인 외계인 침공 이야기를 재구성한 펄프 공상과학 테마의 손에 땀을 쥐게 하는 협동 보드 게임인 FAZA를 설립했습니다.

Ben의 트위터 스레드에서 반려견 배변 이야기를 확인해보세요.

Ben의 GitHub 저장소는 여기에서 찾을 수 있습니다.