{ "_name_or_path": "nvidia/mit-b0", "architectures": [ "SegformerForSemanticSegmentation" ], "attention_probs_dropout_prob": 0.0, "classifier_dropout_prob": 0.1, "decoder_hidden_size": 256, "depths": [ 2, 2, 2, 2 ], "downsampling_rates": [ 1, 4, 8, 16 ], "drop_path_rate": 0.1, "hidden_act": "gelu", "hidden_dropout_prob": 0.0, "hidden_sizes": [ 32, 64, 160, 256 ], "id2label": { "0": "wall", "1": "building", "2": "sky", "3": "floor", "4": "tree", "5": "ceiling", "6": "road", "7": "bed ", "8": "windowpane", "9": "grass", "10": "cabinet", "11": "sidewalk", "12": "person", "13": "earth", "14": "door", "15": "table", "16": "mountain", "17": "plant", "18": "curtain", "19": "chair", "20": "car", "21": "water", "22": "painting", "23": "sofa", "24": "shelf", "25": "house", "26": "sea", "27": "mirror", "28": "rug", "29": "field", "30": "armchair", "31": "seat", "32": "fence", "33": "desk", "34": "rock", "35": "wardrobe", "36": "lamp", "37": "bathtub", "38": "railing", "39": "cushion", "40": "base", "41": "box", "42": "column", "43": "signboard", "44": "chest of drawers", "45": "counter", "46": "sand", "47": "sink", "48": "skyscraper", "49": "fireplace", "50": "refrigerator", "51": "grandstand", "52": "path", "53": "stairs", "54": "runway", "55": "case", "56": "pool table", "57": "pillow", "58": "screen door", "59": "stairway", "60": "river", "61": "bridge", "62": "bookcase", "63": "blind", "64": "coffee table", "65": "toilet", "66": "flower", "67": "book", "68": "hill", "69": "bench", "70": "countertop", "71": "stove", "72": "palm", "73": "kitchen island", "74": "computer", "75": "swivel chair", "76": "boat", "77": "bar", "78": "arcade machine", "79": "hovel", "80": "bus", "81": "towel", "82": "light", "83": "truck", "84": "tower", "85": "chandelier", "86": "awning", "87": "streetlight", "88": "booth", "89": "television receiver", "90": "airplane", "91": "dirt track", "92": "apparel", "93": "pole", "94": "land", "95": "bannister", "96": "escalator", "97": "ottoman", "98": "bottle", "99": "buffet", "100": "poster", "101": "stage", "102": "van", "103": "ship", "104": "fountain", "105": "conveyer belt", "106": "canopy", "107": "washer", "108": "plaything", "109": "swimming pool", "110": "stool", "111": "barrel", "112": "basket", "113": "waterfall", "114": "tent", "115": "bag", "116": "minibike", "117": "cradle", "118": "oven", "119": "ball", "120": "food", "121": "step", "122": "tank", "123": "trade name", "124": "microwave", "125": "pot", "126": "animal", "127": "bicycle", "128": "lake", "129": "dishwasher", "130": "screen", "131": "blanket", "132": "sculpture", "133": "hood", "134": "sconce", "135": "vase", "136": "traffic light", "137": "tray", "138": "ashcan", "139": "fan", "140": "pier", "141": "crt screen", "142": "plate", "143": "monitor", "144": "bulletin board", "145": "shower", "146": "radiator", "147": "glass", "148": "clock", "149": "flag" }, "image_size": 224, "initializer_range": 0.02, "label2id": { "airplane": 90, "animal": 126, "apparel": 92, "arcade machine": 78, "armchair": 30, "ashcan": 138, "awning": 86, "bag": 115, "ball": 119, "bannister": 95, "bar": 77, "barrel": 111, "base": 40, "basket": 112, "bathtub": 37, "bed ": 7, "bench": 69, "bicycle": 127, "blanket": 131, "blind": 63, "boat": 76, "book": 67, "bookcase": 62, "booth": 88, "bottle": 98, "box": 41, "bridge": 61, "buffet": 99, "building": 1, "bulletin board": 144, "bus": 80, "cabinet": 10, "canopy": 106, "car": 20, "case": 55, "ceiling": 5, "chair": 19, "chandelier": 85, "chest of drawers": 44, "clock": 148, "coffee table": 64, "column": 42, "computer": 74, "conveyer belt": 105, "counter": 45, "countertop": 70, "cradle": 117, "crt screen": 141, "curtain": 18, "cushion": 39, "desk": 33, "dirt track": 91, "dishwasher": 129, "door": 14, "earth": 13, "escalator": 96, "fan": 139, "fence": 32, "field": 29, "fireplace": 49, "flag": 149, "floor": 3, "flower": 66, "food": 120, "fountain": 104, "glass": 147, "grandstand": 51, "grass": 9, "hill": 68, "hood": 133, "house": 25, "hovel": 79, "kitchen island": 73, "lake": 128, "lamp": 36, "land": 94, "light": 82, "microwave": 124, "minibike": 116, "mirror": 27, "monitor": 143, "mountain": 16, "ottoman": 97, "oven": 118, "painting": 22, "palm": 72, "path": 52, "person": 12, "pier": 140, "pillow": 57, "plant": 17, "plate": 142, "plaything": 108, "pole": 93, "pool table": 56, "poster": 100, "pot": 125, "radiator": 146, "railing": 38, "refrigerator": 50, "river": 60, "road": 6, "rock": 34, "rug": 28, "runway": 54, "sand": 46, "sconce": 134, "screen": 130, "screen door": 58, "sculpture": 132, "sea": 26, "seat": 31, "shelf": 24, "ship": 103, "shower": 145, "sidewalk": 11, "signboard": 43, "sink": 47, "sky": 2, "skyscraper": 48, "sofa": 23, "stage": 101, "stairs": 53, "stairway": 59, "step": 121, "stool": 110, "stove": 71, "streetlight": 87, "swimming pool": 109, "swivel chair": 75, "table": 15, "tank": 122, "television receiver": 89, "tent": 114, "toilet": 65, "towel": 81, "tower": 84, "trade name": 123, "traffic light": 136, "tray": 137, "tree": 4, "truck": 83, "van": 102, "vase": 135, "wall": 0, "wardrobe": 35, "washer": 107, "water": 21, "waterfall": 113, "windowpane": 8 }, "layer_norm_eps": 1e-06, "mlp_ratios": [ 4, 4, 4, 4 ], "model_type": "segformer", "num_attention_heads": [ 1, 2, 5, 8 ], "num_channels": 3, "num_encoder_blocks": 4, "patch_sizes": [ 7, 3, 3, 3 ], "reshape_last_stage": true, "semantic_loss_ignore_index": 255, "sr_ratios": [ 8, 4, 2, 1 ], "strides": [ 4, 2, 2, 2 ], "torch_dtype": "float32", "transformers_version": "4.28.0" }