{ "_name_or_path": "nvidia/segformer-b0-finetuned-ade-512-512", "architectures": [ "SegformerForImageClassification" ], "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": "00002", "1": "00003", "2": "00005", "3": "00019", "4": "00028", "5": "00029", "6": "00070", "7": "00071", "8": "00081", "9": "00086", "10": "00093", "11": "00095", "12": "00096", "13": "00107", "14": "00108", "15": "00140", "16": "00146", "17": "00154", "18": "00155", "19": "00157", "20": "00162", "21": "00182", "22": "00184", "23": "00185", "24": "00188", "25": "00193", "26": "00210", "27": "00247", "28": "00254", "29": "00256", "30": "00268", "31": "00283", "32": "00383", "33": "00465", "34": "00468", "35": "00469", "36": "00470", "37": "00472", "38": "00473", "39": "00474", "40": "00491", "41": "00498", "42": "00500", "43": "00501", "44": "00505", "45": "00508", "46": "00519", "47": "00522", "48": "00531", "49": "00532", "50": "00533", "51": "00537", "52": "00543", "53": "00547", "54": "00556", "55": "00562", "56": "00563", "57": "00564", "58": "00565", "59": "00566", "60": "00567", "61": "00568", "62": "00569", "63": "00570", "64": "00571", "65": "00572", "66": "00573", "67": "00574", "68": "00575", "69": "00576", "70": "00577", "71": "00578", "72": "00579", "73": "00580", "74": "00581", "75": "00582", "76": "00583", "77": "00584", "78": "00585", "79": "00586", "80": "00587", "81": "00588", "82": "00589", "83": "00590", "84": "00591", "85": "00592", "86": "00593", "87": "00594", "88": "00595", "89": "00596", "90": "00597", "91": "00598", "92": "00599", "93": "00600", "94": "00601", "95": "00602", "96": "00603", "97": "00604", "98": "00605", "99": "00606", "100": "00607", "101": "00608", "102": "00609", "103": "00610", "104": "00611", "105": "00612", "106": "00613", "107": "00614", "108": "00615", "109": "00616", "110": "00617", "111": "00618", "112": "00619", "113": "00620", "114": "00621", "115": "00622", "116": "00623", "117": "00624", "118": "00625", "119": "00626", "120": "00627", "121": "00628", "122": "00629", "123": "00630", "124": "00631", "125": "00632", "126": "00633", "127": "00634", "128": "00635", "129": "00636", "130": "00637", "131": "00638", "132": "00639", "133": "00640", "134": "00641", "135": "00642", "136": "00643", "137": "00644", "138": "00645", "139": "00646", "140": "00647", "141": "00648", "142": "00650", "143": "00652", "144": "00653", "145": "00654", "146": "00655", "147": "00656", "148": "00657", "149": "00658", "150": "00659", "151": "00660", "152": "00661", "153": "00662", "154": "00663", "155": "00664", "156": "00665", "157": "00666", "158": "00667", "159": "00668", "160": "00669", "161": "00670", "162": "00671", "163": "00672", "164": "00673", "165": "00674", "166": "00675", "167": "00676", "168": "00677", "169": "00678", "170": "00679", "171": "00680", "172": "00681", "173": "00682", "174": "00683", "175": "00684", "176": "00685", "177": "00686", "178": "00687", "179": "00688", "180": "00689", "181": "00690", "182": "00691", "183": "00692", "184": "00693", "185": "00694", "186": "00695", "187": "00697", "188": "00698", "189": "00699", "190": "00700", "191": "00701", "192": "00703", "193": "00704", "194": "00705", "195": "00706", "196": "00707", "197": "00708", "198": "00709", "199": "00710", "200": "00711", "201": "00712", "202": "00713", "203": "00714", "204": "00715", "205": "00716", "206": "00717", "207": "00718", "208": "00719", "209": "00720", "210": "00721", "211": "00722", "212": "00723", "213": "00724", "214": "00725", "215": "00726", "216": "00727", "217": "00728", "218": "00729", "219": "00730", "220": "00731", "221": "00732", "222": "00733", "223": "00734", "224": "00735", "225": "00736", "226": "00737", "227": "00738", "228": "00739", "229": "00740", "230": "00741", "231": "00742", "232": "00743", "233": "00744", "234": "00745", "235": "00746", "236": "00747", "237": "00749", "238": "00750", "239": "00751", "240": "00752", "241": "00753", "242": "00754", "243": "00755", "244": "00756", "245": "00757", "246": "00758", "247": "00760", "248": "00761", "249": "00762", "250": "00763", "251": "00764", "252": "00765", "253": "00766", "254": "00767", "255": "00768", "256": "00769", "257": "00770", "258": "00771", "259": "00772", "260": "00773", "261": "00774", "262": "00775", "263": "00776", "264": "00777", "265": "00778", "266": "00779", "267": "00780", "268": "00781", "269": "00782", "270": "00783", "271": "00784", "272": "00785", "273": "00786", "274": "00787", "275": "00788", "276": "00789", "277": "00790", "278": "00791", "279": "00792", "280": "00793", "281": "00794", "282": "00795", "283": "00796", "284": "00797", "285": "00798", "286": "00799", "287": "00800", "288": "00801", "289": "00802", "290": "00803", "291": "00804", "292": "00805", "293": "00806", "294": "00807", "295": "00809", "296": "00810", "297": "00811", "298": "00812", "299": "00813", "300": "00814", "301": "00815", "302": "00816", "303": "00817", "304": "00876", "305": "00877", "306": "00878", "307": "00879", "308": "00880", "309": "00882", "310": "00883", "311": "00884", "312": "00885", "313": "00886", "314": "00887", "315": "00888", "316": "00889", "317": "00890", "318": "00891", "319": "00892", "320": "00893", "321": "00894", "322": "00895", "323": "00896", "324": "00897", "325": "00898", "326": "00899", "327": "00900", "328": "00901", "329": "00902", "330": "00903", "331": "00904", "332": "00905", "333": "00906", "334": "00907", "335": "00908", "336": "00909", "337": "00910", "338": "00911", "339": "00912", "340": "00913", "341": "00914", "342": "00915", "343": "00916", "344": "00917", "345": "00918", "346": "00919", "347": "00920", "348": "00921", "349": "00922", "350": "00923", "351": "00924", "352": "00925", "353": "00926", "354": "00927", "355": "00928", "356": "00929", "357": "00930", "358": "00931", "359": "00932", "360": "00933", "361": "00934", "362": "00935", "363": "00936", "364": "00937", "365": "00938", "366": "00939", "367": "00940", "368": "00941", "369": "00942", "370": "00943", "371": "00944", "372": "00945", "373": "00946", "374": "00947", "375": "00948", "376": "00949", "377": "00950", "378": "00951", "379": "00952", "380": "00953", "381": "00954", "382": "00955", "383": "00956", "384": "00957", "385": "00958", "386": "00959", "387": "00960", "388": "00961", "389": "00962", "390": "00963", "391": "00964", "392": "00965", "393": "00966", "394": "00967", "395": "00968", "396": "00969", "397": "00970", "398": "00971", "399": "00972", "400": "00973", "401": "00974", "402": "00976", "403": "00977", "404": "00978", "405": "00979", "406": "00980", "407": "00981", "408": "00982", "409": "00983", "410": "00984", "411": "00985", "412": "00986", "413": "00987", "414": "00988", "415": "00989", "416": "00990", "417": "00991", "418": "00992", "419": "00993", "420": "00994", "421": "00995", "422": "00996", "423": "00997", "424": "00998", "425": "00999", "426": "01000", "427": "01001", "428": "01002", "429": "01003", "430": "01004", "431": "01005", "432": "01006", "433": "01007", "434": "01008", "435": "01009", "436": "01010", "437": "01011", "438": "01012" }, "image_size": 224, "initializer_range": 0.02, "label2id": { "00002": 0, "00003": 1, "00005": 2, "00019": 3, "00028": 4, "00029": 5, "00070": 6, "00071": 7, "00081": 8, "00086": 9, "00093": 10, "00095": 11, "00096": 12, "00107": 13, "00108": 14, "00140": 15, "00146": 16, "00154": 17, "00155": 18, "00157": 19, "00162": 20, "00182": 21, "00184": 22, "00185": 23, "00188": 24, "00193": 25, "00210": 26, "00247": 27, "00254": 28, "00256": 29, "00268": 30, "00283": 31, "00383": 32, "00465": 33, "00468": 34, "00469": 35, "00470": 36, "00472": 37, "00473": 38, "00474": 39, "00491": 40, "00498": 41, "00500": 42, "00501": 43, "00505": 44, "00508": 45, "00519": 46, "00522": 47, "00531": 48, "00532": 49, "00533": 50, "00537": 51, "00543": 52, "00547": 53, "00556": 54, "00562": 55, "00563": 56, "00564": 57, "00565": 58, "00566": 59, "00567": 60, "00568": 61, "00569": 62, "00570": 63, "00571": 64, "00572": 65, "00573": 66, "00574": 67, "00575": 68, "00576": 69, "00577": 70, "00578": 71, "00579": 72, "00580": 73, "00581": 74, "00582": 75, "00583": 76, "00584": 77, "00585": 78, "00586": 79, "00587": 80, "00588": 81, "00589": 82, "00590": 83, "00591": 84, "00592": 85, "00593": 86, "00594": 87, "00595": 88, "00596": 89, "00597": 90, "00598": 91, "00599": 92, "00600": 93, "00601": 94, "00602": 95, "00603": 96, "00604": 97, "00605": 98, "00606": 99, "00607": 100, "00608": 101, "00609": 102, "00610": 103, "00611": 104, "00612": 105, "00613": 106, "00614": 107, "00615": 108, "00616": 109, "00617": 110, "00618": 111, "00619": 112, "00620": 113, "00621": 114, "00622": 115, "00623": 116, "00624": 117, "00625": 118, "00626": 119, "00627": 120, "00628": 121, "00629": 122, "00630": 123, "00631": 124, "00632": 125, "00633": 126, "00634": 127, "00635": 128, "00636": 129, "00637": 130, "00638": 131, "00639": 132, "00640": 133, "00641": 134, "00642": 135, "00643": 136, "00644": 137, "00645": 138, "00646": 139, "00647": 140, "00648": 141, "00650": 142, "00652": 143, "00653": 144, "00654": 145, "00655": 146, "00656": 147, "00657": 148, "00658": 149, "00659": 150, "00660": 151, "00661": 152, "00662": 153, "00663": 154, "00664": 155, "00665": 156, "00666": 157, "00667": 158, "00668": 159, "00669": 160, "00670": 161, "00671": 162, "00672": 163, "00673": 164, "00674": 165, "00675": 166, "00676": 167, "00677": 168, "00678": 169, "00679": 170, "00680": 171, "00681": 172, "00682": 173, "00683": 174, "00684": 175, "00685": 176, "00686": 177, "00687": 178, "00688": 179, "00689": 180, "00690": 181, "00691": 182, "00692": 183, "00693": 184, "00694": 185, "00695": 186, "00697": 187, "00698": 188, "00699": 189, "00700": 190, "00701": 191, "00703": 192, "00704": 193, "00705": 194, "00706": 195, "00707": 196, "00708": 197, "00709": 198, "00710": 199, "00711": 200, "00712": 201, "00713": 202, "00714": 203, "00715": 204, "00716": 205, "00717": 206, "00718": 207, "00719": 208, "00720": 209, "00721": 210, "00722": 211, "00723": 212, "00724": 213, "00725": 214, "00726": 215, "00727": 216, "00728": 217, "00729": 218, "00730": 219, "00731": 220, "00732": 221, "00733": 222, "00734": 223, "00735": 224, "00736": 225, "00737": 226, "00738": 227, "00739": 228, "00740": 229, "00741": 230, "00742": 231, "00743": 232, "00744": 233, "00745": 234, "00746": 235, "00747": 236, "00749": 237, "00750": 238, "00751": 239, "00752": 240, "00753": 241, "00754": 242, "00755": 243, "00756": 244, "00757": 245, "00758": 246, "00760": 247, "00761": 248, "00762": 249, "00763": 250, "00764": 251, "00765": 252, "00766": 253, "00767": 254, "00768": 255, "00769": 256, "00770": 257, "00771": 258, "00772": 259, "00773": 260, "00774": 261, "00775": 262, "00776": 263, "00777": 264, "00778": 265, "00779": 266, "00780": 267, "00781": 268, "00782": 269, "00783": 270, "00784": 271, "00785": 272, "00786": 273, "00787": 274, "00788": 275, "00789": 276, "00790": 277, "00791": 278, "00792": 279, "00793": 280, "00794": 281, "00795": 282, "00796": 283, "00797": 284, "00798": 285, "00799": 286, "00800": 287, "00801": 288, "00802": 289, "00803": 290, "00804": 291, "00805": 292, "00806": 293, "00807": 294, "00809": 295, "00810": 296, "00811": 297, "00812": 298, "00813": 299, "00814": 300, "00815": 301, "00816": 302, "00817": 303, "00876": 304, "00877": 305, "00878": 306, "00879": 307, "00880": 308, "00882": 309, "00883": 310, "00884": 311, "00885": 312, "00886": 313, "00887": 314, "00888": 315, "00889": 316, "00890": 317, "00891": 318, "00892": 319, "00893": 320, "00894": 321, "00895": 322, "00896": 323, "00897": 324, "00898": 325, "00899": 326, "00900": 327, "00901": 328, "00902": 329, "00903": 330, "00904": 331, "00905": 332, "00906": 333, "00907": 334, "00908": 335, "00909": 336, "00910": 337, "00911": 338, "00912": 339, "00913": 340, "00914": 341, "00915": 342, "00916": 343, "00917": 344, "00918": 345, "00919": 346, "00920": 347, "00921": 348, "00922": 349, "00923": 350, "00924": 351, "00925": 352, "00926": 353, "00927": 354, "00928": 355, "00929": 356, "00930": 357, "00931": 358, "00932": 359, "00933": 360, "00934": 361, "00935": 362, "00936": 363, "00937": 364, "00938": 365, "00939": 366, "00940": 367, "00941": 368, "00942": 369, "00943": 370, "00944": 371, "00945": 372, "00946": 373, "00947": 374, "00948": 375, "00949": 376, "00950": 377, "00951": 378, "00952": 379, "00953": 380, "00954": 381, "00955": 382, "00956": 383, "00957": 384, "00958": 385, "00959": 386, "00960": 387, "00961": 388, "00962": 389, "00963": 390, "00964": 391, "00965": 392, "00966": 393, "00967": 394, "00968": 395, "00969": 396, "00970": 397, "00971": 398, "00972": 399, "00973": 400, "00974": 401, "00976": 402, "00977": 403, "00978": 404, "00979": 405, "00980": 406, "00981": 407, "00982": 408, "00983": 409, "00984": 410, "00985": 411, "00986": 412, "00987": 413, "00988": 414, "00989": 415, "00990": 416, "00991": 417, "00992": 418, "00993": 419, "00994": 420, "00995": 421, "00996": 422, "00997": 423, "00998": 424, "00999": 425, "01000": 426, "01001": 427, "01002": 428, "01003": 429, "01004": 430, "01005": 431, "01006": 432, "01007": 433, "01008": 434, "01009": 435, "01010": 436, "01011": 437, "01012": 438 }, "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 ], "problem_type": "single_label_classification", "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.39.3" }