Returns pokemon type given an image.
See https://www.kaggle.com/code/dima806/pokemon-common-types-image-detection-vit for more details.
Accuracy: 0.9588
F1 Score: 0.9459
Classification report:
precision recall f1-score support
Wartortle 0.9615 0.9615 0.9615 26
Arcanine 1.0000 1.0000 1.0000 27
Staryu 1.0000 1.0000 1.0000 27
Arbok 1.0000 1.0000 1.0000 26
Butterfree 0.0000 0.0000 0.0000 26
Geodude 1.0000 1.0000 1.0000 27
Seaking 1.0000 1.0000 1.0000 26
Diglett 1.0000 1.0000 1.0000 27
Jynx 1.0000 1.0000 1.0000 26
Sandslash 0.9286 1.0000 0.9630 26
Magneton 1.0000 1.0000 1.0000 27
Scyther 1.0000 1.0000 1.0000 27
Kabuto 1.0000 1.0000 1.0000 26
Cubone 0.8276 0.9231 0.8727 26
Golem 1.0000 1.0000 1.0000 26
Dewgong 0.9630 1.0000 0.9811 26
Pidgey 1.0000 0.9259 0.9615 27
Kadabra 0.5200 1.0000 0.6842 26
Ditto 1.0000 1.0000 1.0000 26
Venomoth 0.5400 1.0000 0.7013 27
Rattata 1.0000 1.0000 1.0000 27
Alakazam 0.0000 0.0000 0.0000 26
Machoke 1.0000 0.9615 0.9804 26
Farfetchd 1.0000 1.0000 1.0000 27
Omastar 1.0000 0.9615 0.9804 26
Machamp 0.9630 1.0000 0.9811 26
Jigglypuff 1.0000 1.0000 1.0000 27
Dragonite 1.0000 1.0000 1.0000 26
Weepinbell 1.0000 1.0000 1.0000 26
Sandshrew 1.0000 1.0000 1.0000 26
Dugtrio 1.0000 1.0000 1.0000 27
Mankey 0.8276 0.8889 0.8571 27
Hitmonchan 0.8667 1.0000 0.9286 26
Spearow 1.0000 1.0000 1.0000 26
Caterpie 1.0000 1.0000 1.0000 27
Dratini 0.0000 0.0000 0.0000 26
Bulbasaur 1.0000 1.0000 1.0000 26
Tentacool 1.0000 1.0000 1.0000 26
Gengar 1.0000 1.0000 1.0000 26
Machop 0.9643 1.0000 0.9818 27
Raichu 1.0000 1.0000 1.0000 26
Alolan Sandslash 0.0000 0.0000 0.0000 26
Eevee 1.0000 1.0000 1.0000 27
Abra 1.0000 1.0000 1.0000 27
Haunter 1.0000 1.0000 1.0000 27
Metapod 1.0000 1.0000 1.0000 27
Fearow 0.9630 1.0000 0.9811 26
Nidorina 0.8966 1.0000 0.9455 26
Zapdos 1.0000 1.0000 1.0000 27
Ninetales 1.0000 0.9630 0.9811 27
Chansey 1.0000 1.0000 1.0000 27
Kangaskhan 0.9630 1.0000 0.9811 26
Poliwrath 1.0000 0.9630 0.9811 27
Gyarados 1.0000 1.0000 1.0000 27
Charmeleon 1.0000 1.0000 1.0000 26
Vulpix 1.0000 1.0000 1.0000 26
Pidgeot 1.0000 0.8846 0.9388 26
Blastoise 0.9630 1.0000 0.9811 26
Porygon 1.0000 1.0000 1.0000 26
Psyduck 0.9643 1.0000 0.9818 27
Dragonair 0.5400 1.0000 0.7013 27
Raticate 0.9630 1.0000 0.9811 26
Squirtle 1.0000 0.9615 0.9804 26
Charizard 1.0000 1.0000 1.0000 26
Electrode 1.0000 0.9615 0.9804 26
Flareon 1.0000 1.0000 1.0000 26
Exeggutor 0.9643 1.0000 0.9818 27
Pikachu 1.0000 1.0000 1.0000 26
Wigglytuff 1.0000 1.0000 1.0000 26
Venusaur 1.0000 0.9615 0.9804 26
Mewtwo 1.0000 1.0000 1.0000 26
Clefable 1.0000 1.0000 1.0000 27
Oddish 1.0000 1.0000 1.0000 26
Ekans 1.0000 1.0000 1.0000 26
Shellder 1.0000 1.0000 1.0000 26
Marowak 0.9130 0.8077 0.8571 26
Kakuna 1.0000 1.0000 1.0000 27
Rapidash 1.0000 0.9615 0.9804 26
Rhydon 1.0000 0.9630 0.9811 27
Ivysaur 1.0000 1.0000 1.0000 26
Slowpoke 1.0000 1.0000 1.0000 26
Lapras 1.0000 1.0000 1.0000 27
Clefairy 1.0000 1.0000 1.0000 26
Hitmonlee 1.0000 1.0000 1.0000 26
Jolteon 1.0000 1.0000 1.0000 26
Growlithe 1.0000 1.0000 1.0000 27
Gastly 1.0000 1.0000 1.0000 27
Aerodactyl 1.0000 1.0000 1.0000 27
Weedle 1.0000 1.0000 1.0000 26
Tauros 1.0000 1.0000 1.0000 27
Seel 0.8929 0.9615 0.9259 26
Zubat 1.0000 1.0000 1.0000 26
Meowth 0.0000 0.0000 0.0000 26
Persian 0.6341 1.0000 0.7761 26
Articuno 0.9310 1.0000 0.9643 27
Weezing 0.9643 1.0000 0.9818 27
Magnemite 1.0000 1.0000 1.0000 27
Omanyte 0.9630 1.0000 0.9811 26
Mew 1.0000 1.0000 1.0000 26
Vileplume 1.0000 1.0000 1.0000 27
Nidoqueen 0.9615 0.9259 0.9434 27
Vaporeon 0.9000 1.0000 0.9474 27
Ponyta 0.9630 1.0000 0.9811 26
Moltres 1.0000 1.0000 1.0000 27
Voltorb 0.9630 1.0000 0.9811 26
Magikarp 1.0000 1.0000 1.0000 27
Beedrill 1.0000 1.0000 1.0000 26
Nidoking 1.0000 1.0000 1.0000 27
Paras 1.0000 1.0000 1.0000 26
Grimer 1.0000 0.9615 0.9804 26
Dodrio 1.0000 1.0000 1.0000 26
Charmander 1.0000 1.0000 1.0000 26
Muk 1.0000 0.9615 0.9804 26
Primeape 0.8966 0.9630 0.9286 27
Victreebel 1.0000 1.0000 1.0000 26
Golbat 1.0000 1.0000 1.0000 26
Horsea 1.0000 1.0000 1.0000 27
Goldeen 1.0000 1.0000 1.0000 27
Pidgeotto 0.8966 1.0000 0.9455 26
Koffing 0.9630 1.0000 0.9811 26
Seadra 0.5870 1.0000 0.7397 27
Tentacruel 1.0000 1.0000 1.0000 26
Pinsir 1.0000 1.0000 1.0000 26
Cloyster 1.0000 1.0000 1.0000 26
Gloom 1.0000 1.0000 1.0000 26
Graveler 1.0000 1.0000 1.0000 26
Magmar 1.0000 1.0000 1.0000 27
Krabby 0.9286 1.0000 0.9630 26
Electabuzz 1.0000 1.0000 1.0000 27
Poliwhirl 0.9643 1.0000 0.9818 27
Golduck 0.9310 1.0000 0.9643 27
Onix 1.0000 1.0000 1.0000 27
Nidorino 1.0000 1.0000 1.0000 27
Snorlax 0.9630 1.0000 0.9811 26
Starmie 1.0000 1.0000 1.0000 27
Slowbro 1.0000 1.0000 1.0000 26
MrMime 1.0000 1.0000 1.0000 26
Venonat 1.0000 1.0000 1.0000 27
Kabutops 1.0000 1.0000 1.0000 26
Drowzee 1.0000 1.0000 1.0000 26
Rhyhorn 1.0000 1.0000 1.0000 26
Tangela 1.0000 1.0000 1.0000 27
Doduo 1.0000 1.0000 1.0000 27
Exeggcute 1.0000 1.0000 1.0000 26
Poliwag 1.0000 1.0000 1.0000 27
Lickitung 1.0000 1.0000 1.0000 26
Hypno 0.9286 1.0000 0.9630 26
Bellsprout 1.0000 1.0000 1.0000 27
Parasect 1.0000 1.0000 1.0000 26
Kingler 1.0000 0.9231 0.9600 26
accuracy 0.9588 3960
macro avg 0.9382 0.9583 0.9459 3960
weighted avg 0.9386 0.9588 0.9463 3960
- Downloads last month
- 5
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/pokemon_types_image_detection
Base model
google/vit-base-patch16-224-in21k