Hugging Face's logo

dima806
/
pokemon_types_image_detection

Image Classification
Transformers
PyTorch
Safetensors
vit
Model card Files
xet
 Community
dima806 commited on
Commit
ff45820
·
verified ·
1 Parent(s): ce2934c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +167 -1
README.md CHANGED
@@ -4,4 +4,170 @@ metrics:
4
  - accuracy
5
  - f1
6
  ---
7
- See https://www.kaggle.com/code/dima806/pokemon-common-types-image-detection-vit for more details.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  - accuracy
5
  - f1
6
  ---
7
+ Returns pokemon type given an image.
8
+
9
+ See https://www.kaggle.com/code/dima806/pokemon-common-types-image-detection-vit for more details.
10
+
11
+ ```
12
+ Accuracy: 0.9588
13
+ F1 Score: 0.9459
14
+
15
+ Classification report:
16
+
17
+ precision recall f1-score support
18
+
19
+ Wartortle 0.9615 0.9615 0.9615 26
20
+ Arcanine 1.0000 1.0000 1.0000 27
21
+ Staryu 1.0000 1.0000 1.0000 27
22
+ Arbok 1.0000 1.0000 1.0000 26
23
+ Butterfree 0.0000 0.0000 0.0000 26
24
+ Geodude 1.0000 1.0000 1.0000 27
25
+ Seaking 1.0000 1.0000 1.0000 26
26
+ Diglett 1.0000 1.0000 1.0000 27
27
+ Jynx 1.0000 1.0000 1.0000 26
28
+ Sandslash 0.9286 1.0000 0.9630 26
29
+ Magneton 1.0000 1.0000 1.0000 27
30
+ Scyther 1.0000 1.0000 1.0000 27
31
+ Kabuto 1.0000 1.0000 1.0000 26
32
+ Cubone 0.8276 0.9231 0.8727 26
33
+ Golem 1.0000 1.0000 1.0000 26
34
+ Dewgong 0.9630 1.0000 0.9811 26
35
+ Pidgey 1.0000 0.9259 0.9615 27
36
+ Kadabra 0.5200 1.0000 0.6842 26
37
+ Ditto 1.0000 1.0000 1.0000 26
38
+ Venomoth 0.5400 1.0000 0.7013 27
39
+ Rattata 1.0000 1.0000 1.0000 27
40
+ Alakazam 0.0000 0.0000 0.0000 26
41
+ Machoke 1.0000 0.9615 0.9804 26
42
+ Farfetchd 1.0000 1.0000 1.0000 27
43
+ Omastar 1.0000 0.9615 0.9804 26
44
+ Machamp 0.9630 1.0000 0.9811 26
45
+ Jigglypuff 1.0000 1.0000 1.0000 27
46
+ Dragonite 1.0000 1.0000 1.0000 26
47
+ Weepinbell 1.0000 1.0000 1.0000 26
48
+ Sandshrew 1.0000 1.0000 1.0000 26
49
+ Dugtrio 1.0000 1.0000 1.0000 27
50
+ Mankey 0.8276 0.8889 0.8571 27
51
+ Hitmonchan 0.8667 1.0000 0.9286 26
52
+ Spearow 1.0000 1.0000 1.0000 26
53
+ Caterpie 1.0000 1.0000 1.0000 27
54
+ Dratini 0.0000 0.0000 0.0000 26
55
+ Bulbasaur 1.0000 1.0000 1.0000 26
56
+ Tentacool 1.0000 1.0000 1.0000 26
57
+ Gengar 1.0000 1.0000 1.0000 26
58
+ Machop 0.9643 1.0000 0.9818 27
59
+ Raichu 1.0000 1.0000 1.0000 26
60
+ Alolan Sandslash 0.0000 0.0000 0.0000 26
61
+ Eevee 1.0000 1.0000 1.0000 27
62
+ Abra 1.0000 1.0000 1.0000 27
63
+ Haunter 1.0000 1.0000 1.0000 27
64
+ Metapod 1.0000 1.0000 1.0000 27
65
+ Fearow 0.9630 1.0000 0.9811 26
66
+ Nidorina 0.8966 1.0000 0.9455 26
67
+ Zapdos 1.0000 1.0000 1.0000 27
68
+ Ninetales 1.0000 0.9630 0.9811 27
69
+ Chansey 1.0000 1.0000 1.0000 27
70
+ Kangaskhan 0.9630 1.0000 0.9811 26
71
+ Poliwrath 1.0000 0.9630 0.9811 27
72
+ Gyarados 1.0000 1.0000 1.0000 27
73
+ Charmeleon 1.0000 1.0000 1.0000 26
74
+ Vulpix 1.0000 1.0000 1.0000 26
75
+ Pidgeot 1.0000 0.8846 0.9388 26
76
+ Blastoise 0.9630 1.0000 0.9811 26
77
+ Porygon 1.0000 1.0000 1.0000 26
78
+ Psyduck 0.9643 1.0000 0.9818 27
79
+ Dragonair 0.5400 1.0000 0.7013 27
80
+ Raticate 0.9630 1.0000 0.9811 26
81
+ Squirtle 1.0000 0.9615 0.9804 26
82
+ Charizard 1.0000 1.0000 1.0000 26
83
+ Electrode 1.0000 0.9615 0.9804 26
84
+ Flareon 1.0000 1.0000 1.0000 26
85
+ Exeggutor 0.9643 1.0000 0.9818 27
86
+ Pikachu 1.0000 1.0000 1.0000 26
87
+ Wigglytuff 1.0000 1.0000 1.0000 26
88
+ Venusaur 1.0000 0.9615 0.9804 26
89
+ Mewtwo 1.0000 1.0000 1.0000 26
90
+ Clefable 1.0000 1.0000 1.0000 27
91
+ Oddish 1.0000 1.0000 1.0000 26
92
+ Ekans 1.0000 1.0000 1.0000 26
93
+ Shellder 1.0000 1.0000 1.0000 26
94
+ Marowak 0.9130 0.8077 0.8571 26
95
+ Kakuna 1.0000 1.0000 1.0000 27
96
+ Rapidash 1.0000 0.9615 0.9804 26
97
+ Rhydon 1.0000 0.9630 0.9811 27
98
+ Ivysaur 1.0000 1.0000 1.0000 26
99
+ Slowpoke 1.0000 1.0000 1.0000 26
100
+ Lapras 1.0000 1.0000 1.0000 27
101
+ Clefairy 1.0000 1.0000 1.0000 26
102
+ Hitmonlee 1.0000 1.0000 1.0000 26
103
+ Jolteon 1.0000 1.0000 1.0000 26
104
+ Growlithe 1.0000 1.0000 1.0000 27
105
+ Gastly 1.0000 1.0000 1.0000 27
106
+ Aerodactyl 1.0000 1.0000 1.0000 27
107
+ Weedle 1.0000 1.0000 1.0000 26
108
+ Tauros 1.0000 1.0000 1.0000 27
109
+ Seel 0.8929 0.9615 0.9259 26
110
+ Zubat 1.0000 1.0000 1.0000 26
111
+ Meowth 0.0000 0.0000 0.0000 26
112
+ Persian 0.6341 1.0000 0.7761 26
113
+ Articuno 0.9310 1.0000 0.9643 27
114
+ Weezing 0.9643 1.0000 0.9818 27
115
+ Magnemite 1.0000 1.0000 1.0000 27
116
+ Omanyte 0.9630 1.0000 0.9811 26
117
+ Mew 1.0000 1.0000 1.0000 26
118
+ Vileplume 1.0000 1.0000 1.0000 27
119
+ Nidoqueen 0.9615 0.9259 0.9434 27
120
+ Vaporeon 0.9000 1.0000 0.9474 27
121
+ Ponyta 0.9630 1.0000 0.9811 26
122
+ Moltres 1.0000 1.0000 1.0000 27
123
+ Voltorb 0.9630 1.0000 0.9811 26
124
+ Magikarp 1.0000 1.0000 1.0000 27
125
+ Beedrill 1.0000 1.0000 1.0000 26
126
+ Nidoking 1.0000 1.0000 1.0000 27
127
+ Paras 1.0000 1.0000 1.0000 26
128
+ Grimer 1.0000 0.9615 0.9804 26
129
+ Dodrio 1.0000 1.0000 1.0000 26
130
+ Charmander 1.0000 1.0000 1.0000 26
131
+ Muk 1.0000 0.9615 0.9804 26
132
+ Primeape 0.8966 0.9630 0.9286 27
133
+ Victreebel 1.0000 1.0000 1.0000 26
134
+ Golbat 1.0000 1.0000 1.0000 26
135
+ Horsea 1.0000 1.0000 1.0000 27
136
+ Goldeen 1.0000 1.0000 1.0000 27
137
+ Pidgeotto 0.8966 1.0000 0.9455 26
138
+ Koffing 0.9630 1.0000 0.9811 26
139
+ Seadra 0.5870 1.0000 0.7397 27
140
+ Tentacruel 1.0000 1.0000 1.0000 26
141
+ Pinsir 1.0000 1.0000 1.0000 26
142
+ Cloyster 1.0000 1.0000 1.0000 26
143
+ Gloom 1.0000 1.0000 1.0000 26
144
+ Graveler 1.0000 1.0000 1.0000 26
145
+ Magmar 1.0000 1.0000 1.0000 27
146
+ Krabby 0.9286 1.0000 0.9630 26
147
+ Electabuzz 1.0000 1.0000 1.0000 27
148
+ Poliwhirl 0.9643 1.0000 0.9818 27
149
+ Golduck 0.9310 1.0000 0.9643 27
150
+ Onix 1.0000 1.0000 1.0000 27
151
+ Nidorino 1.0000 1.0000 1.0000 27
152
+ Snorlax 0.9630 1.0000 0.9811 26
153
+ Starmie 1.0000 1.0000 1.0000 27
154
+ Slowbro 1.0000 1.0000 1.0000 26
155
+ MrMime 1.0000 1.0000 1.0000 26
156
+ Venonat 1.0000 1.0000 1.0000 27
157
+ Kabutops 1.0000 1.0000 1.0000 26
158
+ Drowzee 1.0000 1.0000 1.0000 26
159
+ Rhyhorn 1.0000 1.0000 1.0000 26
160
+ Tangela 1.0000 1.0000 1.0000 27
161
+ Doduo 1.0000 1.0000 1.0000 27
162
+ Exeggcute 1.0000 1.0000 1.0000 26
163
+ Poliwag 1.0000 1.0000 1.0000 27
164
+ Lickitung 1.0000 1.0000 1.0000 26
165
+ Hypno 0.9286 1.0000 0.9630 26
166
+ Bellsprout 1.0000 1.0000 1.0000 27
167
+ Parasect 1.0000 1.0000 1.0000 26
168
+ Kingler 1.0000 0.9231 0.9600 26
169
+
170
+ accuracy 0.9588 3960
171
+ macro avg 0.9382 0.9583 0.9459 3960
172
+ weighted avg 0.9386 0.9588 0.9463 3960
173
+ ```