-
Update README.md
-
Update README.md
-
Update car view model
-
Update README.md
-
Update README.md
-
Create README.md
# AutoInspect — Car View Classifier (ResNet18)
Модель для определения ракурса автомобиля на изображении.
Часть проекта **AutoInspect** (pipeline: *view classification → car parts segmentation → damage segmentation*).
## Task
Multi-class классификация ракурса автомобиля (9 классов).
## Labels
```python
CLASS_NAMES = [
"back",
"back-left",
"back-right",
"front",
"front-left",
"front-right",
"left",
"other",
"right",
]
```
## How to use
### Load model
```python
import torch
import torch.nn as nn
from torchvision import models, transforms
from PIL import Image
CLASS_NAMES = [
"back", "back-left", "back-right",
"front", "front-left", "front-right",
"left", "other", "right"
]
device = "cuda" if torch.cuda.is_available() else "cpu"
model = models.resnet18(weights=None)
model.fc = nn.Linear(model.fc.in_features, len(CLASS_NAMES))
state_dict = torch.load("best_car_view_model.pth", map_location=device)
model.load_state_dict(state_dict… -
Upload car_view_model.pth
-
initial commit
