YOLOvME: 개 배설물의 미스터리한 사건

다음번에 마당에서 반려견이 배변하도록 둘 때는 한 번 더 생각해 보시는 것이 좋겠습니다. 벤 파라함드는 딸이 마당에 개 배설물이 계속 늘어난다고 알리기 시작하면서 이 문제를 직접 겪게 되었습니다. 누가 그런 일을 하는지 몰랐던 벤은 범인을 현장에서 붙잡을 방법을 고민하기 시작했습니다.

이야기가 어떻게 끝났는지 궁금해 벤을 만나 Ultralytics YOLOv5와 비전 AI를 사용한 경험에 대해 몇 가지 질문을 했습니다.

객체 감지 도구를 만드는 과정은 어땠나요?#

문제 파악#

“문제가 보였어요. ‘그래, 딸이 마당에서 놀다가 개 배설물을 발견하는구나. 개가 마당에 배변하도록 두고 치우지 않는 사람과 이야기하고 싶었어요.’라고 생각했죠.”라고 벤은 말합니다. 벤은 24시간 내내 창밖을 바라보는 것보다 나은 해결책을 찾아야 했습니다.

적합한 ML 모델 선택#

간단히 Google 검색을 해 본 Ben은 Ultralytics YOLOv5를 발견하고 모델 학습을 시작했습니다. “사용할 수 있는 모델 몇 가지를 찾았습니다. YOLOv5는 문서가 잘 갖춰져 있는 것 같아서 그 모델로 시작했습니다.” 이후 Ben은 Twitter에 방문해 잠복 관찰 중 얻은 정보를 기록했고, 우리를 긴장감 속에 빠뜨렸습니다.

ML 모델 학습#

초기에 벤은 몇 가지 장애물을 만났습니다. 처음에는 벤의 모델이 작은 개를 감지하는 데 어려움을 겪었으며, 작은 개들도 여전히 용의자로 간주되고 있었습니다. 때로는 작은 개를 새로 분류했고, 어떤 때는 아예 감지하지 못했습니다. 한 사례에서는 크고 복슬복슬한 흰 개를 양으로 분류했습니다. 이 문제를 해결하기 위해 벤은 사용하던 작은 신경망을 더 큰 신경망으로 바꾸었고, 그 결과 문제가 해결되었습니다. 작은 개는 더 이상 새로 식별되지 않았고, 양으로 잘못 분류됐던 개도 마침내 개로 정확하게 식별되었습니다.

범인을 결국 잡았나요?#

“잡았어요. 알고 보니 주인들이 자주 우리 집 앞을 지나갔어요. 어떤 때는 아버지가, 어떤 때는 딸이 지나갔죠. 데이터를 확인해 보니 아버지는 항상 개 배설물을 치웠지만 딸은 그러지 않았어요.” 벤의 동네 상황은 나아진 것 같습니다. “편지를 쓰고 직접 이야기한 뒤로는 개 배설물이 더 이상 나오지 않았어요. 하지만 가끔 밖에 나갔을 때 지나가는 개들을 알아보게 되니 조금 섬뜩하기도 해요. 내가 지켜본다는 걸 그 개들이 알고 있을까 생각하게 되죠.”

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

“대학원 시절 OpenCV를 알고 있어서 조금 다뤄 본 적이 있는데, 우연히 시작하게 됐어요. 하지만 제 대학원 과정은 엔지니어링보다는 사회적 영향력을 위한 디자인에 초점을 맞췄어요. 그저 이것저것 만져 봤죠. 최근에 제가 원하던 일을 할 수 있다는 건 알았지만 실제로 할 수 있을지는 100% 확신하지 못했어요. 얼마나 쉽게 할 수 있는지도 몰랐고요. 훌륭한 문서를 제공해 주셔서 정말 감사합니다. 문서를 작성한 분은 연봉을 올려야 해요.”

Ultralytics YOLOv5로 또 어떤 문제를 해결하고 싶으신가요?#

“지금 가장 먼저 떠오르는 건 동네에 있는 정지 표지판을 사람들이 계속 무시하고 지나간다는 거예요. 누군가 이미 이런 애플리케이션을 만들었을 수도 있지만, 누가 계속 정지 표지판을 무시하는지 알아내고 싶어요. 다만 이런 장치를 설치하는 것이 법적으로 문제가 없는지는 모르겠네요.”

벤 파라함드는 데이터 리터러시 교육을 촉진하고 교사가 수학 및 과학 수업에 실생활 학습을 도입하도록 지원하는 소규모 교육 회사인 Tuva Labs의 제품 관리자입니다. Tuva Labs는 모든 종류의 통계 수업을 교육과정에 포함할 수 있도록 데이터셋, 수업 자료 및 그래프 작성 소프트웨어를 제공합니다. 벤은 고전적인 외계인 침공 이야기를 펄프 SF 테마로 새롭게 풀어낸 긴장감 넘치는 협동 보드게임 FAZA도 설립했습니다.

Ben의 Twitter 스레드에서 개똥 이야기를 확인해 보세요.

Ben의 코드는 dog-detector GitHub 저장소에서 확인하세요.