YOLOvME: 개똥 미스터리 사건

다음에 다른 사람의 마당에 개가 똥을 싸게 할 때는 다시 한번 생각해 보세요. Ben Farahmand는 그의 딸이 마당에 나타나기 시작한 개똥의 유입을 그에게 알리기 시작했을 때 바로 이 문제를 깨달았습니다. 누가 배후에 있는지 확신하지 못한 Ben은 범인을 현장에서 잡을 수 있는 방법을 브레인스토밍하기 시작했습니다.

저희는 이야기가 어떻게 끝나는지 알고 싶어서 Ben과 함께 앉아 YOLOv5 및 비전 AI에 대한 그의 경험에 대해 몇 가지 질문을 했습니다.

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

과제 식별

“문제를 발견했습니다. 제 딸이 마당에서 노는데 강아지 똥이 있는 것을 발견했습니다. 강아지 똥을 치우지 않고 그냥 두는 사람에게 말을 걸고 싶었습니다.”라고 벤은 말합니다. 벤은 24시간 내내 창밖을 쳐다보는 것보다 더 나은 해결책을 찾아야 했습니다.

적합한 ML 모델 선택

그래서 빠른 Google 검색 후 Ben은 YOLOv5를 발견하고 모델 학습을 시작했습니다. “몇 가지 다른 사용 가능한 모델을 발견했습니다. YOLOv5는 문서화가 잘 되어 있는 것 같아서 그걸로 시작했습니다.” 그런 다음 Ben은 트위터로 눈에 띄는 정보를 문서화하여 우리를 긴장하게 만들었습니다.

ML 모델 훈련

처음에 벤은 몇 가지 장애물에 부딪혔습니다. 벤의 모델은 처음에 사건의 용의자로 간주되었던 작은 개를 탐지하는 데 어려움을 겪었습니다. 때로는 작은 개를 새로 분류하고, 때로는 완전히 놓치기도 했습니다. 특정 사례에서 크고 푹신한 흰 개가 양으로 분류되었습니다. 이 문제를 해결하기 위해 벤은 사용하던 작은 신경망을 더 큰 신경망으로 교체하여 성공을 거두었습니다. 작은 개는 더 이상 새로 식별되지 않았고, 이전의 양은 마침내 개로 올바르게 식별되었습니다.

범인을 잡은 적이 있나요?

“제가 해봤는데 결국 주인들이 제 마당을 지나가는 일이 잦았습니다. 아버지가 지나갈 때도 있고, 딸이 지나갈 때도 있었습니다. 데이터를 살펴보니 아버지는 항상 강아지 똥을 치우는데 딸은 그렇지 않다는 것을 알게 되었습니다.” 벤의 동네는 상황이 개선된 것 같습니다. “편지를 쓰고 이야기를 나눴더니 지금까지는 더 이상 강아지 똥이 없었습니다. 하지만 이제는 좀 으스스하게 느껴집니다. 가끔 밖에 나가면 이 개들이 지나가는 것을 알아봅니다. 그들이 내가 자신들을 지켜보고 있다는 것을 알고 있을까 궁금합니다.”

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

“글쎄요, 우연히 OpenCV에 대해 알게 되었고 대학원 시절에 그것을 가지고 놀았습니다. 하지만 제 대학원 프로그램은 엔지니어링에 초점을 두지 않고 사회적 영향에 대한 디자인에 더 중점을 두었습니다. 그냥 조금씩 만지작거렸습니다. 최근에 제가 원하는 것을 할 수 있다는 것을 알았지만 100% 확신하지 못했습니다. 또한 얼마나 쉬운지 깨닫지 못했습니다. 훌륭한 문서를 만들어 주셔서 감사합니다. 문서를 작성한 사람은 월급을 올려줘야 합니다.”

YOLOv5로 해결하고 싶은 다른 과제는 무엇인가요?

“지금 당장 떠오르는 생각은 우리 동네에 사람들이 항상 무시하고 지나가는 정지 표지판이 있다는 것입니다. 아마 다른 누군가가 이미 이와 유사한 애플리케이션을 만들었을 수도 있지만, 저는 누가 계속해서 정지 표지판을 무시하는지 알아내고 싶습니다. 하지만 제가 이런 것을 설치하는 것이 얼마나 합법적인지는 확실히 말할 수 없습니다.”

‍

‍Ben Farahmand는 데이터 리터러시 교육을 촉진하고 교사가 실제 학습을 수학 및 과학 교육에 도입할 수 있도록 지원하는 소규모 교육 회사인 Tuva Labs의 제품 관리자입니다. Tuva Labs는 모든 종류의 통계 수업을 커리큘럼에 통합할 수 있도록 데이터 세트, 수업 및 그래프 작성 소프트웨어를 제공합니다. Ben은 또한 고전적인 외계인 침략 이야기를 다시 들려주는 펄프 공상 과학 테마의 손에 땀을 쥐게 하는 협동 보드 게임인 FAZA를 설립했습니다.

‍

Ben의 트위터 스레드에서 개똥 이야기를 확인하세요.

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