Обнаружение объектов с TensorFlow
Довольно давно я обучал ИИ для обнаружения объектов
Однажды холодным июльским зимним днем в Австралии… я почувствовал срочную необходимость обучить модель ИИ для обнаружения не заглушенных арматурных стержней из бетона…
Поэтому я купил в местном магазине Bunnings Warehouse несколько новых блестящих арматурных стержней, таких как эти
и защитные колпачки для стержней, такие как эти
Начало с колпачков
Сделал сотню фотографий с разными композициями с этими желтыми колпачками для арматурных стержней, разметил их в LabelImg, взял модель SSD, затем другую EfficientNet, немного покодил на Python - и модель быстро поняла, что я от нее хотел.
изначально модель принимала почти все желтые цилиндрические объекты за колпачки для арматурных стержней
И ограничивающие рамки могли быть точнее…
Затем появились стержни
Взял несколько других полезных инструментов из гаража и сделал еще сотню фотографий, как эта, и модель стала видеть лучше…
Да, определенно лучше
Мобильное приложение
В конце концов обучил небольшую модель SSD ИИ для работы на мобильном телефоне без большой задержки, создал простое приложение для Android, и вот как оно выглядело на Samsung S8:
Понятно.
Полезные ссылки
- MMdetection больше не поддерживается
- Рабочий процесс для обучения детектора объектов с Label Studio & MMDetection
- LabelImg: https://github.com/HumanSignal/labelImg
- TensorFlow: https://www.tensorflow.org/
- Шпаргалка по Python
- Шпаргалка по Conda
- Шпаргалка по venv
- Шпаргалка по Bash
- Шпаргалка по Ollama
- Генерация PDF в Python - библиотеки и примеры
