Image Segmentation
Transformers
PyTorch
upernet
mccaly commited on
Commit
409c30a
·
1 Parent(s): 7f00903

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +208 -300
config.json CHANGED
@@ -122,309 +122,217 @@
122
  },
123
  "hidden_size": 512,
124
  "id2label": {
125
- "0": "wall",
126
- "1": "building",
127
- "2": "sky",
128
- "3": "floor",
129
- "4": "tree",
130
- "5": "ceiling",
131
- "6": "road",
132
- "7": "bed ",
133
- "8": "windowpane",
134
- "9": "grass",
135
- "10": "cabinet",
136
- "11": "sidewalk",
137
- "12": "person",
138
- "13": "earth",
139
- "14": "door",
140
- "15": "table",
141
- "16": "mountain",
142
- "17": "plant",
143
- "18": "curtain",
144
- "19": "chair",
145
- "20": "car",
146
- "21": "water",
147
- "22": "painting",
148
- "23": "sofa",
149
- "24": "shelf",
150
- "25": "house",
151
- "26": "sea",
152
- "27": "mirror",
153
- "28": "rug",
154
- "29": "field",
155
- "30": "armchair",
156
- "31": "seat",
157
- "32": "fence",
158
- "33": "desk",
159
- "34": "rock",
160
- "35": "wardrobe",
161
- "36": "lamp",
162
- "37": "bathtub",
163
- "38": "railing",
164
- "39": "cushion",
165
- "40": "base",
166
- "41": "box",
167
- "42": "column",
168
- "43": "signboard",
169
- "44": "chest of drawers",
170
- "45": "counter",
171
- "46": "sand",
172
- "47": "sink",
173
- "48": "skyscraper",
174
- "49": "fireplace",
175
- "50": "refrigerator",
176
- "51": "grandstand",
177
- "52": "path",
178
- "53": "stairs",
179
- "54": "runway",
180
- "55": "case",
181
- "56": "pool table",
182
- "57": "pillow",
183
- "58": "screen door",
184
- "59": "stairway",
185
- "60": "river",
186
- "61": "bridge",
187
- "62": "bookcase",
188
- "63": "blind",
189
- "64": "coffee table",
190
- "65": "toilet",
191
- "66": "flower",
192
- "67": "book",
193
- "68": "hill",
194
- "69": "bench",
195
- "70": "countertop",
196
- "71": "stove",
197
- "72": "palm",
198
- "73": "kitchen island",
199
- "74": "computer",
200
- "75": "swivel chair",
201
- "76": "boat",
202
- "77": "bar",
203
- "78": "arcade machine",
204
- "79": "hovel",
205
- "80": "bus",
206
- "81": "towel",
207
- "82": "light",
208
- "83": "truck",
209
- "84": "tower",
210
- "85": "chandelier",
211
- "86": "awning",
212
- "87": "streetlight",
213
- "88": "booth",
214
- "89": "television receiver",
215
- "90": "airplane",
216
- "91": "dirt track",
217
- "92": "apparel",
218
- "93": "pole",
219
- "94": "land",
220
- "95": "bannister",
221
- "96": "escalator",
222
- "97": "ottoman",
223
- "98": "bottle",
224
- "99": "buffet",
225
- "100": "poster",
226
- "101": "stage",
227
- "102": "van",
228
- "103": "ship",
229
- "104": "fountain",
230
- "105": "conveyer belt",
231
- "106": "canopy",
232
- "107": "washer",
233
- "108": "plaything",
234
- "109": "swimming pool",
235
- "110": "stool",
236
- "111": "barrel",
237
- "112": "basket",
238
- "113": "waterfall",
239
- "114": "tent",
240
- "115": "bag",
241
- "116": "minibike",
242
- "117": "cradle",
243
- "118": "oven",
244
- "119": "ball",
245
- "120": "food",
246
- "121": "step",
247
- "122": "tank",
248
- "123": "trade name",
249
- "124": "microwave",
250
- "125": "pot",
251
- "126": "animal",
252
- "127": "bicycle",
253
- "128": "lake",
254
- "129": "dishwasher",
255
- "130": "screen",
256
- "131": "blanket",
257
- "132": "sculpture",
258
- "133": "hood",
259
- "134": "sconce",
260
- "135": "vase",
261
- "136": "traffic light",
262
- "137": "tray",
263
- "138": "ashcan",
264
- "139": "fan",
265
- "140": "pier",
266
- "141": "crt screen",
267
- "142": "plate",
268
- "143": "monitor",
269
- "144": "bulletin board",
270
- "145": "shower",
271
- "146": "radiator",
272
- "147": "glass",
273
- "148": "clock",
274
- "149": "flag"
275
  },
