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