YOLOvME: загадочное дело о собачьих экскрементах
Узнай, как Бен Фарахманд использовал YOLOv5 и компьютерное зрение, чтобы решить соседскую проблему с экскрементами собак и улучшить отношения в сообществе.
В следующий раз, когда ты позволишь своей собаке сходить в туалет на чужом газоне, возможно, стоит подумать дважды. Бен Фарахманд столкнулся именно с такой проблемой, когда его дочь начала сообщать ему о потоке собачьих «сюрпризов», появляющихся на его лужайке. Не зная, кто за этим стоит, Бен начал размышлять, как поймать виновника с поличным.
Нам нужно было узнать, чем закончилась эта история, поэтому мы пообщались с Беном и задали ему несколько вопросов о его опыте работы с Ultralytics YOLOv5 и визуальным ИИ.
Как проходил процесс создания твоего инструмента для обнаружения объектов?#
Определение проблемы#
«Я увидел проблему. Я подумал: окей, моя дочь играет во дворе и замечает там собачьи экскременты. Я хотел иметь возможность поговорить с человеком, который позволяет собаке гадить и не убирает за ней», — говорит Бен. Бену нужно было найти решение получше, чем круглосуточно смотреть в окно.
Выбор правильной модели машинного обучения#
Итак, после быстрого поиска в Google Бен обнаружил Ultralytics YOLOv5 и приступил к обучению своей модели. «Я наткнулся на пару разных доступных моделей. У YOLOv5 оказалась хорошая документация, так что я просто начал с неё». Затем Бен обратился к Twitter и документировал информацию со своей засады, держа нас в напряжении.
Обучение модели машинного обучения#
В начале Бен столкнулся с парой препятствий. Его модель поначалу с трудом обнаруживала маленьких собак, которые все еще считались подозреваемыми по делу. Иногда она классифицировала маленьких собак как птиц, а иногда пропускала их вовсе. В одном конкретном случае большая пушистая белая собака была классифицирована как овца. Чтобы исправить эту проблему, Бен заменил небольшую нейронную сеть, которую использовал, на более крупную, что принесло успех. Маленькие собаки больше не определялись как птицы, а бывшая «овца» наконец была правильно идентифицирована как собака.
Удалось ли тебе поймать виновника?#
«Да, и вот что оказалось: владельцы часто проходили мимо моего двора. Иногда проходил отец, иногда дочь. Когда я просмотрел свои данные, я обнаружил, что отец всегда убирал за собакой, а дочь — нет». Похоже, ситуация в районе Бена улучшилась. «Мы написали письмо и поговорили с ними, и пока что собачьих экскрементов больше не было. Но теперь это кажется немного жутковатым, потому что иногда я выхожу на улицу и узнаю этих собак, проходящих мимо. Мне интересно, знают ли они, что я наблюдаю за ними».
Как ты начал работать с Ultralytics и YOLOv5?#
«Ну, случайно, потому что я знал об OpenCV, я немного экспериментировал с ним еще в магистратуре. Но моя магистерская программа была больше про дизайн для социального воздействия, без какого-либо инженерного фокуса. Я просто немного возился с этим. Недавно я понял, что могу сделать то, что хочу, я просто не был на 100% уверен, справлюсь ли. Я также не осознавал, насколько это легко. Честь и хвала вам, ребята, за отличную документацию. Тот, кто писал вашу документацию, заслуживает повышения».
Какие еще задачи ты хотел бы решить с помощью Ultralytics YOLOv5?#
«На данный момент первое, что приходит на ум, — это то, что у нас в районе есть знак «Стоп», на который люди постоянно не обращают внимания. Может быть, кто-то уже создал подобное приложение, но я хотел бы выяснить, кто продолжает игнорировать этот знак. Однако я не могу сказать, насколько законно было бы для меня настраивать что-то подобное».
Ben Farahmand — продакт-менеджер в небольшой образовательной компании Tuva Labs, которая развивает образование в сфере грамотности работы с данными и помогает учителям внедрять реальные кейсы в обучение математике и естественным наукам. Tuva Labs предоставляет датасеты, уроки и ПО для построения графиков, позволяя добавлять любые статистические задачи в учебные планы. Бен также основал FAZA — увлекательную кооперативную настольную игру в жанре научно-фантастического палпа, переосмысляющую классическую историю об инопланетном вторжении.
Посмотри тред Бена в Twitter, где описана история с собачьими фекалиями.
Репозиторий Бена на GitHub можно найти здесь.
