Edit model card

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
Downloads last month
25
Safetensors
Model size
85.9M 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/animal_151_types_image_detection

Finetuned
(1694)
this model