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.

Classification report:

              precision    recall  f1-score   support

         SUV     0.8780    0.9000    0.8889        40
         bus     1.0000    1.0000    1.0000        40
family sedan     0.8571    0.9000    0.8780        40
 fire engine     0.8444    0.9500    0.8941        40
 heavy truck     0.9459    0.8750    0.9091        40
        jeep     0.9512    0.9750    0.9630        40
     minibus     0.9500    0.9500    0.9500        40
  racing car     1.0000    0.9500    0.9744        40
        taxi     0.9750    0.9750    0.9750        40
       truck     0.9722    0.8750    0.9211        40

    accuracy                         0.9350       400
   macro avg     0.9374    0.9350    0.9354       400
weighted avg     0.9374    0.9350    0.9354       400
Downloads last month
34
Safetensors
Model size
85.8M params
Tensor type
F32
·
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Model tree for dima806/vehicle_10_types_image_detection

Finetuned
(1769)
this model