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

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

우리는 이야기가 어떻게 끝났는지 알고 싶었기에, Ben과 만나 Ultralytics YOLOv5 및 비전 AI 경험에 관해 몇 가지 질문을 나누었습니다.

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

도전 과제 확인#

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

적절한 ML 모델 선택#

그래서 간단한 Google 검색 끝에 Ben은 Ultralytics YOLOv5를 발견하고 모델 학습을 시작했습니다. "몇 가지 다른 사용 가능한 모델을 접했습니다. YOLOv5는 문서화가 잘 되어 있는 것 같아서 그냥 그것으로 시작했습니다." 그런 후 Ben은 Twitter에 접속하여 감시 활동 정보를 기록하며 우리를 긴장감 속에 몰아넣었습니다.

ML 모델 학습#

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

범인을 잡았나요?#

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

Ultralytics와 YOLOv5를 어떻게 시작하게 되었나요?#

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

Ultralytics YOLOv5를 사용하여 해결하고 싶은 다른 과제는 무엇인가요?#

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

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

반려견 배변 이야기를 기록한 Ben의 Twitter 스레드를 확인해 보세요.

Ben의 GitHub 저장소는 여기에서 확인하세요.