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