Returns a common mammal type given an image.
See https://www.kaggle.com/code/dima806/mammals-45-types-image-classification-vit for more details.
Classification report:
precision recall f1-score support
african_elephant 1.0000 1.0000 1.0000 71
alpaca 0.9200 0.9718 0.9452 71
american_bison 1.0000 1.0000 1.0000 71
anteater 0.9853 0.9437 0.9640 71
arctic_fox 0.9286 0.9155 0.9220 71
armadillo 0.9726 1.0000 0.9861 71
baboon 0.9718 0.9718 0.9718 71
badger 1.0000 0.9718 0.9857 71
blue_whale 0.9710 0.9437 0.9571 71
brown_bear 0.9722 0.9859 0.9790 71
camel 0.9861 1.0000 0.9930 71
dolphin 0.8974 0.9859 0.9396 71
giraffe 0.9857 0.9718 0.9787 71
groundhog 0.9714 0.9577 0.9645 71
highland_cattle 0.9859 0.9859 0.9859 71
horse 1.0000 0.9859 0.9929 71
jackal 0.9577 0.9444 0.9510 72
kangaroo 0.8415 0.9583 0.8961 72
koala 0.9589 0.9859 0.9722 71
manatee 0.9861 0.9861 0.9861 72
mongoose 0.9483 0.7746 0.8527 71
mountain_goat 0.9855 0.9577 0.9714 71
opossum 1.0000 0.9577 0.9784 71
orangutan 1.0000 1.0000 1.0000 71
otter 1.0000 0.9577 0.9784 71
polar_bear 0.9706 0.9296 0.9496 71
porcupine 1.0000 0.9722 0.9859 72
red_panda 0.9718 0.9718 0.9718 71
rhinoceros 0.9859 0.9859 0.9859 71
sea_lion 0.7600 0.8028 0.7808 71
seal 0.8308 0.7500 0.7883 72
snow_leopard 1.0000 1.0000 1.0000 71
squirrel 0.9444 0.9577 0.9510 71
sugar_glider 0.8554 1.0000 0.9221 71
tapir 1.0000 1.0000 1.0000 71
vampire_bat 1.0000 0.9861 0.9930 72
vicuna 1.0000 0.8873 0.9403 71
walrus 0.9342 0.9861 0.9595 72
warthog 0.9571 0.9437 0.9504 71
water_buffalo 0.9333 0.9859 0.9589 71
weasel 0.9583 0.9583 0.9583 72
wildebeest 0.9577 0.9444 0.9510 72
wombat 0.8947 0.9577 0.9252 71
yak 1.0000 0.9437 0.9710 71
zebra 0.9595 1.0000 0.9793 71
accuracy 0.9572 3204
macro avg 0.9587 0.9573 0.9572 3204
weighted avg 0.9586 0.9572 0.9572 3204
- Downloads last month
- 24