This is an ONNX version of dima806/animal_151_types_image_detection. It was automatically converted and uploaded using this Hugging Face Space.
See the pipeline documentation for image-classification: https://huggingface.co/docs/transformers.js/api/pipelines#module_pipelines.ImageClassificationPipeline
Returns animal type given image with about 99% accuracy.
See https://www.kaggle.com/code/dima806/animal-151-types-detection-vit for more details.
Classification report:
precision recall f1-score support
acinonyx-jubatus 1.0000 1.0000 1.0000 12
aethia-cristatella 1.0000 0.9167 0.9565 12
agalychnis-callidryas 1.0000 1.0000 1.0000 12
agkistrodon-contortrix 1.0000 1.0000 1.0000 12
ailuropoda-melanoleuca 1.0000 1.0000 1.0000 12
ailurus-fulgens 1.0000 1.0000 1.0000 12
alces-alces 1.0000 1.0000 1.0000 12
anas-platyrhynchos 1.0000 1.0000 1.0000 12
ankylosaurus-magniventris 0.9167 0.9167 0.9167 12
apis-mellifera 1.0000 1.0000 1.0000 12
aptenodytes-forsteri 1.0000 1.0000 1.0000 12
aquila-chrysaetos 1.0000 1.0000 1.0000 12
ara-macao 1.0000 1.0000 1.0000 12
architeuthis-dux 0.9231 1.0000 0.9600 12
ardea-herodias 1.0000 1.0000 1.0000 12
balaenoptera-musculus 1.0000 1.0000 1.0000 12
betta-splendens 1.0000 1.0000 1.0000 12
bison-bison 1.0000 1.0000 1.0000 12
bos-gaurus 1.0000 1.0000 1.0000 12
bos-taurus 1.0000 1.0000 1.0000 12
bradypus-variegatus 1.0000 1.0000 1.0000 12
branta-canadensis 1.0000 1.0000 1.0000 12
canis-lupus 1.0000 1.0000 1.0000 12
canis-lupus-familiaris 1.0000 1.0000 1.0000 12
carcharodon-carcharias 1.0000 1.0000 1.0000 12
cardinalis-cardinalis 1.0000 1.0000 1.0000 12
cathartes-aura 1.0000 1.0000 1.0000 12
centrochelys-sulcata 1.0000 1.0000 1.0000 12
centruroides-vittatus 1.0000 1.0000 1.0000 12
ceratitis-capitata 1.0000 0.9167 0.9565 12
ceratotherium-simum 1.0000 1.0000 1.0000 12
chelonia-mydas 1.0000 1.0000 1.0000 12
chrysemys-picta 1.0000 1.0000 1.0000 12
circus-hudsonius 1.0000 1.0000 1.0000 12
codium-fragile 1.0000 1.0000 1.0000 12
coelacanthiformes 0.9231 1.0000 0.9600 12
colaptes-auratus 1.0000 1.0000 1.0000 12
connochaetes-gnou 1.0000 1.0000 1.0000 12
correlophus-ciliatus 1.0000 1.0000 1.0000 12
crocodylus-niloticus 1.0000 1.0000 1.0000 12
crotalus-atrox 1.0000 1.0000 1.0000 12
crotophaga-sulcirostris 1.0000 1.0000 1.0000 12
cryptoprocta-ferox 1.0000 1.0000 1.0000 12
cyanocitta-cristata 1.0000 1.0000 1.0000 12
danaus-plexippus 1.0000 1.0000 1.0000 12
dasypus-novemcinctus 1.0000 0.9167 0.9565 12
delphinapterus-leucas 1.0000 1.0000 1.0000 12
dendrobatidae 1.0000 1.0000 1.0000 12
dermochelys-coriacea 0.9231 1.0000 0.9600 12
desmodus-rotundus 1.0000 0.9167 0.9565 12
diplodocus 1.0000 1.0000 1.0000 12
dugong-dugon 1.0000 1.0000 1.0000 12
eidolon-helvum 1.0000 1.0000 1.0000 12
enhydra-lutris 1.0000 1.0000 1.0000 12
enteroctopus-dofleini 1.0000 1.0000 1.0000 12
equus-caballus 0.9231 1.0000 0.9600 12
equus-quagga 1.0000 1.0000 1.0000 12
eudocimus-albus 1.0000 1.0000 1.0000 12
eunectes-murinus 1.0000 1.0000 1.0000 12
falco-peregrinus 1.0000 1.0000 1.0000 12
felis-catus 1.0000 1.0000 1.0000 12
formicidae 1.0000 1.0000 1.0000 12
gallus-gallus-domesticus 1.0000 1.0000 1.0000 12
gavialis-gangeticus 1.0000 1.0000 1.0000 12
geococcyx-californianus 1.0000 1.0000 1.0000 12
giraffa-camelopardalis 1.0000 1.0000 1.0000 12
gorilla-gorilla 1.0000 1.0000 1.0000 12
haliaeetus-leucocephalus 1.0000 1.0000 1.0000 12
hapalochlaena-maculosa 1.0000 1.0000 1.0000 12
heloderma-suspectum 1.0000 1.0000 1.0000 12
heterocera 0.9231 1.0000 0.9600 12
hippopotamus-amphibius 1.0000 1.0000 1.0000 12
homo-sapiens 0.9231 1.0000 0.9600 12
hydrurga-leptonyx 0.9231 1.0000 0.9600 12
icterus-galbula 1.0000 1.0000 1.0000 12
icterus-gularis 1.0000 1.0000 1.0000 12
icterus-spurius 1.0000 1.0000 1.0000 12
iguana-iguana 1.0000 1.0000 1.0000 12
iguanodon-bernissartensis 1.0000 1.0000 1.0000 12
inia-geoffrensis 1.0000 1.0000 1.0000 12
lampropeltis-triangulum 1.0000 1.0000 1.0000 12
lemur-catta 1.0000 1.0000 1.0000 12
lepus-americanus 1.0000 1.0000 1.0000 12
loxodonta-africana 1.0000 1.0000 1.0000 12
macropus-giganteus 1.0000 1.0000 1.0000 12
malayopython-reticulatus 1.0000 1.0000 1.0000 12
mammuthus-primigeniu 1.0000 1.0000 1.0000 12
martes-americana 1.0000 1.0000 1.0000 12
megaptera-novaeangliae 1.0000 1.0000 1.0000 12
melanerpes-carolinus 1.0000 1.0000 1.0000 12
mellisuga-helenae 1.0000 1.0000 1.0000 12
mergus-serrator 1.0000 1.0000 1.0000 12
mimus-polyglottos 1.0000 1.0000 1.0000 12
monodon-monoceros 0.9231 1.0000 0.9600 12
musca-domestica 1.0000 1.0000 1.0000 12
odobenus-rosmarus 1.0000 1.0000 1.0000 12
okapia-johnstoni 1.0000 1.0000 1.0000 12
ophiophagus-hannah 1.0000 1.0000 1.0000 12
orcinus-orca 1.0000 1.0000 1.0000 12
ornithorhynchus-anatinus 1.0000 1.0000 1.0000 12
ovis-aries 1.0000 1.0000 1.0000 12
ovis-canadensis 1.0000 1.0000 1.0000 12
panthera-leo 1.0000 0.9167 0.9565 12
panthera-onca 0.8571 1.0000 0.9231 12
panthera-pardus 1.0000 0.8333 0.9091 12
panthera-tigris 1.0000 1.0000 1.0000 12
pantherophis-alleghaniensis 1.0000 1.0000 1.0000 12
pantherophis-guttatus 1.0000 1.0000 1.0000 12
papilio-glaucus 1.0000 0.9167 0.9565 12
passerina-ciris 1.0000 1.0000 1.0000 12
pavo-cristatus 1.0000 1.0000 1.0000 12
periplaneta-americana 1.0000 1.0000 1.0000 12
phascolarctos-cinereus 1.0000 1.0000 1.0000 12
phoebetria-fusca 1.0000 1.0000 1.0000 12
phoenicopterus-ruber 1.0000 1.0000 1.0000 12
phyllobates-terribilis 1.0000 1.0000 1.0000 12
physalia-physalis 1.0000 1.0000 1.0000 12
physeter-macrocephalus 0.9231 1.0000 0.9600 12
poecile-atricapillus 1.0000 1.0000 1.0000 12
pongo-abelii 1.0000 1.0000 1.0000 12
procyon-lotor 1.0000 1.0000 1.0000 12
pteranodon-longiceps 1.0000 1.0000 1.0000 12
pterois-mombasae 1.0000 0.8333 0.9091 12
pterois-volitans 0.8571 1.0000 0.9231 12
puma-concolor 1.0000 0.9167 0.9565 12
rattus-rattus 1.0000 1.0000 1.0000 12
rusa-unicolor 1.0000 1.0000 1.0000 12
salmo-salar 1.0000 1.0000 1.0000 12
sciurus-carolinensis 1.0000 1.0000 1.0000 12
smilodon-populator 1.0000 1.0000 1.0000 12
spheniscus-demersus 1.0000 1.0000 1.0000 12
sphyrna-mokarran 1.0000 1.0000 1.0000 12
spinosaurus-aegyptiacus 1.0000 1.0000 1.0000 12
stegosaurus-stenops 1.0000 1.0000 1.0000 12
struthio-camelus 1.0000 1.0000 1.0000 12
tapirus 1.0000 1.0000 1.0000 12
tarsius-pumilus 1.0000 1.0000 1.0000 12
taurotragus-oryx 1.0000 1.0000 1.0000 12
telmatobufo-bullocki 1.0000 1.0000 1.0000 12
thryothorus-ludovicianus 1.0000 1.0000 1.0000 12
triceratops-horridus 1.0000 0.9167 0.9565 12
trilobita 1.0000 0.9167 0.9565 12
turdus-migratorius 1.0000 1.0000 1.0000 12
tursiops-truncatus 1.0000 1.0000 1.0000 12
tyrannosaurus-rex 1.0000 1.0000 1.0000 12
tyrannus-tyrannus 1.0000 1.0000 1.0000 12
ursus-arctos-horribilis 1.0000 1.0000 1.0000 12
ursus-maritimus 1.0000 1.0000 1.0000 12
varanus-komodoensis 1.0000 1.0000 1.0000 12
vulpes-vulpes 1.0000 1.0000 1.0000 12
vultur-gryphus 1.0000 1.0000 1.0000 12
accuracy 0.9923 1812
macro avg 0.9930 0.9923 0.9922 1812
weighted avg 0.9930 0.9923 0.9922 1812
onnx/model.onnx 344 MBonnx/model_fp16.onnx 172 MBonnx/model_uint8.onnx 87 MBonnx/model_int8.onnx 87 MBonnx/model_quantized.onnx 87 MBonnx/model_bnb4.onnx 57 MBonnx/model_q4.onnx 57 MBonnx/model_q4f16.onnx 50 MB