{ "_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.33.0.dev0" }