doodle-dash2 / config.json
laszlokiss27's picture
Upload 8 files
5bd74a2 verified
raw
history blame
15.9 kB
{
"_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
}