276
  "initializer_range": 0.02,
277
  "label2id": {
278
- "airplane": 90,
279
- "animal": 126,
280
- "apparel": 92,
281
- "arcade machine": 78,
282
- "armchair": 30,
283
- "ashcan": 138,
284
- "awning": 86,
285
- "bag": 115,
286
- "ball": 119,
287
- "bannister": 95,
288
- "bar": 77,
289
- "barrel": 111,
290
- "base": 40,
291
- "basket": 112,
292
- "bathtub": 37,
293
- "bed ": 7,
294
- "bench": 69,
295
- "bicycle": 127,
296
- "blanket": 131,
297
- "blind": 63,
298
- "boat": 76,
299
- "book": 67,
300
- "bookcase": 62,
301
- "booth": 88,
302
- "bottle": 98,
303
- "box": 41,
304
- "bridge": 61,
305
- "buffet": 99,
306
- "building": 1,
307
- "bulletin board": 144,
308
- "bus": 80,
309
- "cabinet": 10,
310
- "canopy": 106,
311
- "car": 20,
312
- "case": 55,
313
- "ceiling": 5,
314
- "chair": 19,
315
- "chandelier": 85,
316
- "chest of drawers": 44,
317
- "clock": 148,
318
- "coffee table": 64,
319
- "column": 42,
320
- "computer": 74,
321
- "conveyer belt": 105,
322
- "counter": 45,
323
- "countertop": 70,
324
- "cradle": 117,
325
- "crt screen": 141,
326
- "curtain": 18,
327
- "cushion": 39,
328
- "desk": 33,
329
- "dirt track": 91,
330
- "dishwasher": 129,
331
- "door": 14,
332
- "earth": 13,
333
- "escalator": 96,
334
- "fan": 139,
335
- "fence": 32,
336
- "field": 29,
337
- "fireplace": 49,
338
- "flag": 149,
339
- "floor": 3,
340
- "flower": 66,
341
- "food": 120,
342
- "fountain": 104,
343
- "glass": 147,
344
- "grandstand": 51,
345
- "grass": 9,
346
- "hill": 68,
347
- "hood": 133,
348
- "house": 25,
349
- "hovel": 79,
350
- "kitchen island": 73,
351
- "lake": 128,
352
- "lamp": 36,
353
- "land": 94,
354
- "light": 82,
355
- "microwave": 124,
356
- "minibike": 116,
357
- "mirror": 27,
358
- "monitor": 143,
359
- "mountain": 16,
360
- "ottoman": 97,
361
- "oven": 118,
362
- "painting": 22,
363
- "palm": 72,
364
- "path": 52,
365
- "person": 12,
366
- "pier": 140,
367
- "pillow": 57,
368
- "plant": 17,
369
- "plate": 142,
370
- "plaything": 108,
371
- "pole": 93,
372
- "pool table": 56,
373
- "poster": 100,
374
- "pot": 125,
375
- "radiator": 146,
376
- "railing": 38,
377
- "refrigerator": 50,
378
- "river": 60,
379
- "road": 6,
380
- "rock": 34,
381
- "rug": 28,
382
- "runway": 54,
383
- "sand": 46,
384
- "sconce": 134,
385
- "screen": 130,
386
- "screen door": 58,
387
- "sculpture": 132,
388
- "sea": 26,
389
- "seat": 31,
390
- "shelf": 24,
391
- "ship": 103,
392
- "shower": 145,
393
- "sidewalk": 11,
394
- "signboard": 43,
395
- "sink": 47,
396
- "sky": 2,
397
- "skyscraper": 48,
398
- "sofa": 23,
399
- "stage": 101,
400
- "stairs": 53,
401
- "stairway": 59,
402
- "step": 121,
403
- "stool": 110,
404
- "stove": 71,
405
- "streetlight": 87,
406
- "swimming pool": 109,
407
- "swivel chair": 75,
408
- "table": 15,
409
- "tank": 122,
410
- "television receiver": 89,
411
- "tent": 114,
412
- "toilet": 65,
413
- "towel": 81,
414
- "tower": 84,
415
- "trade name": 123,
416
- "traffic light": 136,
417
- "tray": 137,
418
- "tree": 4,
419
- "truck": 83,
420
- "van": 102,
421
- "vase": 135,
422
- "wall": 0,
423
- "wardrobe": 35,
424
- "washer": 107,
425
- "water": 21,
426
- "waterfall": 113,
427
- "windowpane": 8
428
  },
