Upload ViTForImageClassification

config.json

@@ -0,0 +1,530 @@
+{
+  "_name_or_path": "google/vit-base-patch16-224",
+  "architectures": [
+    "ViTForImageClassification"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "000_airplane",
+    "1": "001_alarm_clock",
+    "2": "002_angel",
+    "3": "003_ant",
+    "4": "004_apple",
+    "5": "005_arm",
+    "6": "006_armchair",
+    "7": "007_ashtray",
+    "8": "008_axe",
+    "9": "009_backpack",
+    "10": "010_banana",
+    "11": "011_barn",
+    "12": "012_baseball_bat",
+    "13": "013_basket",
+    "14": "014_bathtub",
+    "15": "015_bear_(animal)",
+    "16": "016_bed",
+    "17": "017_bee",
+    "18": "018_beer-mug",
+    "19": "019_bell",
+    "20": "020_bench",
+    "21": "021_bicycle",
+    "22": "022_binoculars",
+    "23": "023_blimp",
+    "24": "024_book",
+    "25": "025_bookshelf",
+    "26": "026_boomerang",
+    "27": "027_bottle_opener",
+    "28": "028_bowl",
+    "29": "029_brain",
+    "30": "030_bread",
+    "31": "031_bridge",
+    "32": "032_bulldozer",
+    "33": "033_bus",
+    "34": "034_bush",
+    "35": "035_butterfly",
+    "36": "036_cabinet",
+    "37": "037_cactus",
+    "38": "038_cake",
+    "39": "039_calculator",
+    "40": "040_camel",
+    "41": "041_camera",
+    "42": "042_candle",
+    "43": "043_cannon",
+    "44": "044_canoe",
+    "45": "045_car_(sedan)",
+    "46": "046_carrot",
+    "47": "047_castle",
+    "48": "048_cat",
+    "49": "049_cell_phone",
+    "50": "050_chair",
+    "51": "051_chandelier",
+    "52": "052_church",
+    "53": "053_cigarette",
+    "54": "054_cloud",
+    "55": "055_comb",
+    "56": "056_computer_monitor",
+    "57": "057_computer-mouse",
+    "58": "058_couch",
+    "59": "059_cow",
+    "60": "060_crab",
+    "61": "061_crane_(machine)",
+    "62": "062_crocodile",
+    "63": "063_crown",
+    "64": "064_cup",
+    "65": "065_diamond",
+    "66": "066_dog",
+    "67": "067_dolphin",
+    "68": "068_donut",
+    "69": "069_door",
+    "70": "070_door_handle",
+    "71": "071_dragon",
+    "72": "072_duck",
+    "73": "073_ear",
+    "74": "074_elephant",
+    "75": "075_envelope",
+    "76": "076_eye",
+    "77": "077_eyeglasses",
+    "78": "078_face",
+    "79": "079_fan",
+    "80": "080_feather",
+    "81": "081_fire_hydrant",
+    "82": "082_fish",
+    "83": "083_flashlight",
+    "84": "084_floor_lamp",
+    "85": "085_flower_with_stem",
+    "86": "086_flying_bird",
+    "87": "087_flying_saucer",
+    "88": "088_foot",
+    "89": "089_fork",
+    "90": "090_frog",
+    "91": "091_frying-pan",
+    "92": "092_giraffe",
+    "93": "093_grapes",
+    "94": "094_grenade",
+    "95": "095_guitar",
+    "96": "096_hamburger",
+    "97": "097_hammer",
+    "98": "098_hand",
+    "99": "099_harp",
+    "100": "100_hat",
+    "101": "101_head",
+    "102": "102_head-phones",
+    "103": "103_hedgehog",
+    "104": "104_helicopter",
+    "105": "105_helmet",
+    "106": "106_horse",
+    "107": "107_hot_air_balloon",
+    "108": "108_hot-dog",
+    "109": "109_hourglass",
+    "110": "110_house",
+    "111": "111_human-skeleton",
+    "112": "112_ice-cream-cone",
+    "113": "113_ipod",
+    "114": "114_kangaroo",
+    "115": "115_key",
+    "116": "116_keyboard",
+    "117": "117_knife",
+    "118": "118_ladder",
+    "119": "119_laptop",
+    "120": "120_leaf",
+    "121": "121_lightbulb",
+    "122": "122_lighter",
+    "123": "123_lion",
+    "124": "124_lobster",
+    "125": "125_loudspeaker",
+    "126": "126_mailbox",
+    "127": "127_megaphone",
+    "128": "128_mermaid",
+    "129": "129_microphone",
+    "130": "130_microscope",
+    "131": "131_monkey",
+    "132": "132_moon",
+    "133": "133_mosquito",
+    "134": "134_motorbike",
+    "135": "135_mouse_(animal)",
+    "136": "136_mouth",
+    "137": "137_mug",
+    "138": "138_mushroom",
+    "139": "139_nose",
+    "140": "140_octopus",
+    "141": "141_owl",
+    "142": "142_palm_tree",
+    "143": "143_panda",
+    "144": "144_paper_clip",
+    "145": "145_parachute",
+    "146": "146_parking_meter",
+    "147": "147_parrot",
+    "148": "148_pear",
+    "149": "149_pen",
+    "150": "150_penguin",
+    "151": "151_person_sitting",
+    "152": "152_person_walking",
+    "153": "153_piano",
+    "154": "154_pickup_truck",
+    "155": "155_pig",
+    "156": "156_pigeon",
+    "157": "157_pineapple",
+    "158": "158_pipe_(for_smoking)",
+    "159": "159_pizza",
+    "160": "160_potted_plant",
+    "161": "161_power_outlet",
+    "162": "162_present",
+    "163": "163_pretzel",
+    "164": "164_pumpkin",
+    "165": "165_purse",
+    "166": "166_rabbit",
+    "167": "167_race_car",
+    "168": "168_radio",
+    "169": "169_rainbow",
+    "170": "170_revolver",
+    "171": "171_rifle",
+    "172": "172_rollerblades",
+    "173": "173_rooster",
+    "174": "174_sailboat",
+    "175": "175_santa_claus",
+    "176": "176_satellite",
+    "177": "177_satellite_dish",
+    "178": "178_saxophone",
+    "179": "179_scissors",
+    "180": "180_scorpion",
+    "181": "181_screwdriver",
+    "182": "182_sea_turtle",
+    "183": "183_seagull",
+    "184": "184_shark",
+    "185": "185_sheep",
+    "186": "186_ship",
+    "187": "187_shoe",
+    "188": "188_shovel",
+    "189": "189_skateboard",
+    "190": "190_skull",
+    "191": "191_skyscraper",
+    "192": "192_snail",
+    "193": "193_snake",
+    "194": "194_snowboard",
+    "195": "195_snowman",
+    "196": "196_socks",
+    "197": "197_space_shuttle",
+    "198": "198_speed-boat",
+    "199": "199_spider",
+    "200": "200_sponge_bob",
+    "201": "201_spoon",
+    "202": "202_squirrel",
+    "203": "203_standing_bird",
+    "204": "204_stapler",
+    "205": "205_strawberry",
+    "206": "206_streetlight",
+    "207": "207_submarine",
+    "208": "208_suitcase",
+    "209": "209_sun",
+    "210": "210_suv",
+    "211": "211_swan",
+    "212": "212_sword",
+    "213": "213_syringe",
+    "214": "214_t-shirt",
+    "215": "215_table",
+    "216": "216_tablelamp",
+    "217": "217_teacup",
+    "218": "218_teapot",
+    "219": "219_teddy-bear",
+    "220": "220_telephone",
+    "221": "221_tennis-racket",
+    "222": "222_tent",
+    "223": "223_tiger",
+    "224": "224_tire",
+    "225": "225_toilet",
+    "226": "226_tomato",
+    "227": "227_tooth",
+    "228": "228_toothbrush",
+    "229": "229_tractor",
+    "230": "230_traffic_light",
+    "231": "231_train",
+    "232": "232_tree",
+    "233": "233_trombone",
+    "234": "234_trousers",
+    "235": "235_truck",
+    "236": "236_trumpet",
+    "237": "237_tv",
+    "238": "238_umbrella",
+    "239": "239_van",
+    "240": "240_vase",
+    "241": "241_violin",
+    "242": "242_walkie_talkie",
+    "243": "243_wheel",
+    "244": "244_wheelbarrow",
+    "245": "245_windmill",
+    "246": "246_wine-bottle",
+    "247": "247_wineglass",
+    "248": "248_wrist-watch",
+    "249": "249_zebra",
+    "250": "mistery_category"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "000_airplane": 0,
+    "001_alarm_clock": 1,
+    "002_angel": 2,
+    "003_ant": 3,
+    "004_apple": 4,
+    "005_arm": 5,
+    "006_armchair": 6,
+    "007_ashtray": 7,
+    "008_axe": 8,
+    "009_backpack": 9,
+    "010_banana": 10,
+    "011_barn": 11,
+    "012_baseball_bat": 12,
+    "013_basket": 13,
+    "014_bathtub": 14,
+    "015_bear_(animal)": 15,
+    "016_bed": 16,
+    "017_bee": 17,
+    "018_beer-mug": 18,
+    "019_bell": 19,
+    "020_bench": 20,
+    "021_bicycle": 21,
+    "022_binoculars": 22,
+    "023_blimp": 23,
+    "024_book": 24,
+    "025_bookshelf": 25,
+    "026_boomerang": 26,
+    "027_bottle_opener": 27,
+    "028_bowl": 28,
+    "029_brain": 29,
+    "030_bread": 30,
+    "031_bridge": 31,
+    "032_bulldozer": 32,
+    "033_bus": 33,
+    "034_bush": 34,
+    "035_butterfly": 35,
+    "036_cabinet": 36,
+    "037_cactus": 37,
+    "038_cake": 38,
+    "039_calculator": 39,
+    "040_camel": 40,
+    "041_camera": 41,
+    "042_candle": 42,
+    "043_cannon": 43,
+    "044_canoe": 44,
+    "045_car_(sedan)": 45,
+    "046_carrot": 46,
+    "047_castle": 47,
+    "048_cat": 48,
+    "049_cell_phone": 49,
+    "050_chair": 50,
+    "051_chandelier": 51,
+    "052_church": 52,
+    "053_cigarette": 53,
+    "054_cloud": 54,
+    "055_comb": 55,
+    "056_computer_monitor": 56,
+    "057_computer-mouse": 57,
+    "058_couch": 58,
+    "059_cow": 59,
+    "060_crab": 60,
+    "061_crane_(machine)": 61,
+    "062_crocodile": 62,
+    "063_crown": 63,
+    "064_cup": 64,
+    "065_diamond": 65,
+    "066_dog": 66,
+    "067_dolphin": 67,
+    "068_donut": 68,
+    "069_door": 69,
+    "070_door_handle": 70,
+    "071_dragon": 71,
+    "072_duck": 72,
+    "073_ear": 73,
+    "074_elephant": 74,
+    "075_envelope": 75,
+    "076_eye": 76,
+    "077_eyeglasses": 77,
+    "078_face": 78,
+    "079_fan": 79,
+    "080_feather": 80,
+    "081_fire_hydrant": 81,
+    "082_fish": 82,
+    "083_flashlight": 83,
+    "084_floor_lamp": 84,
+    "085_flower_with_stem": 85,
+    "086_flying_bird": 86,
+    "087_flying_saucer": 87,
+    "088_foot": 88,
+    "089_fork": 89,
+    "090_frog": 90,
+    "091_frying-pan": 91,
+    "092_giraffe": 92,
+    "093_grapes": 93,
+    "094_grenade": 94,
+    "095_guitar": 95,
+    "096_hamburger": 96,
+    "097_hammer": 97,
+    "098_hand": 98,
+    "099_harp": 99,
+    "100_hat": 100,
+    "101_head": 101,
+    "102_head-phones": 102,
+    "103_hedgehog": 103,
+    "104_helicopter": 104,
+    "105_helmet": 105,
+    "106_horse": 106,
+    "107_hot_air_balloon": 107,
+    "108_hot-dog": 108,
+    "109_hourglass": 109,
+    "110_house": 110,
+    "111_human-skeleton": 111,
+    "112_ice-cream-cone": 112,
+    "113_ipod": 113,
+    "114_kangaroo": 114,
+    "115_key": 115,
+    "116_keyboard": 116,
+    "117_knife": 117,
+    "118_ladder": 118,
+    "119_laptop": 119,
+    "120_leaf": 120,
+    "121_lightbulb": 121,
+    "122_lighter": 122,
+    "123_lion": 123,
+    "124_lobster": 124,
+    "125_loudspeaker": 125,
+    "126_mailbox": 126,
+    "127_megaphone": 127,
+    "128_mermaid": 128,
+    "129_microphone": 129,
+    "130_microscope": 130,
+    "131_monkey": 131,
+    "132_moon": 132,
+    "133_mosquito": 133,
+    "134_motorbike": 134,
+    "135_mouse_(animal)": 135,
+    "136_mouth": 136,
+    "137_mug": 137,
+    "138_mushroom": 138,
+    "139_nose": 139,
+    "140_octopus": 140,
+    "141_owl": 141,
+    "142_palm_tree": 142,
+    "143_panda": 143,
+    "144_paper_clip": 144,
+    "145_parachute": 145,
+    "146_parking_meter": 146,
+    "147_parrot": 147,
+    "148_pear": 148,
+    "149_pen": 149,
+    "150_penguin": 150,
+    "151_person_sitting": 151,
+    "152_person_walking": 152,
+    "153_piano": 153,
+    "154_pickup_truck": 154,
+    "155_pig": 155,
+    "156_pigeon": 156,
+    "157_pineapple": 157,
+    "158_pipe_(for_smoking)": 158,
+    "159_pizza": 159,
+    "160_potted_plant": 160,
+    "161_power_outlet": 161,
+    "162_present": 162,
+    "163_pretzel": 163,
+    "164_pumpkin": 164,
+    "165_purse": 165,
+    "166_rabbit": 166,
+    "167_race_car": 167,
+    "168_radio": 168,
+    "169_rainbow": 169,
+    "170_revolver": 170,
+    "171_rifle": 171,
+    "172_rollerblades": 172,
+    "173_rooster": 173,
+    "174_sailboat": 174,
+    "175_santa_claus": 175,
+    "176_satellite": 176,
+    "177_satellite_dish": 177,
+    "178_saxophone": 178,
+    "179_scissors": 179,
+    "180_scorpion": 180,
+    "181_screwdriver": 181,
+    "182_sea_turtle": 182,
+    "183_seagull": 183,
+    "184_shark": 184,
+    "185_sheep": 185,
+    "186_ship": 186,
+    "187_shoe": 187,
+    "188_shovel": 188,
+    "189_skateboard": 189,
+    "190_skull": 190,
+    "191_skyscraper": 191,
+    "192_snail": 192,
+    "193_snake": 193,
+    "194_snowboard": 194,
+    "195_snowman": 195,
+    "196_socks": 196,
+    "197_space_shuttle": 197,
+    "198_speed-boat": 198,
+    "199_spider": 199,
+    "200_sponge_bob": 200,
+    "201_spoon": 201,
+    "202_squirrel": 202,
+    "203_standing_bird": 203,
+    "204_stapler": 204,
+    "205_strawberry": 205,
+    "206_streetlight": 206,
+    "207_submarine": 207,
+    "208_suitcase": 208,
+    "209_sun": 209,
+    "210_suv": 210,
+    "211_swan": 211,
+    "212_sword": 212,
+    "213_syringe": 213,
+    "214_t-shirt": 214,
+    "215_table": 215,
+    "216_tablelamp": 216,
+    "217_teacup": 217,
+    "218_teapot": 218,
+    "219_teddy-bear": 219,
+    "220_telephone": 220,
+    "221_tennis-racket": 221,
+    "222_tent": 222,
+    "223_tiger": 223,
+    "224_tire": 224,
+    "225_toilet": 225,
+    "226_tomato": 226,
+    "227_tooth": 227,
+    "228_toothbrush": 228,
+    "229_tractor": 229,
+    "230_traffic_light": 230,
+    "231_train": 231,
+    "232_tree": 232,
+    "233_trombone": 233,
+    "234_trousers": 234,
+    "235_truck": 235,
+    "236_trumpet": 236,
+    "237_tv": 237,
+    "238_umbrella": 238,
+    "239_van": 239,
+    "240_vase": 240,
+    "241_violin": 241,
+    "242_walkie_talkie": 242,
+    "243_wheel": 243,
+    "244_wheelbarrow": 244,
+    "245_windmill": 245,
+    "246_wine-bottle": 246,
+    "247_wineglass": 247,
+    "248_wrist-watch": 248,
+    "249_zebra": 249,
+    "mistery_category": 250
+  },
+  "layer_norm_eps": 1e-12,
+  "model_type": "vit",
+  "num_attention_heads": 12,
+  "num_channels": 3,
+  "num_hidden_layers": 12,
+  "patch_size": 16,
+  "problem_type": "single_label_classification",
+  "qkv_bias": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.35.2"
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9804a63ea574354fbdee7b965aa1b694b3cc5ded16b3637b6fa9d75b6a5e94c2
+size 343989916