Задача: Определение дистанции до автомобиля по фото.
Решение данной задачи производилось в несколько этапов:
- Детектирование автомобиля и номерного знака на фото
- Использование детекций в качестве признаков, а также расчет дополнительных геометрических характеристик bbox, таких как площади, отношения сторон, диагонали и пр.
- Прогнозирование расстояния регрессионной моделью на основе полносвязной нейронной сети
Обработка данных и определение расстояния описаны в файле main.ipynb
Обучения детекционной и полнозвязной моделей описаны в файлах yolo_training.ipynb и main_model_training.ipynb соответственно.
YOLO обучалась на фото размером 1600, детекции производились по фото с меньшим размером 1200.
Полносвязная модель построена на базе Tensorflow, имеет два hidden слоя с регуляризацией Dropout.