From 409aa6238678e52586425365e0dc7096fb467800 Mon Sep 17 00:00:00 2001 From: Niels Rogge Date: Wed, 28 Jul 2021 15:11:16 +0200 Subject: [PATCH] Add new weights as decode head is modularized --- config.json | 603 ++++++++++++++++++++------------------- preprocessor_config.json | 14 +- pytorch_model.bin | 4 +- 3 files changed, 316 insertions(+), 305 deletions(-) diff --git a/config.json b/config.json index fc6095a..bdae24c 100644 --- a/config.json +++ b/config.json @@ -27,310 +27,310 @@ 256 ], "id2label": { - "0": "LABEL_0", - "1": "LABEL_1", - "2": "LABEL_2", - "3": "LABEL_3", - "4": "LABEL_4", - "5": "LABEL_5", - "6": "LABEL_6", - "7": "LABEL_7", - "8": "LABEL_8", - "9": "LABEL_9", - "10": "LABEL_10", - "11": "LABEL_11", - "12": "LABEL_12", - "13": "LABEL_13", - "14": "LABEL_14", - "15": "LABEL_15", - "16": "LABEL_16", - "17": "LABEL_17", - "18": "LABEL_18", - "19": "LABEL_19", - "20": "LABEL_20", - "21": "LABEL_21", - "22": "LABEL_22", - "23": "LABEL_23", - "24": "LABEL_24", - "25": "LABEL_25", - "26": "LABEL_26", - "27": "LABEL_27", - "28": "LABEL_28", - "29": "LABEL_29", - "30": "LABEL_30", - "31": "LABEL_31", - "32": "LABEL_32", - "33": "LABEL_33", - "34": "LABEL_34", - "35": "LABEL_35", - "36": "LABEL_36", - "37": "LABEL_37", - "38": "LABEL_38", - "39": "LABEL_39", - "40": "LABEL_40", - "41": "LABEL_41", - "42": "LABEL_42", - "43": "LABEL_43", - "44": "LABEL_44", - "45": "LABEL_45", - "46": "LABEL_46", - "47": "LABEL_47", - "48": "LABEL_48", - "49": "LABEL_49", - "50": "LABEL_50", - "51": "LABEL_51", - "52": "LABEL_52", - "53": "LABEL_53", - "54": "LABEL_54", - "55": "LABEL_55", - "56": "LABEL_56", - "57": "LABEL_57", - "58": "LABEL_58", - "59": "LABEL_59", - "60": "LABEL_60", - "61": "LABEL_61", - "62": "LABEL_62", - "63": "LABEL_63", - "64": "LABEL_64", - "65": "LABEL_65", - "66": "LABEL_66", - "67": "LABEL_67", - "68": "LABEL_68", - "69": "LABEL_69", - "70": "LABEL_70", - "71": "LABEL_71", - "72": "LABEL_72", - "73": "LABEL_73", - "74": "LABEL_74", - "75": "LABEL_75", - "76": "LABEL_76", - "77": "LABEL_77", - "78": "LABEL_78", - "79": "LABEL_79", - "80": "LABEL_80", - "81": "LABEL_81", - "82": "LABEL_82", - "83": "LABEL_83", - "84": "LABEL_84", - "85": "LABEL_85", - "86": "LABEL_86", - "87": "LABEL_87", - "88": "LABEL_88", - "89": "LABEL_89", - "90": "LABEL_90", - "91": "LABEL_91", - "92": "LABEL_92", - "93": "LABEL_93", - "94": "LABEL_94", - "95": "LABEL_95", - "96": "LABEL_96", - "97": "LABEL_97", - "98": "LABEL_98", - "99": "LABEL_99", - "100": "LABEL_100", - "101": "LABEL_101", - "102": "LABEL_102", - "103": "LABEL_103", - "104": "LABEL_104", - "105": "LABEL_105", - "106": "LABEL_106", - "107": "LABEL_107", - "108": "LABEL_108", - "109": "LABEL_109", - "110": "LABEL_110", - "111": "LABEL_111", - "112": "LABEL_112", - "113": "LABEL_113", - "114": "LABEL_114", - "115": "LABEL_115", - "116": "LABEL_116", - "117": "LABEL_117", - "118": "LABEL_118", - "119": "LABEL_119", - "120": "LABEL_120", - "121": "LABEL_121", - "122": "LABEL_122", - "123": "LABEL_123", - "124": "LABEL_124", - "125": "LABEL_125", - "126": "LABEL_126", - "127": "LABEL_127", - "128": "LABEL_128", - "129": "LABEL_129", - "130": "LABEL_130", - "131": "LABEL_131", - "132": "LABEL_132", - "133": "LABEL_133", - "134": "LABEL_134", - "135": "LABEL_135", - "136": "LABEL_136", - "137": "LABEL_137", - "138": "LABEL_138", - "139": "LABEL_139", - "140": "LABEL_140", - "141": "LABEL_141", - "142": "LABEL_142", - "143": "LABEL_143", - "144": "LABEL_144", - "145": "LABEL_145", - "146": "LABEL_146", - "147": "LABEL_147", - "148": "LABEL_148", - "149": "LABEL_149" + "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": { - "LABEL_0": 0, - "LABEL_1": 1, - "LABEL_10": 10, - "LABEL_100": 100, - "LABEL_101": 101, - "LABEL_102": 102, - "LABEL_103": 103, - "LABEL_104": 104, - "LABEL_105": 105, - "LABEL_106": 106, - "LABEL_107": 107, - "LABEL_108": 108, - "LABEL_109": 109, - "LABEL_11": 11, - "LABEL_110": 110, - "LABEL_111": 111, - "LABEL_112": 112, - "LABEL_113": 113, - "LABEL_114": 114, - "LABEL_115": 115, - "LABEL_116": 116, - "LABEL_117": 117, - "LABEL_118": 118, - "LABEL_119": 119, - "LABEL_12": 12, - "LABEL_120": 120, - "LABEL_121": 121, - "LABEL_122": 122, - "LABEL_123": 123, - "LABEL_124": 124, - "LABEL_125": 125, - "LABEL_126": 126, - "LABEL_127": 127, - "LABEL_128": 128, - "LABEL_129": 129, - "LABEL_13": 13, - "LABEL_130": 130, - "LABEL_131": 131, - "LABEL_132": 132, - "LABEL_133": 133, - "LABEL_134": 134, - "LABEL_135": 135, - "LABEL_136": 136, - "LABEL_137": 137, - "LABEL_138": 138, - "LABEL_139": 139, - "LABEL_14": 14, - "LABEL_140": 140, - "LABEL_141": 141, - "LABEL_142": 142, - "LABEL_143": 143, - "LABEL_144": 144, - "LABEL_145": 145, - "LABEL_146": 146, - "LABEL_147": 147, - "LABEL_148": 148, - "LABEL_149": 149, - "LABEL_15": 15, - "LABEL_16": 16, - "LABEL_17": 17, - "LABEL_18": 18, - "LABEL_19": 19, - "LABEL_2": 2, - "LABEL_20": 20, - "LABEL_21": 21, - "LABEL_22": 22, - "LABEL_23": 23, - "LABEL_24": 24, - "LABEL_25": 25, - "LABEL_26": 26, - "LABEL_27": 27, - "LABEL_28": 28, - "LABEL_29": 29, - "LABEL_3": 3, - "LABEL_30": 30, - "LABEL_31": 31, - "LABEL_32": 32, - "LABEL_33": 33, - "LABEL_34": 34, - "LABEL_35": 35, - "LABEL_36": 36, - "LABEL_37": 37, - "LABEL_38": 38, - "LABEL_39": 39, - "LABEL_4": 4, - "LABEL_40": 40, - "LABEL_41": 41, - "LABEL_42": 42, - "LABEL_43": 43, - "LABEL_44": 44, - "LABEL_45": 45, - "LABEL_46": 46, - "LABEL_47": 47, - "LABEL_48": 48, - "LABEL_49": 49, - "LABEL_5": 5, - "LABEL_50": 50, - "LABEL_51": 51, - "LABEL_52": 52, - "LABEL_53": 53, - "LABEL_54": 54, - "LABEL_55": 55, - "LABEL_56": 56, - "LABEL_57": 57, - "LABEL_58": 58, - "LABEL_59": 59, - "LABEL_6": 6, - "LABEL_60": 60, - "LABEL_61": 61, - "LABEL_62": 62, - "LABEL_63": 63, - "LABEL_64": 64, - "LABEL_65": 65, - "LABEL_66": 66, - "LABEL_67": 67, - "LABEL_68": 68, - "LABEL_69": 69, - "LABEL_7": 7, - "LABEL_70": 70, - "LABEL_71": 71, - "LABEL_72": 72, - "LABEL_73": 73, - "LABEL_74": 74, - "LABEL_75": 75, - "LABEL_76": 76, - "LABEL_77": 77, - "LABEL_78": 78, - "LABEL_79": 79, - "LABEL_8": 8, - "LABEL_80": 80, - "LABEL_81": 81, - "LABEL_82": 82, - "LABEL_83": 83, - "LABEL_84": 84, - "LABEL_85": 85, - "LABEL_86": 86, - "LABEL_87": 87, - "LABEL_88": 88, - "LABEL_89": 89, - "LABEL_9": 9, - "LABEL_90": 90, - "LABEL_91": 91, - "LABEL_92": 92, - "LABEL_93": 93, - "LABEL_94": 94, - "LABEL_95": 95, - "LABEL_96": 96, - "LABEL_97": 97, - "LABEL_98": 98, - "LABEL_99": 99 + "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": [ @@ -366,5 +366,6 @@ 2, 2 ], - "transformers_version": "4.9.0.dev0" + "torch_dtype": "float32", + "transformers_version": "4.10.0.dev0" } diff --git a/preprocessor_config.json b/preprocessor_config.json index 8e75f7b..0ee8471 100644 --- a/preprocessor_config.json +++ b/preprocessor_config.json @@ -1,5 +1,12 @@ { + "align": false, + "crop_size": [ + 512, + 512 + ], "do_normalize": true, + "do_pad": true, + "do_random_crop": false, "do_resize": true, "feature_extractor_type": "SegFormerFeatureExtractor", "image_mean": [ @@ -8,7 +15,7 @@ 0.406 ], "image_scale": [ - 2048, + 512, 512 ], "image_std": [ @@ -16,7 +23,10 @@ 0.224, 0.225 ], - "keep_ratio": true, + "keep_ratio": false, + "padding_value": 0, + "reduce_zero_label": false, "resample": 2, + "segmentation_padding_value": -100, "size_divisor": 32 } diff --git a/pytorch_model.bin b/pytorch_model.bin index f1cdb13..b2e9983 100644 --- a/pytorch_model.bin +++ b/pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:1f9993d65731c081d1830c7c01254ee6af2b8d062b4b539bd1f18051cac13c83 -size 15096815 +oid sha256:0f4df97633cbedd558ecffa3ad228ace5af37e082678390b45a9d22745787c61 +size 15092257