Edit model card

Returns jellyfish type based on image.

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

Classification report:

                         precision    recall  f1-score   support

         blue_jellyfish     1.0000    1.0000    1.0000        30
       barrel_jellyfish     1.0000    1.0000    1.0000        30
mauve_stinger_jellyfish     1.0000    1.0000    1.0000        30
         Moon_jellyfish     1.0000    1.0000    1.0000        30
      compass_jellyfish     1.0000    1.0000    1.0000        30
   lions_mane_jellyfish     1.0000    1.0000    1.0000        30

               accuracy                         1.0000       180
              macro avg     1.0000    1.0000    1.0000       180
           weighted avg     1.0000    1.0000    1.0000       180
Downloads last month
14
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/jellyfish_types_image_detection

Finetuned
(1708)
this model