{
  "_name_or_path": "nvidia/mit-b3",
  "architectures": [
    "SegformerForSemanticSegmentation"
  ],
  "attention_probs_dropout_prob": 0.0,
  "classifier_dropout_prob": 0.1,
  "decoder_hidden_size": 768,
  "depths": [
    3,
    4,
    18,
    3
  ],
  "downsampling_rates": [
    1,
    4,
    8,
    16
  ],
  "drop_path_rate": 0.1,
  "hidden_act": "gelu",
  "hidden_dropout_prob": 0.0,
  "hidden_sizes": [
    64,
    128,
    320,
    512
  ],
  "id2label": {
    "0": "wall",
    "1": "building;edifice",
    "2": "sky",
    "3": "floor",
    "4": "tree",
    "5": "ceiling",
    "6": "road;route",
    "7": "bed",
    "8": "windowpane;window",
    "9": "grass",
    "10": "cabinet",
    "11": "sidewalk;pavement",
    "12": "person",
    "13": "earth;ground",
    "14": "door;double;door",
    "15": "table",
    "16": "mountain;mount",
    "17": "plant;flora;plant;life",
    "18": "curtain;drape;drapery;mantle;pall",
    "19": "chair",
    "20": "car;auto",
    "21": "water",
    "22": "painting;picture",
    "23": "sofa;couch;lounge",
    "24": "rack",
    "25": "house",
    "26": "sea",
    "27": "mirror",
    "28": "rug;carpet;carpeting",
    "29": "field",
    "30": "armchair",
    "31": "seat",
    "32": "fence;fencing",
    "33": "desk",
    "34": "rock;stone",
    "35": "wardrobe;closet;press",
    "36": "lamp",
    "37": "bathtub;bathing;tub;bath;tub",
    "38": "railing;rail",
    "39": "cushion",
    "40": "base;pedestal;stand",
    "41": "box",
    "42": "pillar",
    "43": "sign",
    "44": "chest;of;drawers;chest;bureau;dresser",
    "45": "counter",
    "46": "sand",
    "47": "sink",
    "48": "skyscraper",
    "49": "fireplace;hearth;open;fireplace",
    "50": "refrigerator;icebox",
    "51": "grandstand;covered;stand",
    "52": "path",
    "53": "stairs;steps",
    "54": "runway",
    "55": "case;display;case;showcase;vitrine",
    "56": "pool;table;billiard;table;snooker;table",
    "57": "pillow",
    "58": "screen;door;screen",
    "59": "stairway;staircase",
    "60": "river",
    "61": "bridge;span",
    "62": "bookcase",
    "63": "blind;screen",
    "64": "coffee;table;cocktail;table",
    "65": "toilet;can;commode;crapper;pot;potty;stool;throne",
    "66": "flower",
    "67": "book",
    "68": "hill",
    "69": "bench",
    "70": "countertop",
    "71": "stove;kitchen;stove;range;kitchen;range;cooking;stove",
    "72": "palm;palm;tree",
    "73": "kitchen;island",
    "74": "computer",
    "75": "swivel;chair",
    "76": "boat",
    "77": "bar",
    "78": "arcade;machine",
    "79": "hovel;hut;hutch;shack;shanty",
    "80": "bus;autobus",
    "81": "towel",
    "82": "light;light;source",
    "83": "truck;motortruck",
    "84": "tower",
    "85": "chandelier;pendant;pendent",
    "86": "awning;sunshade;sunblind",
    "87": "streetlight;street;lamp",
    "88": "booth;cubicle;stall;kiosk",
    "89": "television",
    "90": "airplane;aeroplane;plane",
    "91": "dirt;track",
    "92": "apparel;wearing;apparel;dress;clothes",
    "93": "pole",
    "94": "land;ground;soil",
    "95": "bannister;banister;balustrade;balusters;handrail",
    "96": "escalator;moving;staircase;moving;stairway",
    "97": "ottoman;pouf;pouffe;puff;hassock",
    "98": "bottle",
    "99": "buffet;counter;sideboard",
    "100": "poster;posting;placard;notice;bill;card",
    "101": "stage",
    "102": "van",
    "103": "ship",
    "104": "fountain",
    "105": "conveyer;belt;conveyor;belt;conveyer;conveyor;transporter",
    "106": "canopy",
    "107": "washer;automatic;washer;washing;machine",
    "108": "plaything;toy",
    "109": "swimming;pool;swimming;bath;natatorium",
    "110": "stool",
    "111": "barel",
    "112": "basket;handbasket",
    "113": "waterfall;falls",
    "114": "tent;collapsible;shelter",
    "115": "bag",
    "116": "minibike;motorbike",
    "117": "cradle",
    "118": "oven",
    "119": "ball",
    "120": "food;solid;food",
    "121": "step;stair",
    "122": "tank;storage;tank",
    "123": "trade;name;brand;name;brand;marque",
    "124": "microwave;microwave;oven",
    "125": "pot;flowerpot",
    "126": "animal;animate;being;beast;brute;creature;fauna",
    "127": "bicycle;bike;wheel;cycle",
    "128": "lake",
    "129": "dishwasher;dish;washer;dishwashing;machine",
    "130": "screen;silver;screen;projection;screen",
    "131": "blanket;cover",
    "132": "sculpture",
    "133": "hood;exhaust;hood",
    "134": "sconce",
    "135": "vase",
    "136": "traffic;light;traffic;signal;stoplight",
    "137": "tray",
    "138": "ashcan;trash;can;garbage;wastebin",
    "139": "fan",
    "140": "pier;wharf;wharfage;dock",
    "141": "crt;screen",
    "142": "plate",
    "143": "monitor;monitoring;device",
    "144": "bulletin;board;notice;board",
    "145": "emergency_board",
    "146": "radiator",
    "147": "glass;drinking;glass",
    "148": "clock",
    "149": "flag",
    "150": "BACKGROUND",
    "151": "paper_note",
    "152": "pallet",
    "153": "UNLABELLED",
    "154": "floor_decal",
    "155": "fire_extinguisher",
    "156": "crate",
    "157": "wire",
    "158": "bracket",
    "159": "fuse_box",
    "160": "forklift",
    "161": "cone",
    "162": "cart",
    "163": "bucket",
    "164": "paper_shortcut",
    "165": "barcode"
  },
  "image_size": 224,
  "initializer_range": 0.02,
  "label2id": {
    "BACKGROUND": 150,
    "UNLABELLED": 153,
    "airplane;aeroplane;plane": 90,
    "animal;animate;being;beast;brute;creature;fauna": 126,
    "apparel;wearing;apparel;dress;clothes": 92,
    "arcade;machine": 78,
    "armchair": 30,
    "ashcan;trash;can;garbage;wastebin": 138,
    "awning;sunshade;sunblind": 86,
    "bag": 115,
    "ball": 119,
    "bannister;banister;balustrade;balusters;handrail": 95,
    "bar": 77,
    "barcode": 165,
    "barel": 111,
    "base;pedestal;stand": 40,
    "basket;handbasket": 112,
    "bathtub;bathing;tub;bath;tub": 37,
    "bed": 7,
    "bench": 69,
    "bicycle;bike;wheel;cycle": 127,
    "blanket;cover": 131,
    "blind;screen": 63,
    "boat": 76,
    "book": 67,
    "bookcase": 62,
    "booth;cubicle;stall;kiosk": 88,
    "bottle": 98,
    "box": 41,
    "bracket": 158,
    "bridge;span": 61,
    "bucket": 163,
    "buffet;counter;sideboard": 99,
    "building;edifice": 1,
    "bulletin;board;notice;board": 144,
    "bus;autobus": 80,
    "cabinet": 10,
    "canopy": 106,
    "car;auto": 20,
    "cart": 162,
    "case;display;case;showcase;vitrine": 55,
    "ceiling": 5,
    "chair": 19,
    "chandelier;pendant;pendent": 85,
    "chest;of;drawers;chest;bureau;dresser": 44,
    "clock": 148,
    "coffee;table;cocktail;table": 64,
    "computer": 74,
    "cone": 161,
    "conveyer;belt;conveyor;belt;conveyer;conveyor;transporter": 105,
    "counter": 45,
    "countertop": 70,
    "cradle": 117,
    "crate": 156,
    "crt;screen": 141,
    "curtain;drape;drapery;mantle;pall": 18,
    "cushion": 39,
    "desk": 33,
    "dirt;track": 91,
    "dishwasher;dish;washer;dishwashing;machine": 129,
    "door;double;door": 14,
    "earth;ground": 13,
    "emergency_board": 145,
    "escalator;moving;staircase;moving;stairway": 96,
    "fan": 139,
    "fence;fencing": 32,
    "field": 29,
    "fire_extinguisher": 155,
    "fireplace;hearth;open;fireplace": 49,
    "flag": 149,
    "floor": 3,
    "floor_decal": 154,
    "flower": 66,
    "food;solid;food": 120,
    "forklift": 160,
    "fountain": 104,
    "fuse_box": 159,
    "glass;drinking;glass": 147,
    "grandstand;covered;stand": 51,
    "grass": 9,
    "hill": 68,
    "hood;exhaust;hood": 133,
    "house": 25,
    "hovel;hut;hutch;shack;shanty": 79,
    "kitchen;island": 73,
    "lake": 128,
    "lamp": 36,
    "land;ground;soil": 94,
    "light;light;source": 82,
    "microwave;microwave;oven": 124,
    "minibike;motorbike": 116,
    "mirror": 27,
    "monitor;monitoring;device": 143,
    "mountain;mount": 16,
    "ottoman;pouf;pouffe;puff;hassock": 97,
    "oven": 118,
    "painting;picture": 22,
    "pallet": 152,
    "palm;palm;tree": 72,
    "paper_note": 151,
    "paper_shortcut": 164,
    "path": 52,
    "person": 12,
    "pier;wharf;wharfage;dock": 140,
    "pillar": 42,
    "pillow": 57,
    "plant;flora;plant;life": 17,
    "plate": 142,
    "plaything;toy": 108,
    "pole": 93,
    "pool;table;billiard;table;snooker;table": 56,
    "poster;posting;placard;notice;bill;card": 100,
    "pot;flowerpot": 125,
    "rack": 24,
    "radiator": 146,
    "railing;rail": 38,
    "refrigerator;icebox": 50,
    "river": 60,
    "road;route": 6,
    "rock;stone": 34,
    "rug;carpet;carpeting": 28,
    "runway": 54,
    "sand": 46,
    "sconce": 134,
    "screen;door;screen": 58,
    "screen;silver;screen;projection;screen": 130,
    "sculpture": 132,
    "sea": 26,
    "seat": 31,
    "ship": 103,
    "sidewalk;pavement": 11,
    "sign": 43,
    "sink": 47,
    "sky": 2,
    "skyscraper": 48,
    "sofa;couch;lounge": 23,
    "stage": 101,
    "stairs;steps": 53,
    "stairway;staircase": 59,
    "step;stair": 121,
    "stool": 110,
    "stove;kitchen;stove;range;kitchen;range;cooking;stove": 71,
    "streetlight;street;lamp": 87,
    "swimming;pool;swimming;bath;natatorium": 109,
    "swivel;chair": 75,
    "table": 15,
    "tank;storage;tank": 122,
    "television": 89,
    "tent;collapsible;shelter": 114,
    "toilet;can;commode;crapper;pot;potty;stool;throne": 65,
    "towel": 81,
    "tower": 84,
    "trade;name;brand;name;brand;marque": 123,
    "traffic;light;traffic;signal;stoplight": 136,
    "tray": 137,
    "tree": 4,
    "truck;motortruck": 83,
    "van": 102,
    "vase": 135,
    "wall": 0,
    "wardrobe;closet;press": 35,
    "washer;automatic;washer;washing;machine": 107,
    "water": 21,
    "waterfall;falls": 113,
    "windowpane;window": 8,
    "wire": 157
  },
  "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.40.2"
}