429
  "loss_ignore_index": 255,
430
  "model_type": "upernet",
 
122
  },
123
  "hidden_size": 512,
124
  "id2label": {
125
+ "0": "background",
126
+ "1": "candy",
127
+ "2": "egg tart",
128
+ "3": "french fries",
129
+ "4": "chocolate",
130
+ "5": "biscuit",
131
+ "6": "popcorn",
132
+ "7": "pudding",
133
+ "8": "ice cream",
134
+ "9": "cheese butter",
135
+ "10": "cake",
136
+ "11": "wine",
137
+ "12": "milkshake",
138
+ "13": "coffee",
139
+ "14": "juice",
140
+ "15": "milk",
141
+ "16": "tea",
142
+ "17": "almond",
143
+ "18": "red beans",
144
+ "19": "cashew",
145
+ "20": "dried cranberries",
146
+ "21": "soy",
147
+ "22": "walnut",
148
+ "23": "peanut",
149
+ "24": "egg",
150
+ "25": "apple",
151
+ "26": "date",
152
+ "27": "apricot",
153
+ "28": "avocado",
154
+ "29": "banana",
155
+ "30": "strawberry",
156
+ "31": "cherry",
157
+ "32": "blueberry",
158
+ "33": "raspberry",
159
+ "34": "mango",
160
+ "35": "olives",
161
+ "36": "peach",
162
+ "37": "lemon",
163
+ "38": "pear",
164
+ "39": "fig",
165
+ "40": "pineapple",
166
+ "41": "grape",
167
+ "42": "kiwi",
168
+ "43": "melon",
169
+ "44": "orange",
170
+ "45": "watermelon",
171
+ "46": "steak",
172
+ "47": "pork",
173
+ "48": "chicken duck",
174
+ "49": "sausage",
175
+ "50": "fried meat",
176
+ "51": "lamb",
177
+ "52": "sauce",
178
+ "53": "crab",
179
+ "54": "fish",
180
+ "55": "shellfish",
181
+ "56": "shrimp",
182
+ "57": "soup",
183
+ "58": "bread",
184
+ "59": "corn",
185
+ "60": "hamburg",
186
+ "61": "pizza",
187
+ "62": "hanamaki baozi",
188
+ "63": "wonton dumplings",
189
+ "64": "pasta",
190
+ "65": "noodles",
191
+ "66": "rice",
192
+ "67": "pie",
193
+ "68": "tofu",
194
+ "69": "eggplant",
195
+ "70": "potato",
196
+ "71": "garlic",
197
+ "72": "cauliflower",
198
+ "73": "tomato",
199
+ "74": "kelp",
200
+ "75": "seaweed",
201
+ "76": "spring onion",
202
+ "77": "rape",
203
+ "78": "ginger",
204
+ "79": "okra",
205
+ "80": "lettuce",
206
+ "81": "pumpkin",
207
+ "82": "cucumber",
208
+ "83": "white radish",
209
+ "84": "carrot",
210
+ "85": "asparagus",
211
+ "86": "bamboo shoots",
212
+ "87": "broccoli",
213
+ "88": "celery stick",
214
+ "89": "cilantro mint",
215
+ "90": "snow peas",
216
+ "91": "cabbage",
217
+ "92": "bean sprouts",
218
+ "93": "onion",
219
+ "94": "pepper",
220
+ "95": "green beans",
221
+ "96": "French beans",
222
+ "97": "king oyster mushroom",
223
+ "98": "shiitake",
224
+ "99": "enoki mushroom",
225
+ "100": "oyster mushroom",
226
+ "101": "white button mushroom",
227
+ "102": "salad",
228
+ "103": "other ingredients"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
229
  },
