Edit model card

Returns ball type given an image.

See https://www.kaggle.com/code/dima806/ball-types-image-detection for more details.

Classification report:

                   precision    recall  f1-score   support

      wiffle ball     1.0000    1.0000    1.0000        39
        puffballs     1.0000    1.0000    1.0000        38
    chrochet ball     1.0000    1.0000    1.0000        39
        golf ball     1.0000    1.0000    1.0000        39
    pokeman balls     1.0000    1.0000    1.0000        39
  water polo ball     1.0000    1.0000    1.0000        39
         football     1.0000    1.0000    1.0000        39
           marble     1.0000    1.0000    1.0000        38
    medicine ball     1.0000    1.0000    1.0000        39
      tether ball     1.0000    1.0000    1.0000        38
    billiard ball     1.0000    1.0000    1.0000        38
      cannon ball     1.0000    1.0000    1.0000        39
     crystal ball     1.0000    1.0000    1.0000        38
     cricket ball     1.0000    1.0000    1.0000        39
sepak takraw ball     1.0000    1.0000    1.0000        39
      tennis ball     1.0000    1.0000    1.0000        39
    wrecking ball     1.0000    1.0000    1.0000        38
  rubberband ball     1.0000    1.0000    1.0000        39
      buckeyballs     1.0000    1.0000    1.0000        39
     bowling ball     1.0000    1.0000    1.0000        38
         eyeballs     1.0000    1.0000    1.0000        38
        meat ball     1.0000    1.0000    1.0000        38
            brass     1.0000    1.0000    1.0000        39
       screwballs     1.0000    1.0000    1.0000        38
         baseball     1.0000    1.0000    1.0000        38
       beachballs     1.0000    1.0000    1.0000        39
      soccer ball     1.0000    1.0000    1.0000        38
       basketball     1.0000    1.0000    1.0000        39
      volley ball     1.0000    1.0000    1.0000        39
      paint balls     1.0000    1.0000    1.0000        39

         accuracy                         1.0000      1158
        macro avg     1.0000    1.0000    1.0000      1158
     weighted avg     1.0000    1.0000    1.0000      1158
Downloads last month
8
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/ball_types_image_detection

Finetuned
(1691)
this model