nvidia
/
segformer-b0-finetuned-ade-512-512
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add new weights as decode head is modularized

This commit is contained in:
Niels Rogge 2021-07-28 15:11:16 +02:00
parent fbfdfbc59c
commit 409aa62386
3 changed files with 316 additions and 305 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

603
config.json
View File

@ -27,310 +27,310 @@
256 256
], ],
"id2label": { "id2label": {
"0": "LABEL_0", "0": "wall",
"1": "LABEL_1", "1": "building",
"2": "LABEL_2", "2": "sky",
"3": "LABEL_3", "3": "floor",
"4": "LABEL_4", "4": "tree",
"5": "LABEL_5", "5": "ceiling",
"6": "LABEL_6", "6": "road",
"7": "LABEL_7", "7": "bed ",
"8": "LABEL_8", "8": "windowpane",
"9": "LABEL_9", "9": "grass",
"10": "LABEL_10", "10": "cabinet",
"11": "LABEL_11", "11": "sidewalk",
"12": "LABEL_12", "12": "person",
"13": "LABEL_13", "13": "earth",
"14": "LABEL_14", "14": "door",
"15": "LABEL_15", "15": "table",
"16": "LABEL_16", "16": "mountain",
"17": "LABEL_17", "17": "plant",
"18": "LABEL_18", "18": "curtain",
"19": "LABEL_19", "19": "chair",
"20": "LABEL_20", "20": "car",
"21": "LABEL_21", "21": "water",
"22": "LABEL_22", "22": "painting",
"23": "LABEL_23", "23": "sofa",
"24": "LABEL_24", "24": "shelf",
"25": "LABEL_25", "25": "house",
"26": "LABEL_26", "26": "sea",
"27": "LABEL_27", "27": "mirror",
"28": "LABEL_28", "28": "rug",
"29": "LABEL_29", "29": "field",
"30": "LABEL_30", "30": "armchair",
"31": "LABEL_31", "31": "seat",
"32": "LABEL_32", "32": "fence",
"33": "LABEL_33", "33": "desk",
"34": "LABEL_34", "34": "rock",
"35": "LABEL_35", "35": "wardrobe",
"36": "LABEL_36", "36": "lamp",
"37": "LABEL_37", "37": "bathtub",
"38": "LABEL_38", "38": "railing",
"39": "LABEL_39", "39": "cushion",
"40": "LABEL_40", "40": "base",
"41": "LABEL_41", "41": "box",
"42": "LABEL_42", "42": "column",
"43": "LABEL_43", "43": "signboard",
"44": "LABEL_44", "44": "chest of drawers",
"45": "LABEL_45", "45": "counter",
"46": "LABEL_46", "46": "sand",
"47": "LABEL_47", "47": "sink",
"48": "LABEL_48", "48": "skyscraper",
"49": "LABEL_49", "49": "fireplace",
"50": "LABEL_50", "50": "refrigerator",
"51": "LABEL_51", "51": "grandstand",
"52": "LABEL_52", "52": "path",
"53": "LABEL_53", "53": "stairs",
"54": "LABEL_54", "54": "runway",
"55": "LABEL_55", "55": "case",
"56": "LABEL_56", "56": "pool table",
"57": "LABEL_57", "57": "pillow",
"58": "LABEL_58", "58": "screen door",
"59": "LABEL_59", "59": "stairway",
"60": "LABEL_60", "60": "river",
"61": "LABEL_61", "61": "bridge",
"62": "LABEL_62", "62": "bookcase",
"63": "LABEL_63", "63": "blind",
"64": "LABEL_64", "64": "coffee table",
"65": "LABEL_65", "65": "toilet",
"66": "LABEL_66", "66": "flower",
"67": "LABEL_67", "67": "book",
"68": "LABEL_68", "68": "hill",
"69": "LABEL_69", "69": "bench",
"70": "LABEL_70", "70": "countertop",
"71": "LABEL_71", "71": "stove",
"72": "LABEL_72", "72": "palm",
"73": "LABEL_73", "73": "kitchen island",
"74": "LABEL_74", "74": "computer",
"75": "LABEL_75", "75": "swivel chair",
"76": "LABEL_76", "76": "boat",
"77": "LABEL_77", "77": "bar",
"78": "LABEL_78", "78": "arcade machine",
"79": "LABEL_79", "79": "hovel",
"80": "LABEL_80", "80": "bus",
"81": "LABEL_81", "81": "towel",
"82": "LABEL_82", "82": "light",
"83": "LABEL_83", "83": "truck",
"84": "LABEL_84", "84": "tower",
"85": "LABEL_85", "85": "chandelier",
"86": "LABEL_86", "86": "awning",
"87": "LABEL_87", "87": "streetlight",
"88": "LABEL_88", "88": "booth",
"89": "LABEL_89", "89": "television receiver",
"90": "LABEL_90", "90": "airplane",
"91": "LABEL_91", "91": "dirt track",
"92": "LABEL_92", "92": "apparel",
"93": "LABEL_93", "93": "pole",
"94": "LABEL_94", "94": "land",
"95": "LABEL_95", "95": "bannister",
"96": "LABEL_96", "96": "escalator",
"97": "LABEL_97", "97": "ottoman",
"98": "LABEL_98", "98": "bottle",
"99": "LABEL_99", "99": "buffet",
"100": "LABEL_100", "100": "poster",
"101": "LABEL_101", "101": "stage",
"102": "LABEL_102", "102": "van",
"103": "LABEL_103", "103": "ship",
"104": "LABEL_104", "104": "fountain",
"105": "LABEL_105", "105": "conveyer belt",
"106": "LABEL_106", "106": "canopy",
"107": "LABEL_107", "107": "washer",
"108": "LABEL_108", "108": "plaything",
"109": "LABEL_109", "109": "swimming pool",
"110": "LABEL_110", "110": "stool",
"111": "LABEL_111", "111": "barrel",
"112": "LABEL_112", "112": "basket",
"113": "LABEL_113", "113": "waterfall",
"114": "LABEL_114", "114": "tent",
"115": "LABEL_115", "115": "bag",
"116": "LABEL_116", "116": "minibike",
"117": "LABEL_117", "117": "cradle",
"118": "LABEL_118", "118": "oven",
"119": "LABEL_119", "119": "ball",
"120": "LABEL_120", "120": "food",
"121": "LABEL_121", "121": "step",
"122": "LABEL_122", "122": "tank",
"123": "LABEL_123", "123": "trade name",
"124": "LABEL_124", "124": "microwave",
"125": "LABEL_125", "125": "pot",
"126": "LABEL_126", "126": "animal",
"127": "LABEL_127", "127": "bicycle",
"128": "LABEL_128", "128": "lake",
"129": "LABEL_129", "129": "dishwasher",
"130": "LABEL_130", "130": "screen",
"131": "LABEL_131", "131": "blanket",
"132": "LABEL_132", "132": "sculpture",
"133": "LABEL_133", "133": "hood",
"134": "LABEL_134", "134": "sconce",
"135": "LABEL_135", "135": "vase",
"136": "LABEL_136", "136": "traffic light",
"137": "LABEL_137", "137": "tray",
"138": "LABEL_138", "138": "ashcan",
"139": "LABEL_139", "139": "fan",
"140": "LABEL_140", "140": "pier",
"141": "LABEL_141", "141": "crt screen",
"142": "LABEL_142", "142": "plate",
"143": "LABEL_143", "143": "monitor",
"144": "LABEL_144", "144": "bulletin board",
"145": "LABEL_145", "145": "shower",
"146": "LABEL_146", "146": "radiator",
"147": "LABEL_147", "147": "glass",
"148": "LABEL_148", "148": "clock",
"149": "LABEL_149" "149": "flag"
}, },
"image_size": 224, "image_size": 224,
"initializer_range": 0.02, "initializer_range": 0.02,
"label2id": { "label2id": {
"LABEL_0": 0, "airplane": 90,
"LABEL_1": 1, "animal": 126,
"LABEL_10": 10, "apparel": 92,
"LABEL_100": 100, "arcade machine": 78,
"LABEL_101": 101, "armchair": 30,
"LABEL_102": 102, "ashcan": 138,
"LABEL_103": 103, "awning": 86,
"LABEL_104": 104, "bag": 115,
"LABEL_105": 105, "ball": 119,
"LABEL_106": 106, "bannister": 95,
"LABEL_107": 107, "bar": 77,
"LABEL_108": 108, "barrel": 111,
"LABEL_109": 109, "base": 40,
"LABEL_11": 11, "basket": 112,
"LABEL_110": 110, "bathtub": 37,
"LABEL_111": 111, "bed ": 7,
"LABEL_112": 112, "bench": 69,
"LABEL_113": 113, "bicycle": 127,
"LABEL_114": 114, "blanket": 131,
"LABEL_115": 115, "blind": 63,
"LABEL_116": 116, "boat": 76,
"LABEL_117": 117, "book": 67,
"LABEL_118": 118, "bookcase": 62,
"LABEL_119": 119, "booth": 88,
"LABEL_12": 12, "bottle": 98,
"LABEL_120": 120, "box": 41,
"LABEL_121": 121, "bridge": 61,
"LABEL_122": 122, "buffet": 99,
"LABEL_123": 123, "building": 1,
"LABEL_124": 124, "bulletin board": 144,
"LABEL_125": 125, "bus": 80,
"LABEL_126": 126, "cabinet": 10,
"LABEL_127": 127, "canopy": 106,
"LABEL_128": 128, "car": 20,
"LABEL_129": 129, "case": 55,
"LABEL_13": 13, "ceiling": 5,
"LABEL_130": 130, "chair": 19,
"LABEL_131": 131, "chandelier": 85,
"LABEL_132": 132, "chest of drawers": 44,
"LABEL_133": 133, "clock": 148,
"LABEL_134": 134, "coffee table": 64,
"LABEL_135": 135, "column": 42,
"LABEL_136": 136, "computer": 74,
"LABEL_137": 137, "conveyer belt": 105,
"LABEL_138": 138, "counter": 45,
"LABEL_139": 139, "countertop": 70,
"LABEL_14": 14, "cradle": 117,
"LABEL_140": 140, "crt screen": 141,
"LABEL_141": 141, "curtain": 18,
"LABEL_142": 142, "cushion": 39,
"LABEL_143": 143, "desk": 33,
"LABEL_144": 144, "dirt track": 91,
"LABEL_145": 145, "dishwasher": 129,
"LABEL_146": 146, "door": 14,
"LABEL_147": 147, "earth": 13,
"LABEL_148": 148, "escalator": 96,
"LABEL_149": 149, "fan": 139,
"LABEL_15": 15, "fence": 32,
"LABEL_16": 16, "field": 29,
"LABEL_17": 17, "fireplace": 49,
"LABEL_18": 18, "flag": 149,
"LABEL_19": 19, "floor": 3,
"LABEL_2": 2, "flower": 66,
"LABEL_20": 20, "food": 120,
"LABEL_21": 21, "fountain": 104,
"LABEL_22": 22, "glass": 147,
"LABEL_23": 23, "grandstand": 51,
"LABEL_24": 24, "grass": 9,
"LABEL_25": 25, "hill": 68,
"LABEL_26": 26, "hood": 133,
"LABEL_27": 27, "house": 25,
"LABEL_28": 28, "hovel": 79,
"LABEL_29": 29, "kitchen island": 73,
"LABEL_3": 3, "lake": 128,
"LABEL_30": 30, "lamp": 36,
"LABEL_31": 31, "land": 94,
"LABEL_32": 32, "light": 82,
"LABEL_33": 33, "microwave": 124,
"LABEL_34": 34, "minibike": 116,
"LABEL_35": 35, "mirror": 27,
"LABEL_36": 36, "monitor": 143,
"LABEL_37": 37, "mountain": 16,
"LABEL_38": 38, "ottoman": 97,
"LABEL_39": 39, "oven": 118,
"LABEL_4": 4, "painting": 22,
"LABEL_40": 40, "palm": 72,
"LABEL_41": 41, "path": 52,
"LABEL_42": 42, "person": 12,
"LABEL_43": 43, "pier": 140,
"LABEL_44": 44, "pillow": 57,
"LABEL_45": 45, "plant": 17,
"LABEL_46": 46, "plate": 142,
"LABEL_47": 47, "plaything": 108,
"LABEL_48": 48, "pole": 93,
"LABEL_49": 49, "pool table": 56,
"LABEL_5": 5, "poster": 100,
"LABEL_50": 50, "pot": 125,
"LABEL_51": 51, "radiator": 146,
"LABEL_52": 52, "railing": 38,
"LABEL_53": 53, "refrigerator": 50,
"LABEL_54": 54, "river": 60,
"LABEL_55": 55, "road": 6,
"LABEL_56": 56, "rock": 34,
"LABEL_57": 57, "rug": 28,
"LABEL_58": 58, "runway": 54,
"LABEL_59": 59, "sand": 46,
"LABEL_6": 6, "sconce": 134,
"LABEL_60": 60, "screen": 130,
"LABEL_61": 61, "screen door": 58,
"LABEL_62": 62, "sculpture": 132,
"LABEL_63": 63, "sea": 26,
"LABEL_64": 64, "seat": 31,
"LABEL_65": 65, "shelf": 24,
"LABEL_66": 66, "ship": 103,
"LABEL_67": 67, "shower": 145,
"LABEL_68": 68, "sidewalk": 11,
"LABEL_69": 69, "signboard": 43,
"LABEL_7": 7, "sink": 47,
"LABEL_70": 70, "sky": 2,
"LABEL_71": 71, "skyscraper": 48,
"LABEL_72": 72, "sofa": 23,
"LABEL_73": 73, "stage": 101,
"LABEL_74": 74, "stairs": 53,
"LABEL_75": 75, "stairway": 59,
"LABEL_76": 76, "step": 121,
"LABEL_77": 77, "stool": 110,
"LABEL_78": 78, "stove": 71,
"LABEL_79": 79, "streetlight": 87,
"LABEL_8": 8, "swimming pool": 109,
"LABEL_80": 80, "swivel chair": 75,
"LABEL_81": 81, "table": 15,
"LABEL_82": 82, "tank": 122,
"LABEL_83": 83, "television receiver": 89,
"LABEL_84": 84, "tent": 114,
"LABEL_85": 85, "toilet": 65,
"LABEL_86": 86, "towel": 81,
"LABEL_87": 87, "tower": 84,
"LABEL_88": 88, "trade name": 123,
"LABEL_89": 89, "traffic light": 136,
"LABEL_9": 9, "tray": 137,
"LABEL_90": 90, "tree": 4,
"LABEL_91": 91, "truck": 83,
"LABEL_92": 92, "van": 102,
"LABEL_93": 93, "vase": 135,
"LABEL_94": 94, "wall": 0,
"LABEL_95": 95, "wardrobe": 35,
"LABEL_96": 96, "washer": 107,
"LABEL_97": 97, "water": 21,
"LABEL_98": 98, "waterfall": 113,
"LABEL_99": 99 "windowpane": 8
}, },
"layer_norm_eps": 1e-06, "layer_norm_eps": 1e-06,
"mlp_ratios": [ "mlp_ratios": [
@ -366,5 +366,6 @@
2, 2,
2 2
], ],
"transformers_version": "4.9.0.dev0" "torch_dtype": "float32",
"transformers_version": "4.10.0.dev0"
} }

14
preprocessor_config.json
View File

@ -1,5 +1,12 @@
{ {
"align": false,
"crop_size": [
512,
512
],
"do_normalize": true, "do_normalize": true,
"do_pad": true,
"do_random_crop": false,
"do_resize": true, "do_resize": true,
"feature_extractor_type": "SegFormerFeatureExtractor", "feature_extractor_type": "SegFormerFeatureExtractor",
"image_mean": [ "image_mean": [
@ -8,7 +15,7 @@
0.406 0.406
], ],
"image_scale": [ "image_scale": [
2048, 512,
512 512
], ],
"image_std": [ "image_std": [
@ -16,7 +23,10 @@
0.224, 0.224,
0.225 0.225
], ],
"keep_ratio": true, "keep_ratio": false,
"padding_value": 0,
"reduce_zero_label": false,
"resample": 2, "resample": 2,
"segmentation_padding_value": -100,
"size_divisor": 32 "size_divisor": 32
} }

BIN
pytorch_model.bin (Stored with Git LFS)
View File

Binary file not shown.