Распознавание дорожных знаков в реальных дорожных условиях — самостоятельное
обучение нейросети для детекции объектов
Видеоурок 42 — обучение нейронной сети для детекции объектов: дорожные знаки в реальных дорожных условиях: https://singtown.com/learn/50918/
Код, модель и набор данных находятся здесь: https://github.com/SingTown/Traffic-Sign-FOMO
Инструкция на китайском: https://github.com/SingTown/Traffic-Sign-FOMO/blob/main/README-CN.md
Всего было обучено две модели
Первая модель: разрешение при обучении — 128*128, использовано трансферное обучение на базе MobileNetV2 0.35, показатель S1 Score 92%, частота кадров работы около 9 кадров/с, работает только на OpenMV4 Plus. (У OpenMV4 недостаточно памяти)
Вторая модель: разрешение при обучении 96*96, использовано трансферное обучение на базе MobileNetV2 0.35, показатель S1 Score 91%, частота кадров работы около 12 кадров/с, пиковое потребление ОЗУ 244 КБ, может работать как на OpenMV4, так и на OpenMV4 Plus.
Сайт партнёра OpenMV EdgeImpulse для онлайн-обучения: https://www.edgeimpulse.com/
Примечание:
- Всего есть две модели. Для OpenMV4 можно использовать только эту небольшую модель, обученную с разрешением 96*96. На Plus можно использовать обе модели.
- Набор данных содержит пять папок: «Остановка запрещена» x, «Звуковой сигнал запрещён» n, «Ограничение скорости 80», «Въезд запрещён»- и тестовые изображения. В самой модели и в уроке мы использовали только «Остановка запрещена» x, «Звуковой сигнал запрещён» n, «Ограничение скорости 80» и тестовые изображения. Размеченного и обученного набора данных для «Въезд запрещён»- нет. Вы можете сами решить, размечать и обучать ли знак «Въезд запрещён».
- Я разметил только три класса объектов: «Остановка запрещена» x, «Звуковой сигнал запрещён» n и «Ограничение скорости 80». Другие знаки в наборе данных (например, «Ограничение скорости 60», «Ограничение скорости 30», «Въезд запрещён» и т. д.) я не размечал и не обучал. Слишком похожие знаки, такие как «Ограничение скорости 80» и «Ограничение скорости 60», могут распознаваться неправильно. Вы также можете добавить разметку и обучить новые классы, например «Ограничение скорости 60» и т. д.
- Благодарности — источник набора данных дорожных знаков в реальных дорожных условиях Китая (Институт автоматизации Китайской академии наук): http://www.nlpr.ia.ac.cn/pal/trafficdata/detection.html