config.json
6.8 KB · 374 lines · json Raw
1 {
2 "_name_or_path": "nvidia/segformer-b0-finetuned-ade-512-512",
3 "architectures": [
4 "SegformerForSemanticSegmentation"
5 ],
6 "attention_probs_dropout_prob": 0.0,
7 "classifier_dropout_prob": 0.1,
8 "decoder_hidden_size": 256,
9 "depths": [
10 2,
11 2,
12 2,
13 2
14 ],
15 "downsampling_rates": [
16 1,
17 4,
18 8,
19 16
20 ],
21 "drop_path_rate": 0.1,
22 "hidden_act": "gelu",
23 "hidden_dropout_prob": 0.0,
24 "hidden_sizes": [
25 32,
26 64,
27 160,
28 256
29 ],
30 "id2label": {
31 "0": "wall",
32 "1": "building",
33 "2": "sky",
34 "3": "floor",
35 "4": "tree",
36 "5": "ceiling",
37 "6": "road",
38 "7": "bed ",
39 "8": "windowpane",
40 "9": "grass",
41 "10": "cabinet",
42 "11": "sidewalk",
43 "12": "person",
44 "13": "earth",
45 "14": "door",
46 "15": "table",
47 "16": "mountain",
48 "17": "plant",
49 "18": "curtain",
50 "19": "chair",
51 "20": "car",
52 "21": "water",
53 "22": "painting",
54 "23": "sofa",
55 "24": "shelf",
56 "25": "house",
57 "26": "sea",
58 "27": "mirror",
59 "28": "rug",
60 "29": "field",
61 "30": "armchair",
62 "31": "seat",
63 "32": "fence",
64 "33": "desk",
65 "34": "rock",
66 "35": "wardrobe",
67 "36": "lamp",
68 "37": "bathtub",
69 "38": "railing",
70 "39": "cushion",
71 "40": "base",
72 "41": "box",
73 "42": "column",
74 "43": "signboard",
75 "44": "chest of drawers",
76 "45": "counter",
77 "46": "sand",
78 "47": "sink",
79 "48": "skyscraper",
80 "49": "fireplace",
81 "50": "refrigerator",
82 "51": "grandstand",
83 "52": "path",
84 "53": "stairs",
85 "54": "runway",
86 "55": "case",
87 "56": "pool table",
88 "57": "pillow",
89 "58": "screen door",
90 "59": "stairway",
91 "60": "river",
92 "61": "bridge",
93 "62": "bookcase",
94 "63": "blind",
95 "64": "coffee table",
96 "65": "toilet",
97 "66": "flower",
98 "67": "book",
99 "68": "hill",
100 "69": "bench",
101 "70": "countertop",
102 "71": "stove",
103 "72": "palm",
104 "73": "kitchen island",
105 "74": "computer",
106 "75": "swivel chair",
107 "76": "boat",
108 "77": "bar",
109 "78": "arcade machine",
110 "79": "hovel",
111 "80": "bus",
112 "81": "towel",
113 "82": "light",
114 "83": "truck",
115 "84": "tower",
116 "85": "chandelier",
117 "86": "awning",
118 "87": "streetlight",
119 "88": "booth",
120 "89": "television receiver",
121 "90": "airplane",
122 "91": "dirt track",
123 "92": "apparel",
124 "93": "pole",
125 "94": "land",
126 "95": "bannister",
127 "96": "escalator",
128 "97": "ottoman",
129 "98": "bottle",
130 "99": "buffet",
131 "100": "poster",
132 "101": "stage",
133 "102": "van",
134 "103": "ship",
135 "104": "fountain",
136 "105": "conveyer belt",
137 "106": "canopy",
138 "107": "washer",
139 "108": "plaything",
140 "109": "swimming pool",
141 "110": "stool",
142 "111": "barrel",
143 "112": "basket",
144 "113": "waterfall",
145 "114": "tent",
146 "115": "bag",
147 "116": "minibike",
148 "117": "cradle",
149 "118": "oven",
150 "119": "ball",
151 "120": "food",
152 "121": "step",
153 "122": "tank",
154 "123": "trade name",
155 "124": "microwave",
156 "125": "pot",
157 "126": "animal",
158 "127": "bicycle",
159 "128": "lake",
160 "129": "dishwasher",
161 "130": "screen",
162 "131": "blanket",
163 "132": "sculpture",
164 "133": "hood",
165 "134": "sconce",
166 "135": "vase",
167 "136": "traffic light",
168 "137": "tray",
169 "138": "ashcan",
170 "139": "fan",
171 "140": "pier",
172 "141": "crt screen",
173 "142": "plate",
174 "143": "monitor",
175 "144": "bulletin board",
176 "145": "shower",
177 "146": "radiator",
178 "147": "glass",
179 "148": "clock",
180 "149": "flag"
181 },
182 "image_size": 224,
183 "initializer_range": 0.02,
184 "label2id": {
185 "airplane": 90,
186 "animal": 126,
187 "apparel": 92,
188 "arcade machine": 78,
189 "armchair": 30,
190 "ashcan": 138,
191 "awning": 86,
192 "bag": 115,
193 "ball": 119,
194 "bannister": 95,
195 "bar": 77,
196 "barrel": 111,
197 "base": 40,
198 "basket": 112,
199 "bathtub": 37,
200 "bed ": 7,
201 "bench": 69,
202 "bicycle": 127,
203 "blanket": 131,
204 "blind": 63,
205 "boat": 76,
206 "book": 67,
207 "bookcase": 62,
208 "booth": 88,
209 "bottle": 98,
210 "box": 41,
211 "bridge": 61,
212 "buffet": 99,
213 "building": 1,
214 "bulletin board": 144,
215 "bus": 80,
216 "cabinet": 10,
217 "canopy": 106,
218 "car": 20,
219 "case": 55,
220 "ceiling": 5,
221 "chair": 19,
222 "chandelier": 85,
223 "chest of drawers": 44,
224 "clock": 148,
225 "coffee table": 64,
226 "column": 42,
227 "computer": 74,
228 "conveyer belt": 105,
229 "counter": 45,
230 "countertop": 70,
231 "cradle": 117,
232 "crt screen": 141,
233 "curtain": 18,
234 "cushion": 39,
235 "desk": 33,
236 "dirt track": 91,
237 "dishwasher": 129,
238 "door": 14,
239 "earth": 13,
240 "escalator": 96,
241 "fan": 139,
242 "fence": 32,
243 "field": 29,
244 "fireplace": 49,
245 "flag": 149,
246 "floor": 3,
247 "flower": 66,
248 "food": 120,
249 "fountain": 104,
250 "glass": 147,
251 "grandstand": 51,
252 "grass": 9,
253 "hill": 68,
254 "hood": 133,
255 "house": 25,
256 "hovel": 79,
257 "kitchen island": 73,
258 "lake": 128,
259 "lamp": 36,
260 "land": 94,
261 "light": 82,
262 "microwave": 124,
263 "minibike": 116,
264 "mirror": 27,
265 "monitor": 143,
266 "mountain": 16,
267 "ottoman": 97,
268 "oven": 118,
269 "painting": 22,
270 "palm": 72,
271 "path": 52,
272 "person": 12,
273 "pier": 140,
274 "pillow": 57,
275 "plant": 17,
276 "plate": 142,
277 "plaything": 108,
278 "pole": 93,
279 "pool table": 56,
280 "poster": 100,
281 "pot": 125,
282 "radiator": 146,
283 "railing": 38,
284 "refrigerator": 50,
285 "river": 60,
286 "road": 6,
287 "rock": 34,
288 "rug": 28,
289 "runway": 54,
290 "sand": 46,
291 "sconce": 134,
292 "screen": 130,
293 "screen door": 58,
294 "sculpture": 132,
295 "sea": 26,
296 "seat": 31,
297 "shelf": 24,
298 "ship": 103,
299 "shower": 145,
300 "sidewalk": 11,
301 "signboard": 43,
302 "sink": 47,
303 "sky": 2,
304 "skyscraper": 48,
305 "sofa": 23,
306 "stage": 101,
307 "stairs": 53,
308 "stairway": 59,
309 "step": 121,
310 "stool": 110,
311 "stove": 71,
312 "streetlight": 87,
313 "swimming pool": 109,
314 "swivel chair": 75,
315 "table": 15,
316 "tank": 122,
317 "television receiver": 89,
318 "tent": 114,
319 "toilet": 65,
320 "towel": 81,
321 "tower": 84,
322 "trade name": 123,
323 "traffic light": 136,
324 "tray": 137,
325 "tree": 4,
326 "truck": 83,
327 "van": 102,
328 "vase": 135,
329 "wall": 0,
330 "wardrobe": 35,
331 "washer": 107,
332 "water": 21,
333 "waterfall": 113,
334 "windowpane": 8
335 },
336 "layer_norm_eps": 1e-06,
337 "mlp_ratios": [
338 4,
339 4,
340 4,
341 4
342 ],
343 "model_type": "segformer",
344 "num_attention_heads": [
345 1,
346 2,
347 5,
348 8
349 ],
350 "num_channels": 3,
351 "num_encoder_blocks": 4,
352 "patch_sizes": [
353 7,
354 3,
355 3,
356 3
357 ],
358 "reshape_last_stage": true,
359 "semantic_loss_ignore_index": 255,
360 "sr_ratios": [
361 8,
362 4,
363 2,
364 1
365 ],
366 "strides": [
367 4,
368 2,
369 2,
370 2
371 ],
372 "transformers_version": "4.37.0.dev0"
373 }
374