kikogazda commited on
Commit
f10ca14
·
verified ·
1 Parent(s): 2ca4ce9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -1
README.md CHANGED
@@ -1,3 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
  # 🚗 TwinCar: Fine-Grained Car Classification on Stanford Cars 196 (EfficientNetV2 Edition)
2
 
3
  > **TwinCar** is a modern deep learning pipeline for car make/model/year classification, featuring a cutting-edge EfficientNetV2 backbone, advanced data augmentation (Mixup, CutMix), robust metric tracking, rich evaluation visuals, and deep model explainability (Grad-CAM++).
@@ -244,4 +256,4 @@ with torch.no_grad():
244
  output = model(input_tensor)
245
  pred_idx = output.argmax(1).item()
246
 
247
- print(f"Predicted class: {class_names[pred_idx]} (index: {pred_idx})")
 
1
+ ---
2
+ license: apache-2.0
3
+ datasets:
4
+ - tanganke/stanford_cars
5
+ language:
6
+ - en
7
+ metrics:
8
+ - accuracy
9
+ base_model:
10
+ - timm/efficientnetv2_rw_s.ra2_in1k
11
+ pipeline_tag: image-classification
12
+ ---
13
  # 🚗 TwinCar: Fine-Grained Car Classification on Stanford Cars 196 (EfficientNetV2 Edition)
14
 
15
  > **TwinCar** is a modern deep learning pipeline for car make/model/year classification, featuring a cutting-edge EfficientNetV2 backbone, advanced data augmentation (Mixup, CutMix), robust metric tracking, rich evaluation visuals, and deep model explainability (Grad-CAM++).
 
256
  output = model(input_tensor)
257
  pred_idx = output.argmax(1).item()
258
 
259
+ print(f"Predicted class: {class_names[pred_idx]} (index: {pred_idx})")