metadata
datasets:
- mitbersh/car-view
base_model:
- microsoft/resnet-18
pipeline_tag: image-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 проекта.