Downloads · 30 days
0
vehicle-model-classifier
myneighborh/vehicle-model-classifier
vehicle-model-classifier is a image classification model from myneighborh. Use it when you need a label for an image. The card lists the license as cc-by-nc-sa-4.0.
This deep learning model classifies 396 real-world used car models from vehicle images.
Access
Public
Updated Jul 18, 2025
Repo size
3.2 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth3.2 GB · 100%
At a glance
- Task
- Image Classification
- License
- cc-by-nc-sa-4.0
- Access
- Public
- Created
- Jul 12, 2025
- Updated
- Jul 18, 2025
- SHA
- c9e2c3b5
Datasets
Model card
From the Hugging Face model README
Vehicle Model Classifier
This deep learning model classifies 396 real-world used car models from vehicle images.
Final Model Details
- Backbone: ConvNeXt-Base
- Framework: PyTorch
- Augmentation: MixUp / CutMix / Albumentations
- Scheduler: CosineAnnealingLR / ReduceLROnPlateau
- Validation: Stratified K-Fold (3-fold)
- Ensemble: Logit & Soft Voting Ensemble
- TTA: 3-view & 5-view Test Time Augmentation
How to Use
from transformers import AutoImageProcessor, AutoModelForImageClassification
from PIL import Image
import torch
processor = AutoImageProcessor.from_pretrained("myneighborh/vehicle-model-classifier")
model = AutoModelForImageClassification.from_pretrained("myneighborh/vehicle-model-classifier")
image = Image.open("path_to_image.jpg")
inputs = processor(images=image, return_tensors="pt")
outputs = model(**inputs)
pred = outputs.logits.argmax(-1).item()
label = model.config.id2label[pred]
print("Predicted label:", label)
- Task
- Image Classification
- License
- cc-by-nc-sa-4.0
- Created
- Jul 12, 2025
- Updated
- Jul 18, 2025