qbourbon commited on
Commit
2861c6a
1 Parent(s): c9000d3

Upload ConvNextForImageClassification

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