230
  "initializer_range": 0.02,
231
  "label2id": {
232
+ "almond": 17,
233
+ "apple": 25,
234
+ "apricot": 27,
235
+ "asparagus": 85,
236
+ "avocado": 28,
237
+ "background": 0,
238
+ "bamboo shoots": 86,
239
+ "banana": 29,
240
+ "bean sprouts": 92,
241
+ "biscuit": 5,
242
+ "blueberry": 32,
243
+ "bread": 58,
244
+ "broccoli": 87,
245
+ "cabbage": 91,
246
+ "cake": 10,
247
+ "candy": 1,
248
+ "carrot": 84,
249
+ "cashew": 19,
250
+ "cauliflower": 72,
251
+ "celery stick": 88,
252
+ "cheese butter": 9,
253
+ "cherry": 31,
254
+ "chicken duck": 48,
255
+ "chocolate": 4,
256
+ "cilantro mint": 89,
257
+ "coffee": 13,
258
+ "corn": 59,
259
+ "crab": 53,
260
+ "cucumber": 82,
261
+ "date": 26,
262
+ "dried cranberries": 20,
263
+ "egg": 24,
264
+ "egg tart": 2,
265
+ "eggplant": 69,
266
+ "enoki mushroom": 99,
267
+ "fig": 39,
268
+ "fish": 54,
269
+ "French beans": 96,
270
+ "french fries": 3,
271
+ "fried meat": 50,
272
+ "garlic": 71,
273
+ "ginger": 78,
274
+ "grape": 41,
275
+ "green beans": 95,
276
+ "hamburg": 60,
277
+ "hanamaki baozi": 62,
278
+ "ice cream": 8,
279
+ "juice": 14,
280
+ "kelp": 74,
281
+ "king oyster mushroom": 97,
282
+ "kiwi": 42,
283
+ "lamb": 51,
284
+ "lemon": 37,
285
+ "lettuce": 80,
286
+ "mango": 34,
287
+ "melon": 43,
288
+ "milk": 15,
289
+ "milkshake": 12,
290
+ "noodles": 65,
291
+ "okra": 79,
292
+ "olives": 35,
293
+ "onion": 93,
294
+ "orange": 44,
295
+ "other ingredients": 103,
296
+ "oyster mushroom": 100,
297
+ "pasta": 64,
298
+ "peach": 36,
299
+ "peanut": 23,
300
+ "pear": 38,
301
+ "pepper": 94,
302
+ "pie": 67,
303
+ "pineapple": 40,
304
+ "pizza": 61,
305
+ "popcorn": 6,
306
+ "pork": 47,
307
+ "potato": 70,
308
+ "pudding": 7,
309
+ "pumpkin": 81,
310
+ "rape": 77,
311
+ "raspberry": 33,
312
+ "red beans": 18,
313
+ "rice": 66,
314
+ "salad": 102,
315
+ "sauce": 52,
316
+ "sausage": 49,
317
+ "seaweed": 75,
318
+ "shellfish": 55,
319
+ "shiitake": 98,
320
+ "shrimp": 56,
321
+ "snow peas": 90,
322
+ "soup": 57,
323
+ "soy": 21,
324
+ "spring onion": 76,
325
+ "steak": 46,
326
+ "strawberry": 30,
327
+ "tea": 16,
328
+ "tofu": 68,
329
+ "tomato": 73,
330
+ "walnut": 22,
331
+ "watermelon": 45,
332
+ "white button mushroom": 101,
333
+ "white radish": 83,
334
+ "wine": 11,
335
+ "wonton dumplings": 63
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
336
  },
337
  "loss_ignore_index": 255,
338
  "model_type": "upernet",