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: 10

Safetensors

Model size

85.8M params

Tensor type

F32

Model tree for dima806/jellyfish_types_image_detection

Base model

google/vit-base-patch16-224-in21k

Finetuned

(2517)

this model