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