Распознание номеров грузовых вагонов по фотографии

Предоставлена выборка фотографий номеров грузовых вагонов с аннотацией.

Задача

Построить модель распознавания(детектирования) изображений.

Инструменты

• YoloV8
• TrOCRProcessor
• VisionEncoderDecoderModel
• Roboflow
• PIL
• IPython
• seaborn
• numpy
• pandas
• matplotlib

Отработанные методы и навыки

1. Загружены данные и проведена предобработка.
2. Построена модель распознования изображений на архитектуре YoloV8m.
3. Проведен анализ и визуализация работы модели.
4. Отработано альтернативное решение на архитектуре TrOCR(оптического распознавания печатных символов).

Итоги проекта

Модель YoloV8m показывала хорошие результаты, производительность модели на разных уровнях сложности обнаружения на тестовой выборке mAP50-95 = 0.87, mAP50 = 0.995. Модель имеет небольшие трудности с распознанием единиц при сложном уровне обнаружения, что является следствием плохого качества изображения этого класса распознания.