|
{ |
|
"_name_or_path": "apple/mobilevitv2-1.0-imagenet1k-256", |
|
"architectures": [ |
|
"MobileViTV2ForImageClassification" |
|
], |
|
"aspp_dropout_prob": 0.1, |
|
"aspp_out_channels": 512, |
|
"atrous_rates": [ |
|
6, |
|
12, |
|
18 |
|
], |
|
"attn_dropout": 0.0, |
|
"base_attn_unit_dims": [ |
|
128, |
|
192, |
|
256 |
|
], |
|
"classifier_dropout_prob": 0.1, |
|
"conv_kernel_size": 3, |
|
"expand_ratio": 2.0, |
|
"ffn_dropout": 0.0, |
|
"ffn_multiplier": 2, |
|
"hidden_act": "swish", |
|
"id2label": { |
|
"0": "aircraft carrier", |
|
"1": "airplane", |
|
"10": "asparagus", |
|
"100": "dumbbell", |
|
"101": "ear", |
|
"102": "elbow", |
|
"103": "elephant", |
|
"104": "envelope", |
|
"105": "eraser", |
|
"106": "eye", |
|
"107": "eyeglasses", |
|
"108": "face", |
|
"109": "fan", |
|
"11": "axe", |
|
"110": "feather", |
|
"111": "fence", |
|
"112": "finger", |
|
"113": "fire hydrant", |
|
"114": "fireplace", |
|
"115": "firetruck", |
|
"116": "fish", |
|
"117": "flamingo", |
|
"118": "flashlight", |
|
"119": "flip flops", |
|
"12": "backpack", |
|
"120": "floor lamp", |
|
"121": "flower", |
|
"122": "flying saucer", |
|
"123": "foot", |
|
"124": "fork", |
|
"125": "frog", |
|
"126": "frying pan", |
|
"127": "garden hose", |
|
"128": "garden", |
|
"129": "giraffe", |
|
"13": "banana", |
|
"130": "goatee", |
|
"131": "golf club", |
|
"132": "grapes", |
|
"133": "grass", |
|
"134": "guitar", |
|
"135": "hamburger", |
|
"136": "hammer", |
|
"137": "hand", |
|
"138": "harp", |
|
"139": "hat", |
|
"14": "bandage", |
|
"140": "headphones", |
|
"141": "hedgehog", |
|
"142": "helicopter", |
|
"143": "helmet", |
|
"144": "hexagon", |
|
"145": "hockey puck", |
|
"146": "hockey stick", |
|
"147": "horse", |
|
"148": "hospital", |
|
"149": "hot air balloon", |
|
"15": "barn", |
|
"150": "hot dog", |
|
"151": "hot tub", |
|
"152": "hourglass", |
|
"153": "house plant", |
|
"154": "house", |
|
"155": "hurricane", |
|
"156": "ice cream", |
|
"157": "jacket", |
|
"158": "jail", |
|
"159": "kangaroo", |
|
"16": "baseball bat", |
|
"160": "key", |
|
"161": "keyboard", |
|
"162": "knee", |
|
"163": "knife", |
|
"164": "ladder", |
|
"165": "lantern", |
|
"166": "laptop", |
|
"167": "leaf", |
|
"168": "leg", |
|
"169": "light bulb", |
|
"17": "baseball", |
|
"170": "lighter", |
|
"171": "lighthouse", |
|
"172": "lightning", |
|
"173": "line", |
|
"174": "lion", |
|
"175": "lipstick", |
|
"176": "lobster", |
|
"177": "lollipop", |
|
"178": "mailbox", |
|
"179": "map", |
|
"18": "basket", |
|
"180": "marker", |
|
"181": "matches", |
|
"182": "megaphone", |
|
"183": "mermaid", |
|
"184": "microphone", |
|
"185": "microwave", |
|
"186": "monkey", |
|
"187": "moon", |
|
"188": "mosquito", |
|
"189": "motorbike", |
|
"19": "basketball", |
|
"190": "mountain", |
|
"191": "mouse", |
|
"192": "moustache", |
|
"193": "mouth", |
|
"194": "mug", |
|
"195": "mushroom", |
|
"196": "nail", |
|
"197": "necklace", |
|
"198": "nose", |
|
"199": "ocean", |
|
"2": "alarm clock", |
|
"20": "bat", |
|
"200": "octagon", |
|
"201": "octopus", |
|
"202": "onion", |
|
"203": "oven", |
|
"204": "owl", |
|
"205": "paint can", |
|
"206": "paintbrush", |
|
"207": "palm tree", |
|
"208": "panda", |
|
"209": "pants", |
|
"21": "bathtub", |
|
"210": "paper clip", |
|
"211": "parachute", |
|
"212": "parrot", |
|
"213": "passport", |
|
"214": "peanut", |
|
"215": "pear", |
|
"216": "peas", |
|
"217": "pencil", |
|
"218": "penguin", |
|
"219": "piano", |
|
"22": "beach", |
|
"220": "pickup truck", |
|
"221": "picture frame", |
|
"222": "pig", |
|
"223": "pillow", |
|
"224": "pineapple", |
|
"225": "pizza", |
|
"226": "pliers", |
|
"227": "police car", |
|
"228": "pond", |
|
"229": "pool", |
|
"23": "bear", |
|
"230": "popsicle", |
|
"231": "postcard", |
|
"232": "potato", |
|
"233": "power outlet", |
|
"234": "purse", |
|
"235": "rabbit", |
|
"236": "raccoon", |
|
"237": "radio", |
|
"238": "rain", |
|
"239": "rainbow", |
|
"24": "beard", |
|
"240": "rake", |
|
"241": "remote control", |
|
"242": "rhinoceros", |
|
"243": "rifle", |
|
"244": "river", |
|
"245": "roller coaster", |
|
"246": "rollerskates", |
|
"247": "sailboat", |
|
"248": "sandwich", |
|
"249": "saw", |
|
"25": "bed", |
|
"250": "saxophone", |
|
"251": "school bus", |
|
"252": "scissors", |
|
"253": "scorpion", |
|
"254": "screwdriver", |
|
"255": "sea turtle", |
|
"256": "see saw", |
|
"257": "shark", |
|
"258": "sheep", |
|
"259": "shoe", |
|
"26": "bee", |
|
"260": "shorts", |
|
"261": "shovel", |
|
"262": "sink", |
|
"263": "skateboard", |
|
"264": "skull", |
|
"265": "skyscraper", |
|
"266": "sleeping bag", |
|
"267": "smiley face", |
|
"268": "snail", |
|
"269": "snake", |
|
"27": "belt", |
|
"270": "snorkel", |
|
"271": "snowflake", |
|
"272": "snowman", |
|
"273": "soccer ball", |
|
"274": "sock", |
|
"275": "speedboat", |
|
"276": "spider", |
|
"277": "spoon", |
|
"278": "spreadsheet", |
|
"279": "square", |
|
"28": "bench", |
|
"280": "squiggle", |
|
"281": "squirrel", |
|
"282": "stairs", |
|
"283": "star", |
|
"284": "steak", |
|
"285": "stereo", |
|
"286": "stethoscope", |
|
"287": "stitches", |
|
"288": "stop sign", |
|
"289": "stove", |
|
"29": "bicycle", |
|
"290": "strawberry", |
|
"291": "streetlight", |
|
"292": "string bean", |
|
"293": "submarine", |
|
"294": "suitcase", |
|
"295": "sun", |
|
"296": "swan", |
|
"297": "sweater", |
|
"298": "swing set", |
|
"299": "sword", |
|
"3": "ambulance", |
|
"30": "binoculars", |
|
"300": "syringe", |
|
"301": "t-shirt", |
|
"302": "table", |
|
"303": "teapot", |
|
"304": "teddy-bear", |
|
"305": "telephone", |
|
"306": "television", |
|
"307": "tennis racquet", |
|
"308": "tent", |
|
"309": "The Eiffel Tower", |
|
"31": "bird", |
|
"310": "The Great Wall of China", |
|
"311": "The Mona Lisa", |
|
"312": "tiger", |
|
"313": "toaster", |
|
"314": "toe", |
|
"315": "toilet", |
|
"316": "tooth", |
|
"317": "toothbrush", |
|
"318": "toothpaste", |
|
"319": "tornado", |
|
"32": "birthday cake", |
|
"320": "tractor", |
|
"321": "traffic light", |
|
"322": "train", |
|
"323": "tree", |
|
"324": "triangle", |
|
"325": "trombone", |
|
"326": "truck", |
|
"327": "trumpet", |
|
"328": "umbrella", |
|
"329": "underwear", |
|
"33": "blackberry", |
|
"330": "van", |
|
"331": "vase", |
|
"332": "violin", |
|
"333": "washing machine", |
|
"334": "watermelon", |
|
"335": "waterslide", |
|
"336": "whale", |
|
"337": "wheel", |
|
"338": "windmill", |
|
"339": "wine bottle", |
|
"34": "blueberry", |
|
"340": "wine glass", |
|
"341": "wristwatch", |
|
"342": "yoga", |
|
"343": "zebra", |
|
"344": "zigzag", |
|
"35": "book", |
|
"36": "boomerang", |
|
"37": "bottlecap", |
|
"38": "bowtie", |
|
"39": "bracelet", |
|
"4": "angel", |
|
"40": "brain", |
|
"41": "bread", |
|
"42": "bridge", |
|
"43": "broccoli", |
|
"44": "broom", |
|
"45": "bucket", |
|
"46": "bulldozer", |
|
"47": "bus", |
|
"48": "bush", |
|
"49": "butterfly", |
|
"5": "animal migration", |
|
"50": "cactus", |
|
"51": "cake", |
|
"52": "calculator", |
|
"53": "calendar", |
|
"54": "camel", |
|
"55": "camera", |
|
"56": "camouflage", |
|
"57": "campfire", |
|
"58": "candle", |
|
"59": "cannon", |
|
"6": "ant", |
|
"60": "canoe", |
|
"61": "car", |
|
"62": "carrot", |
|
"63": "castle", |
|
"64": "cat", |
|
"65": "ceiling fan", |
|
"66": "cell phone", |
|
"67": "cello", |
|
"68": "chair", |
|
"69": "chandelier", |
|
"7": "anvil", |
|
"70": "church", |
|
"71": "circle", |
|
"72": "clarinet", |
|
"73": "clock", |
|
"74": "cloud", |
|
"75": "coffee cup", |
|
"76": "compass", |
|
"77": "computer", |
|
"78": "cookie", |
|
"79": "cooler", |
|
"8": "apple", |
|
"80": "couch", |
|
"81": "cow", |
|
"82": "crab", |
|
"83": "crayon", |
|
"84": "crocodile", |
|
"85": "crown", |
|
"86": "cruise ship", |
|
"87": "cup", |
|
"88": "diamond", |
|
"89": "dishwasher", |
|
"9": "arm", |
|
"90": "diving board", |
|
"91": "dog", |
|
"92": "dolphin", |
|
"93": "donut", |
|
"94": "door", |
|
"95": "dragon", |
|
"96": "dresser", |
|
"97": "drill", |
|
"98": "drums", |
|
"99": "duck" |
|
}, |
|
"image_size": 64, |
|
"initializer_range": 0.02, |
|
"label2id": { |
|
"The Eiffel Tower": "309", |
|
"The Great Wall of China": "310", |
|
"The Mona Lisa": "311", |
|
"aircraft carrier": "0", |
|
"airplane": "1", |
|
"alarm clock": "2", |
|
"ambulance": "3", |
|
"angel": "4", |
|
"animal migration": "5", |
|
"ant": "6", |
|
"anvil": "7", |
|
"apple": "8", |
|
"arm": "9", |
|
"asparagus": "10", |
|
"axe": "11", |
|
"backpack": "12", |
|
"banana": "13", |
|
"bandage": "14", |
|
"barn": "15", |
|
"baseball": "17", |
|
"baseball bat": "16", |
|
"basket": "18", |
|
"basketball": "19", |
|
"bat": "20", |
|
"bathtub": "21", |
|
"beach": "22", |
|
"bear": "23", |
|
"beard": "24", |
|
"bed": "25", |
|
"bee": "26", |
|
"belt": "27", |
|
"bench": "28", |
|
"bicycle": "29", |
|
"binoculars": "30", |
|
"bird": "31", |
|
"birthday cake": "32", |
|
"blackberry": "33", |
|
"blueberry": "34", |
|
"book": "35", |
|
"boomerang": "36", |
|
"bottlecap": "37", |
|
"bowtie": "38", |
|
"bracelet": "39", |
|
"brain": "40", |
|
"bread": "41", |
|
"bridge": "42", |
|
"broccoli": "43", |
|
"broom": "44", |
|
"bucket": "45", |
|
"bulldozer": "46", |
|
"bus": "47", |
|
"bush": "48", |
|
"butterfly": "49", |
|
"cactus": "50", |
|
"cake": "51", |
|
"calculator": "52", |
|
"calendar": "53", |
|
"camel": "54", |
|
"camera": "55", |
|
"camouflage": "56", |
|
"campfire": "57", |
|
"candle": "58", |
|
"cannon": "59", |
|
"canoe": "60", |
|
"car": "61", |
|
"carrot": "62", |
|
"castle": "63", |
|
"cat": "64", |
|
"ceiling fan": "65", |
|
"cell phone": "66", |
|
"cello": "67", |
|
"chair": "68", |
|
"chandelier": "69", |
|
"church": "70", |
|
"circle": "71", |
|
"clarinet": "72", |
|
"clock": "73", |
|
"cloud": "74", |
|
"coffee cup": "75", |
|
"compass": "76", |
|
"computer": "77", |
|
"cookie": "78", |
|
"cooler": "79", |
|
"couch": "80", |
|
"cow": "81", |
|
"crab": "82", |
|
"crayon": "83", |
|
"crocodile": "84", |
|
"crown": "85", |
|
"cruise ship": "86", |
|
"cup": "87", |
|
"diamond": "88", |
|
"dishwasher": "89", |
|
"diving board": "90", |
|
"dog": "91", |
|
"dolphin": "92", |
|
"donut": "93", |
|
"door": "94", |
|
"dragon": "95", |
|
"dresser": "96", |
|
"drill": "97", |
|
"drums": "98", |
|
"duck": "99", |
|
"dumbbell": "100", |
|
"ear": "101", |
|
"elbow": "102", |
|
"elephant": "103", |
|
"envelope": "104", |
|
"eraser": "105", |
|
"eye": "106", |
|
"eyeglasses": "107", |
|
"face": "108", |
|
"fan": "109", |
|
"feather": "110", |
|
"fence": "111", |
|
"finger": "112", |
|
"fire hydrant": "113", |
|
"fireplace": "114", |
|
"firetruck": "115", |
|
"fish": "116", |
|
"flamingo": "117", |
|
"flashlight": "118", |
|
"flip flops": "119", |
|
"floor lamp": "120", |
|
"flower": "121", |
|
"flying saucer": "122", |
|
"foot": "123", |
|
"fork": "124", |
|
"frog": "125", |
|
"frying pan": "126", |
|
"garden": "128", |
|
"garden hose": "127", |
|
"giraffe": "129", |
|
"goatee": "130", |
|
"golf club": "131", |
|
"grapes": "132", |
|
"grass": "133", |
|
"guitar": "134", |
|
"hamburger": "135", |
|
"hammer": "136", |
|
"hand": "137", |
|
"harp": "138", |
|
"hat": "139", |
|
"headphones": "140", |
|
"hedgehog": "141", |
|
"helicopter": "142", |
|
"helmet": "143", |
|
"hexagon": "144", |
|
"hockey puck": "145", |
|
"hockey stick": "146", |
|
"horse": "147", |
|
"hospital": "148", |
|
"hot air balloon": "149", |
|
"hot dog": "150", |
|
"hot tub": "151", |
|
"hourglass": "152", |
|
"house": "154", |
|
"house plant": "153", |
|
"hurricane": "155", |
|
"ice cream": "156", |
|
"jacket": "157", |
|
"jail": "158", |
|
"kangaroo": "159", |
|
"key": "160", |
|
"keyboard": "161", |
|
"knee": "162", |
|
"knife": "163", |
|
"ladder": "164", |
|
"lantern": "165", |
|
"laptop": "166", |
|
"leaf": "167", |
|
"leg": "168", |
|
"light bulb": "169", |
|
"lighter": "170", |
|
"lighthouse": "171", |
|
"lightning": "172", |
|
"line": "173", |
|
"lion": "174", |
|
"lipstick": "175", |
|
"lobster": "176", |
|
"lollipop": "177", |
|
"mailbox": "178", |
|
"map": "179", |
|
"marker": "180", |
|
"matches": "181", |
|
"megaphone": "182", |
|
"mermaid": "183", |
|
"microphone": "184", |
|
"microwave": "185", |
|
"monkey": "186", |
|
"moon": "187", |
|
"mosquito": "188", |
|
"motorbike": "189", |
|
"mountain": "190", |
|
"mouse": "191", |
|
"moustache": "192", |
|
"mouth": "193", |
|
"mug": "194", |
|
"mushroom": "195", |
|
"nail": "196", |
|
"necklace": "197", |
|
"nose": "198", |
|
"ocean": "199", |
|
"octagon": "200", |
|
"octopus": "201", |
|
"onion": "202", |
|
"oven": "203", |
|
"owl": "204", |
|
"paint can": "205", |
|
"paintbrush": "206", |
|
"palm tree": "207", |
|
"panda": "208", |
|
"pants": "209", |
|
"paper clip": "210", |
|
"parachute": "211", |
|
"parrot": "212", |
|
"passport": "213", |
|
"peanut": "214", |
|
"pear": "215", |
|
"peas": "216", |
|
"pencil": "217", |
|
"penguin": "218", |
|
"piano": "219", |
|
"pickup truck": "220", |
|
"picture frame": "221", |
|
"pig": "222", |
|
"pillow": "223", |
|
"pineapple": "224", |
|
"pizza": "225", |
|
"pliers": "226", |
|
"police car": "227", |
|
"pond": "228", |
|
"pool": "229", |
|
"popsicle": "230", |
|
"postcard": "231", |
|
"potato": "232", |
|
"power outlet": "233", |
|
"purse": "234", |
|
"rabbit": "235", |
|
"raccoon": "236", |
|
"radio": "237", |
|
"rain": "238", |
|
"rainbow": "239", |
|
"rake": "240", |
|
"remote control": "241", |
|
"rhinoceros": "242", |
|
"rifle": "243", |
|
"river": "244", |
|
"roller coaster": "245", |
|
"rollerskates": "246", |
|
"sailboat": "247", |
|
"sandwich": "248", |
|
"saw": "249", |
|
"saxophone": "250", |
|
"school bus": "251", |
|
"scissors": "252", |
|
"scorpion": "253", |
|
"screwdriver": "254", |
|
"sea turtle": "255", |
|
"see saw": "256", |
|
"shark": "257", |
|
"sheep": "258", |
|
"shoe": "259", |
|
"shorts": "260", |
|
"shovel": "261", |
|
"sink": "262", |
|
"skateboard": "263", |
|
"skull": "264", |
|
"skyscraper": "265", |
|
"sleeping bag": "266", |
|
"smiley face": "267", |
|
"snail": "268", |
|
"snake": "269", |
|
"snorkel": "270", |
|
"snowflake": "271", |
|
"snowman": "272", |
|
"soccer ball": "273", |
|
"sock": "274", |
|
"speedboat": "275", |
|
"spider": "276", |
|
"spoon": "277", |
|
"spreadsheet": "278", |
|
"square": "279", |
|
"squiggle": "280", |
|
"squirrel": "281", |
|
"stairs": "282", |
|
"star": "283", |
|
"steak": "284", |
|
"stereo": "285", |
|
"stethoscope": "286", |
|
"stitches": "287", |
|
"stop sign": "288", |
|
"stove": "289", |
|
"strawberry": "290", |
|
"streetlight": "291", |
|
"string bean": "292", |
|
"submarine": "293", |
|
"suitcase": "294", |
|
"sun": "295", |
|
"swan": "296", |
|
"sweater": "297", |
|
"swing set": "298", |
|
"sword": "299", |
|
"syringe": "300", |
|
"t-shirt": "301", |
|
"table": "302", |
|
"teapot": "303", |
|
"teddy-bear": "304", |
|
"telephone": "305", |
|
"television": "306", |
|
"tennis racquet": "307", |
|
"tent": "308", |
|
"tiger": "312", |
|
"toaster": "313", |
|
"toe": "314", |
|
"toilet": "315", |
|
"tooth": "316", |
|
"toothbrush": "317", |
|
"toothpaste": "318", |
|
"tornado": "319", |
|
"tractor": "320", |
|
"traffic light": "321", |
|
"train": "322", |
|
"tree": "323", |
|
"triangle": "324", |
|
"trombone": "325", |
|
"truck": "326", |
|
"trumpet": "327", |
|
"umbrella": "328", |
|
"underwear": "329", |
|
"van": "330", |
|
"vase": "331", |
|
"violin": "332", |
|
"washing machine": "333", |
|
"watermelon": "334", |
|
"waterslide": "335", |
|
"whale": "336", |
|
"wheel": "337", |
|
"windmill": "338", |
|
"wine bottle": "339", |
|
"wine glass": "340", |
|
"wristwatch": "341", |
|
"yoga": "342", |
|
"zebra": "343", |
|
"zigzag": "344" |
|
}, |
|
"layer_norm_eps": 1e-05, |
|
"mlp_ratio": 2.0, |
|
"model_type": "mobilevitv2", |
|
"n_attn_blocks": [ |
|
2, |
|
4, |
|
3 |
|
], |
|
"num_channels": 1, |
|
"output_stride": 32, |
|
"patch_size": 2, |
|
"problem_type": "single_label_classification", |
|
"semantic_loss_ignore_index": 255, |
|
"torch_dtype": "float32", |
|
"transformers_version": "4.40.0", |
|
"width_multiplier": 1.0 |
|
} |
|
|