mitbersh's picture
Update README.md
402505b verified
---
datasets:
- mitbersh/car-view
base_model:
- microsoft/resnet-18
pipeline_tag: image-classification
---
# AutoInspect - Car View Classifier (ResNet18)
Модель для определения ракурса автомобиля на изображении.
Часть проекта [**AutoInspect**](https://github.com/DedovInside/AutoInspect/tree/ml/ml) (pipeline: *view classification → car parts segmentation → damage segmentation*).
## Task
Классификация ракурса автомобиля.
## Overview
Модель построена на **ResNet18** и использует две головы:
* left / center / right
* front / center / back
Итоговый ракурс собирается из комбинации этих двух предсказаний.
## Labels
```python
CLASS_NAMES = [
"back",
"back-left",
"back-right",
"front",
"front-left",
"front-right",
"left",
"other",
"right",
]
```
## How to use
Запуск модели описан на [github проекта](https://github.com/DedovInside/AutoInspect/tree/ml/ml/view-classification/model).