qbourbon commited on
Commit
545cd5c
1 Parent(s): c7d3488

Upload ViTForImageClassification

Browse files
Files changed (2) hide show
  1. config.json +534 -0
  2. model.safetensors +3 -0
config.json ADDED
@@ -0,0 +1,534 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "facebook/vit-mae-base",
3
+ "architectures": [
4
+ "ViTForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "decoder_hidden_size": 512,
8
+ "decoder_intermediate_size": 2048,
9
+ "decoder_num_attention_heads": 16,
10
+ "decoder_num_hidden_layers": 8,
11
+ "encoder_stride": 16,
12
+ "hidden_act": "gelu",
13
+ "hidden_dropout_prob": 0.0,
14
+ "hidden_size": 768,
15
+ "id2label": {
16
+ "0": "000_airplane",
17
+ "1": "001_alarm_clock",
18
+ "2": "002_angel",
19
+ "3": "003_ant",
20
+ "4": "004_apple",
21
+ "5": "005_arm",
22
+ "6": "006_armchair",
23
+ "7": "007_ashtray",
24
+ "8": "008_axe",
25
+ "9": "009_backpack",
26
+ "10": "010_banana",
27
+ "11": "011_barn",
28
+ "12": "012_baseball_bat",
29
+ "13": "013_basket",
30
+ "14": "014_bathtub",
31
+ "15": "015_bear_(animal)",
32
+ "16": "016_bed",
33
+ "17": "017_bee",
34
+ "18": "018_beer-mug",
35
+ "19": "019_bell",
36
+ "20": "020_bench",
37
+ "21": "021_bicycle",
38
+ "22": "022_binoculars",
39
+ "23": "023_blimp",
40
+ "24": "024_book",
41
+ "25": "025_bookshelf",
42
+ "26": "026_boomerang",
43
+ "27": "027_bottle_opener",
44
+ "28": "028_bowl",
45
+ "29": "029_brain",
46
+ "30": "030_bread",
47
+ "31": "031_bridge",
48
+ "32": "032_bulldozer",
49
+ "33": "033_bus",
50
+ "34": "034_bush",
51
+ "35": "035_butterfly",
52
+ "36": "036_cabinet",
53
+ "37": "037_cactus",
54
+ "38": "038_cake",
55
+ "39": "039_calculator",
56
+ "40": "040_camel",
57
+ "41": "041_camera",
58
+ "42": "042_candle",
59
+ "43": "043_cannon",
60
+ "44": "044_canoe",
61
+ "45": "045_car_(sedan)",
62
+ "46": "046_carrot",
63
+ "47": "047_castle",
64
+ "48": "048_cat",
65
+ "49": "049_cell_phone",
66
+ "50": "050_chair",
67
+ "51": "051_chandelier",
68
+ "52": "052_church",
69
+ "53": "053_cigarette",
70
+ "54": "054_cloud",
71
+ "55": "055_comb",
72
+ "56": "056_computer_monitor",
73
+ "57": "057_computer-mouse",
74
+ "58": "058_couch",
75
+ "59": "059_cow",
76
+ "60": "060_crab",
77
+ "61": "061_crane_(machine)",
78
+ "62": "062_crocodile",
79
+ "63": "063_crown",
80
+ "64": "064_cup",
81
+ "65": "065_diamond",
82
+ "66": "066_dog",
83
+ "67": "067_dolphin",
84
+ "68": "068_donut",
85
+ "69": "069_door",
86
+ "70": "070_door_handle",
87
+ "71": "071_dragon",
88
+ "72": "072_duck",
89
+ "73": "073_ear",
90
+ "74": "074_elephant",
91
+ "75": "075_envelope",
92
+ "76": "076_eye",
93
+ "77": "077_eyeglasses",
94
+ "78": "078_face",
95
+ "79": "079_fan",
96
+ "80": "080_feather",
97
+ "81": "081_fire_hydrant",
98
+ "82": "082_fish",
99
+ "83": "083_flashlight",
100
+ "84": "084_floor_lamp",
101
+ "85": "085_flower_with_stem",
102
+ "86": "086_flying_bird",
103
+ "87": "087_flying_saucer",
104
+ "88": "088_foot",
105
+ "89": "089_fork",
106
+ "90": "090_frog",
107
+ "91": "091_frying-pan",
108
+ "92": "092_giraffe",
109
+ "93": "093_grapes",
110
+ "94": "094_grenade",
111
+ "95": "095_guitar",
112
+ "96": "096_hamburger",
113
+ "97": "097_hammer",
114
+ "98": "098_hand",
115
+ "99": "099_harp",
116
+ "100": "100_hat",
117
+ "101": "101_head",
118
+ "102": "102_head-phones",
119
+ "103": "103_hedgehog",
120
+ "104": "104_helicopter",
121
+ "105": "105_helmet",
122
+ "106": "106_horse",
123
+ "107": "107_hot_air_balloon",
124
+ "108": "108_hot-dog",
125
+ "109": "109_hourglass",
126
+ "110": "110_house",
127
+ "111": "111_human-skeleton",
128
+ "112": "112_ice-cream-cone",
129
+ "113": "113_ipod",
130
+ "114": "114_kangaroo",
131
+ "115": "115_key",
132
+ "116": "116_keyboard",
133
+ "117": "117_knife",
134
+ "118": "118_ladder",
135
+ "119": "119_laptop",
136
+ "120": "120_leaf",
137
+ "121": "121_lightbulb",
138
+ "122": "122_lighter",
139
+ "123": "123_lion",
140
+ "124": "124_lobster",
141
+ "125": "125_loudspeaker",
142
+ "126": "126_mailbox",
143
+ "127": "127_megaphone",
144
+ "128": "128_mermaid",
145
+ "129": "129_microphone",
146
+ "130": "130_microscope",
147
+ "131": "131_monkey",
148
+ "132": "132_moon",
149
+ "133": "133_mosquito",
150
+ "134": "134_motorbike",
151
+ "135": "135_mouse_(animal)",
152
+ "136": "136_mouth",
153
+ "137": "137_mug",
154
+ "138": "138_mushroom",
155
+ "139": "139_nose",
156
+ "140": "140_octopus",
157
+ "141": "141_owl",
158
+ "142": "142_palm_tree",
159
+ "143": "143_panda",
160
+ "144": "144_paper_clip",
161
+ "145": "145_parachute",
162
+ "146": "146_parking_meter",
163
+ "147": "147_parrot",
164
+ "148": "148_pear",
165
+ "149": "149_pen",
166
+ "150": "150_penguin",
167
+ "151": "151_person_sitting",
168
+ "152": "152_person_walking",
169
+ "153": "153_piano",
170
+ "154": "154_pickup_truck",
171
+ "155": "155_pig",
172
+ "156": "156_pigeon",
173
+ "157": "157_pineapple",
174
+ "158": "158_pipe_(for_smoking)",
175
+ "159": "159_pizza",
176
+ "160": "160_potted_plant",
177
+ "161": "161_power_outlet",
178
+ "162": "162_present",
179
+ "163": "163_pretzel",
180
+ "164": "164_pumpkin",
181
+ "165": "165_purse",
182
+ "166": "166_rabbit",
183
+ "167": "167_race_car",
184
+ "168": "168_radio",
185
+ "169": "169_rainbow",
186
+ "170": "170_revolver",
187
+ "171": "171_rifle",
188
+ "172": "172_rollerblades",
189
+ "173": "173_rooster",
190
+ "174": "174_sailboat",
191
+ "175": "175_santa_claus",
192
+ "176": "176_satellite",
193
+ "177": "177_satellite_dish",
194
+ "178": "178_saxophone",
195
+ "179": "179_scissors",
196
+ "180": "180_scorpion",
197
+ "181": "181_screwdriver",
198
+ "182": "182_sea_turtle",
199
+ "183": "183_seagull",
200
+ "184": "184_shark",
201
+ "185": "185_sheep",
202
+ "186": "186_ship",
203
+ "187": "187_shoe",
204
+ "188": "188_shovel",
205
+ "189": "189_skateboard",
206
+ "190": "190_skull",
207
+ "191": "191_skyscraper",
208
+ "192": "192_snail",
209
+ "193": "193_snake",
210
+ "194": "194_snowboard",
211
+ "195": "195_snowman",
212
+ "196": "196_socks",
213
+ "197": "197_space_shuttle",
214
+ "198": "198_speed-boat",
215
+ "199": "199_spider",
216
+ "200": "200_sponge_bob",
217
+ "201": "201_spoon",
218
+ "202": "202_squirrel",
219
+ "203": "203_standing_bird",
220
+ "204": "204_stapler",
221
+ "205": "205_strawberry",
222
+ "206": "206_streetlight",
223
+ "207": "207_submarine",
224
+ "208": "208_suitcase",
225
+ "209": "209_sun",
226
+ "210": "210_suv",
227
+ "211": "211_swan",
228
+ "212": "212_sword",
229
+ "213": "213_syringe",
230
+ "214": "214_t-shirt",
231
+ "215": "215_table",
232
+ "216": "216_tablelamp",
233
+ "217": "217_teacup",
234
+ "218": "218_teapot",
235
+ "219": "219_teddy-bear",
236
+ "220": "220_telephone",
237
+ "221": "221_tennis-racket",
238
+ "222": "222_tent",
239
+ "223": "223_tiger",
240
+ "224": "224_tire",
241
+ "225": "225_toilet",
242
+ "226": "226_tomato",
243
+ "227": "227_tooth",
244
+ "228": "228_toothbrush",
245
+ "229": "229_tractor",
246
+ "230": "230_traffic_light",
247
+ "231": "231_train",
248
+ "232": "232_tree",
249
+ "233": "233_trombone",
250
+ "234": "234_trousers",
251
+ "235": "235_truck",
252
+ "236": "236_trumpet",
253
+ "237": "237_tv",
254
+ "238": "238_umbrella",
255
+ "239": "239_van",
256
+ "240": "240_vase",
257
+ "241": "241_violin",
258
+ "242": "242_walkie_talkie",
259
+ "243": "243_wheel",
260
+ "244": "244_wheelbarrow",
261
+ "245": "245_windmill",
262
+ "246": "246_wine-bottle",
263
+ "247": "247_wineglass",
264
+ "248": "248_wrist-watch",
265
+ "249": "249_zebra"
266
+ },
267
+ "image_size": 224,
268
+ "initializer_range": 0.02,
269
+ "intermediate_size": 3072,
270
+ "label2id": {
271
+ "000_airplane": 0,
272
+ "001_alarm_clock": 1,
273
+ "002_angel": 2,
274
+ "003_ant": 3,
275
+ "004_apple": 4,
276
+ "005_arm": 5,
277
+ "006_armchair": 6,
278
+ "007_ashtray": 7,
279
+ "008_axe": 8,
280
+ "009_backpack": 9,
281
+ "010_banana": 10,
282
+ "011_barn": 11,
283
+ "012_baseball_bat": 12,
284
+ "013_basket": 13,
285
+ "014_bathtub": 14,
286
+ "015_bear_(animal)": 15,
287
+ "016_bed": 16,
288
+ "017_bee": 17,
289
+ "018_beer-mug": 18,
290
+ "019_bell": 19,
291
+ "020_bench": 20,
292
+ "021_bicycle": 21,
293
+ "022_binoculars": 22,
294
+ "023_blimp": 23,
295
+ "024_book": 24,
296
+ "025_bookshelf": 25,
297
+ "026_boomerang": 26,
298
+ "027_bottle_opener": 27,
299
+ "028_bowl": 28,
300
+ "029_brain": 29,
301
+ "030_bread": 30,
302
+ "031_bridge": 31,
303
+ "032_bulldozer": 32,
304
+ "033_bus": 33,
305
+ "034_bush": 34,
306
+ "035_butterfly": 35,
307
+ "036_cabinet": 36,
308
+ "037_cactus": 37,
309
+ "038_cake": 38,
310
+ "039_calculator": 39,
311
+ "040_camel": 40,
312
+ "041_camera": 41,
313
+ "042_candle": 42,
314
+ "043_cannon": 43,
315
+ "044_canoe": 44,
316
+ "045_car_(sedan)": 45,
317
+ "046_carrot": 46,
318
+ "047_castle": 47,
319
+ "048_cat": 48,
320
+ "049_cell_phone": 49,
321
+ "050_chair": 50,
322
+ "051_chandelier": 51,
323
+ "052_church": 52,
324
+ "053_cigarette": 53,
325
+ "054_cloud": 54,
326
+ "055_comb": 55,
327
+ "056_computer_monitor": 56,
328
+ "057_computer-mouse": 57,
329
+ "058_couch": 58,
330
+ "059_cow": 59,
331
+ "060_crab": 60,
332
+ "061_crane_(machine)": 61,
333
+ "062_crocodile": 62,
334
+ "063_crown": 63,
335
+ "064_cup": 64,
336
+ "065_diamond": 65,
337
+ "066_dog": 66,
338
+ "067_dolphin": 67,
339
+ "068_donut": 68,
340
+ "069_door": 69,
341
+ "070_door_handle": 70,
342
+ "071_dragon": 71,
343
+ "072_duck": 72,
344
+ "073_ear": 73,
345
+ "074_elephant": 74,
346
+ "075_envelope": 75,
347
+ "076_eye": 76,
348
+ "077_eyeglasses": 77,
349
+ "078_face": 78,
350
+ "079_fan": 79,
351
+ "080_feather": 80,
352
+ "081_fire_hydrant": 81,
353
+ "082_fish": 82,
354
+ "083_flashlight": 83,
355
+ "084_floor_lamp": 84,
356
+ "085_flower_with_stem": 85,
357
+ "086_flying_bird": 86,
358
+ "087_flying_saucer": 87,
359
+ "088_foot": 88,
360
+ "089_fork": 89,
361
+ "090_frog": 90,
362
+ "091_frying-pan": 91,
363
+ "092_giraffe": 92,
364
+ "093_grapes": 93,
365
+ "094_grenade": 94,
366
+ "095_guitar": 95,
367
+ "096_hamburger": 96,
368
+ "097_hammer": 97,
369
+ "098_hand": 98,
370
+ "099_harp": 99,
371
+ "100_hat": 100,
372
+ "101_head": 101,
373
+ "102_head-phones": 102,
374
+ "103_hedgehog": 103,
375
+ "104_helicopter": 104,
376
+ "105_helmet": 105,
377
+ "106_horse": 106,
378
+ "107_hot_air_balloon": 107,
379
+ "108_hot-dog": 108,
380
+ "109_hourglass": 109,
381
+ "110_house": 110,
382
+ "111_human-skeleton": 111,
383
+ "112_ice-cream-cone": 112,
384
+ "113_ipod": 113,
385
+ "114_kangaroo": 114,
386
+ "115_key": 115,
387
+ "116_keyboard": 116,
388
+ "117_knife": 117,
389
+ "118_ladder": 118,
390
+ "119_laptop": 119,
391
+ "120_leaf": 120,
392
+ "121_lightbulb": 121,
393
+ "122_lighter": 122,
394
+ "123_lion": 123,
395
+ "124_lobster": 124,
396
+ "125_loudspeaker": 125,
397
+ "126_mailbox": 126,
398
+ "127_megaphone": 127,
399
+ "128_mermaid": 128,
400
+ "129_microphone": 129,
401
+ "130_microscope": 130,
402
+ "131_monkey": 131,
403
+ "132_moon": 132,
404
+ "133_mosquito": 133,
405
+ "134_motorbike": 134,
406
+ "135_mouse_(animal)": 135,
407
+ "136_mouth": 136,
408
+ "137_mug": 137,
409
+ "138_mushroom": 138,
410
+ "139_nose": 139,
411
+ "140_octopus": 140,
412
+ "141_owl": 141,
413
+ "142_palm_tree": 142,
414
+ "143_panda": 143,
415
+ "144_paper_clip": 144,
416
+ "145_parachute": 145,
417
+ "146_parking_meter": 146,
418
+ "147_parrot": 147,
419
+ "148_pear": 148,
420
+ "149_pen": 149,
421
+ "150_penguin": 150,
422
+ "151_person_sitting": 151,
423
+ "152_person_walking": 152,
424
+ "153_piano": 153,
425
+ "154_pickup_truck": 154,
426
+ "155_pig": 155,
427
+ "156_pigeon": 156,
428
+ "157_pineapple": 157,
429
+ "158_pipe_(for_smoking)": 158,
430
+ "159_pizza": 159,
431
+ "160_potted_plant": 160,
432
+ "161_power_outlet": 161,
433
+ "162_present": 162,
434
+ "163_pretzel": 163,
435
+ "164_pumpkin": 164,
436
+ "165_purse": 165,
437
+ "166_rabbit": 166,
438
+ "167_race_car": 167,
439
+ "168_radio": 168,
440
+ "169_rainbow": 169,
441
+ "170_revolver": 170,
442
+ "171_rifle": 171,
443
+ "172_rollerblades": 172,
444
+ "173_rooster": 173,
445
+ "174_sailboat": 174,
446
+ "175_santa_claus": 175,
447
+ "176_satellite": 176,
448
+ "177_satellite_dish": 177,
449
+ "178_saxophone": 178,
450
+ "179_scissors": 179,
451
+ "180_scorpion": 180,
452
+ "181_screwdriver": 181,
453
+ "182_sea_turtle": 182,
454
+ "183_seagull": 183,
455
+ "184_shark": 184,
456
+ "185_sheep": 185,
457
+ "186_ship": 186,
458
+ "187_shoe": 187,
459
+ "188_shovel": 188,
460
+ "189_skateboard": 189,
461
+ "190_skull": 190,
462
+ "191_skyscraper": 191,
463
+ "192_snail": 192,
464
+ "193_snake": 193,
465
+ "194_snowboard": 194,
466
+ "195_snowman": 195,
467
+ "196_socks": 196,
468
+ "197_space_shuttle": 197,
469
+ "198_speed-boat": 198,
470
+ "199_spider": 199,
471
+ "200_sponge_bob": 200,
472
+ "201_spoon": 201,
473
+ "202_squirrel": 202,
474
+ "203_standing_bird": 203,
475
+ "204_stapler": 204,
476
+ "205_strawberry": 205,
477
+ "206_streetlight": 206,
478
+ "207_submarine": 207,
479
+ "208_suitcase": 208,
480
+ "209_sun": 209,
481
+ "210_suv": 210,
482
+ "211_swan": 211,
483
+ "212_sword": 212,
484
+ "213_syringe": 213,
485
+ "214_t-shirt": 214,
486
+ "215_table": 215,
487
+ "216_tablelamp": 216,
488
+ "217_teacup": 217,
489
+ "218_teapot": 218,
490
+ "219_teddy-bear": 219,
491
+ "220_telephone": 220,
492
+ "221_tennis-racket": 221,
493
+ "222_tent": 222,
494
+ "223_tiger": 223,
495
+ "224_tire": 224,
496
+ "225_toilet": 225,
497
+ "226_tomato": 226,
498
+ "227_tooth": 227,
499
+ "228_toothbrush": 228,
500
+ "229_tractor": 229,
501
+ "230_traffic_light": 230,
502
+ "231_train": 231,
503
+ "232_tree": 232,
504
+ "233_trombone": 233,
505
+ "234_trousers": 234,
506
+ "235_truck": 235,
507
+ "236_trumpet": 236,
508
+ "237_tv": 237,
509
+ "238_umbrella": 238,
510
+ "239_van": 239,
511
+ "240_vase": 240,
512
+ "241_violin": 241,
513
+ "242_walkie_talkie": 242,
514
+ "243_wheel": 243,
515
+ "244_wheelbarrow": 244,
516
+ "245_windmill": 245,
517
+ "246_wine-bottle": 246,
518
+ "247_wineglass": 247,
519
+ "248_wrist-watch": 248,
520
+ "249_zebra": 249
521
+ },
522
+ "layer_norm_eps": 1e-12,
523
+ "mask_ratio": 0.75,
524
+ "model_type": "vit",
525
+ "norm_pix_loss": false,
526
+ "num_attention_heads": 12,
527
+ "num_channels": 3,
528
+ "num_hidden_layers": 12,
529
+ "patch_size": 16,
530
+ "problem_type": "single_label_classification",
531
+ "qkv_bias": true,
532
+ "torch_dtype": "float32",
533
+ "transformers_version": "4.35.2"
534
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fe48935f06b930d9f05cfa7142e13b276523f87123e4c5a679e4a13e0e965f2d
3
+ size 343986840