dima806
/

vehicle_10_types_image_detection

Image Classification

Inference Endpoints

Model card Files Files and versions Community

dima806 commited on 13 days ago

Commit

7619f8d

•

1 Parent(s): 66ec88a

Update README.md

Files changed (1) hide show

README.md +3 -1

README.md CHANGED Viewed

@@ -3,8 +3,10 @@ license: apache-2.0
 metrics:
 - accuracy
 - f1
 ---
-Return a vehicle type probability based on an image.
 See https://www.kaggle.com/code/dima806/vehicle-10-types-detection-vit for more details.

 metrics:
 - accuracy
 - f1
+base_model:
+- google/vit-base-patch16-224-in21k
 ---
+Return a vehicle type probability based on an image with about 93% accuracy.
 See https://www.kaggle.com/code/dima806/vehicle-10-types-detection-vit for more details.