dima806's picture
Update README.md
fe4248c verified
|
raw
history blame
919 Bytes
metadata
license: apache-2.0
metrics:
  - accuracy
  - f1
base_model:
  - google/vit-base-patch16-224-in21k

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