mitbersh/car-view-classification

AutoInspect - Car View Classifier (ResNet18)

Модель для определения ракурса автомобиля на изображении. Часть проекта AutoInspect (pipeline: view classification → car parts segmentation → damage segmentation).

Task

Классификация ракурса автомобиля.

Overview

Модель построена на ResNet18 и использует две головы:

• left / center / right
• front / center / back

Итоговый ракурс собирается из комбинации этих двух предсказаний.

Labels

CLASS_NAMES = [
"back",
"back-left",
"back-right",
"front",
"front-left",
"front-right",
"left",
"other",
"right",
]

How to use

Запуск модели описан на github проекта.