Clement commited on
Commit
f084378
·
1 Parent(s): 3e03b42
Files changed (3) hide show
  1. config.json +515 -0
  2. preprocessor_config.json +18 -0
  3. pytorch_model.bin +3 -0
config.json ADDED
@@ -0,0 +1,515 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_commit_hash": "cb488c7620d6b5fc737c94b86b25efe09ea750d1",
3
+ "_name_or_path": "facebook/maskformer-swin-tiny-ade",
4
+ "architectures": ["MaskFormerForInstanceSegmentation"],
5
+ "backbone_config": {
6
+ "_name_or_path": "",
7
+ "add_cross_attention": false,
8
+ "architectures": null,
9
+ "attention_probs_dropout_prob": 0.0,
10
+ "bad_words_ids": null,
11
+ "begin_suppress_tokens": null,
12
+ "bos_token_id": null,
13
+ "chunk_size_feed_forward": 0,
14
+ "cross_attention_hidden_size": null,
15
+ "decoder_start_token_id": null,
16
+ "depths": [2, 2, 6, 2],
17
+ "diversity_penalty": 0.0,
18
+ "do_sample": false,
19
+ "drop_path_rate": 0.3,
20
+ "early_stopping": false,
21
+ "embed_dim": 96,
22
+ "encoder_no_repeat_ngram_size": 0,
23
+ "encoder_stride": 32,
24
+ "eos_token_id": null,
25
+ "exponential_decay_length_penalty": null,
26
+ "finetuning_task": null,
27
+ "forced_bos_token_id": null,
28
+ "forced_eos_token_id": null,
29
+ "hidden_act": "gelu",
30
+ "hidden_dropout_prob": 0.0,
31
+ "hidden_size": 768,
32
+ "id2label": {
33
+ "0": "LABEL_0",
34
+ "1": "LABEL_1"
35
+ },
36
+ "image_size": 224,
37
+ "in_channels": 3,
38
+ "initializer_range": 0.02,
39
+ "is_decoder": false,
40
+ "is_encoder_decoder": false,
41
+ "label2id": {
42
+ "LABEL_0": 0,
43
+ "LABEL_1": 1
44
+ },
45
+ "layer_norm_eps": 1e-5,
46
+ "length_penalty": 1.0,
47
+ "max_length": 20,
48
+ "min_length": 0,
49
+ "mlp_ratio": 4.0,
50
+ "model_type": "swin",
51
+ "no_repeat_ngram_size": 0,
52
+ "num_beam_groups": 1,
53
+ "num_beams": 1,
54
+ "num_channels": 3,
55
+ "num_heads": [3, 6, 12, 24],
56
+ "num_layers": 4,
57
+ "num_return_sequences": 1,
58
+ "out_features": ["stage4"],
59
+ "out_indices": [4],
60
+ "output_attentions": false,
61
+ "output_hidden_states": false,
62
+ "output_scores": false,
63
+ "pad_token_id": null,
64
+ "patch_size": 4,
65
+ "path_norm": true,
66
+ "prefix": null,
67
+ "pretrain_img_size": 224,
68
+ "problem_type": null,
69
+ "pruned_heads": {},
70
+ "qkv_bias": true,
71
+ "remove_invalid_values": false,
72
+ "repetition_penalty": 1.0,
73
+ "return_dict": true,
74
+ "return_dict_in_generate": false,
75
+ "sep_token_id": null,
76
+ "stage_names": ["stem", "stage1", "stage2", "stage3", "stage4"],
77
+ "suppress_tokens": null,
78
+ "task_specific_params": null,
79
+ "temperature": 1.0,
80
+ "tf_legacy_loss": false,
81
+ "tie_encoder_decoder": false,
82
+ "tie_word_embeddings": true,
83
+ "tokenizer_class": null,
84
+ "top_k": 50,
85
+ "top_p": 1.0,
86
+ "torch_dtype": null,
87
+ "torchscript": false,
88
+ "transformers_version": "4.31.0",
89
+ "typical_p": 1.0,
90
+ "use_absolute_embeddings": false,
91
+ "use_bfloat16": false,
92
+ "window_size": 7
93
+ },
94
+ "ce_weight": 1.0,
95
+ "cross_entropy_weight": 1.0,
96
+ "decoder_config": {
97
+ "_commit_hash": null,
98
+ "_name_or_path": "",
99
+ "activation_dropout": 0.0,
100
+ "activation_function": "relu",
101
+ "add_cross_attention": false,
102
+ "architectures": null,
103
+ "attention_dropout": 0.0,
104
+ "auxiliary_loss": false,
105
+ "backbone": "resnet50",
106
+ "backbone_config": null,
107
+ "bad_words_ids": null,
108
+ "bbox_cost": 5,
109
+ "bbox_loss_coefficient": 5,
110
+ "begin_suppress_tokens": null,
111
+ "bos_token_id": null,
112
+ "chunk_size_feed_forward": 0,
113
+ "class_cost": 1,
114
+ "cross_attention_hidden_size": null,
115
+ "d_model": 256,
116
+ "decoder_attention_heads": 8,
117
+ "decoder_ffn_dim": 2048,
118
+ "decoder_layerdrop": 0.0,
119
+ "decoder_layers": 6,
120
+ "decoder_start_token_id": null,
121
+ "dice_loss_coefficient": 1,
122
+ "dilation": false,
123
+ "diversity_penalty": 0.0,
124
+ "do_sample": false,
125
+ "dropout": 0.1,
126
+ "early_stopping": false,
127
+ "encoder_attention_heads": 8,
128
+ "encoder_ffn_dim": 2048,
129
+ "encoder_layerdrop": 0.0,
130
+ "encoder_layers": 6,
131
+ "encoder_no_repeat_ngram_size": 0,
132
+ "eos_coefficient": 0.1,
133
+ "eos_token_id": null,
134
+ "exponential_decay_length_penalty": null,
135
+ "finetuning_task": null,
136
+ "forced_bos_token_id": null,
137
+ "forced_eos_token_id": null,
138
+ "giou_cost": 2,
139
+ "giou_loss_coefficient": 2,
140
+ "id2label": {
141
+ "0": "LABEL_0",
142
+ "1": "LABEL_1"
143
+ },
144
+ "init_std": 0.02,
145
+ "init_xavier_std": 1.0,
146
+ "is_decoder": false,
147
+ "is_encoder_decoder": true,
148
+ "label2id": {
149
+ "LABEL_0": 0,
150
+ "LABEL_1": 1
151
+ },
152
+ "length_penalty": 1.0,
153
+ "mask_loss_coefficient": 1,
154
+ "max_length": 20,
155
+ "max_position_embeddings": 1024,
156
+ "min_length": 0,
157
+ "model_type": "detr",
158
+ "no_repeat_ngram_size": 0,
159
+ "num_beam_groups": 1,
160
+ "num_beams": 1,
161
+ "num_channels": 3,
162
+ "num_hidden_layers": 6,
163
+ "num_queries": 100,
164
+ "num_return_sequences": 1,
165
+ "output_attentions": false,
166
+ "output_hidden_states": false,
167
+ "output_scores": false,
168
+ "pad_token_id": null,
169
+ "position_embedding_type": "sine",
170
+ "prefix": null,
171
+ "problem_type": null,
172
+ "pruned_heads": {},
173
+ "remove_invalid_values": false,
174
+ "repetition_penalty": 1.0,
175
+ "return_dict": true,
176
+ "return_dict_in_generate": false,
177
+ "scale_embedding": false,
178
+ "sep_token_id": null,
179
+ "suppress_tokens": null,
180
+ "task_specific_params": null,
181
+ "temperature": 1.0,
182
+ "tf_legacy_loss": false,
183
+ "tie_encoder_decoder": false,
184
+ "tie_word_embeddings": true,
185
+ "tokenizer_class": null,
186
+ "top_k": 50,
187
+ "top_p": 1.0,
188
+ "torch_dtype": null,
189
+ "torchscript": false,
190
+ "transformers_version": "4.17.0.dev0",
191
+ "typical_p": 1.0,
192
+ "use_bfloat16": false,
193
+ "use_pretrained_backbone": true,
194
+ "use_timm_backbone": true
195
+ },
196
+ "dice_weight": 1.0,
197
+ "fpn_feature_size": 256,
198
+ "id2label": {
199
+ "0": "wall",
200
+ "1": "building",
201
+ "2": "sky",
202
+ "3": "floor",
203
+ "4": "tree",
204
+ "5": "ceiling",
205
+ "6": "road, route",
206
+ "7": "bed",
207
+ "8": "window ",
208
+ "9": "grass",
209
+ "10": "cabinet",
210
+ "11": "sidewalk, pavement",
211
+ "12": "person",
212
+ "13": "earth, ground",
213
+ "14": "door",
214
+ "15": "table",
215
+ "16": "mountain, mount",
216
+ "17": "plant",
217
+ "18": "curtain",
218
+ "19": "chair",
219
+ "20": "car",
220
+ "21": "water",
221
+ "22": "painting, picture",
222
+ "23": "sofa",
223
+ "24": "shelf",
224
+ "25": "house",
225
+ "26": "sea",
226
+ "27": "mirror",
227
+ "28": "rug",
228
+ "29": "field",
229
+ "30": "armchair",
230
+ "31": "seat",
231
+ "32": "fence",
232
+ "33": "desk",
233
+ "34": "rock, stone",
234
+ "35": "wardrobe, closet, press",
235
+ "36": "lamp",
236
+ "37": "tub",
237
+ "38": "rail",
238
+ "39": "cushion",
239
+ "40": "base, pedestal, stand",
240
+ "41": "box",
241
+ "42": "column, pillar",
242
+ "43": "signboard, sign",
243
+ "44": "chest of drawers, chest, bureau, dresser",
244
+ "45": "counter",
245
+ "46": "sand",
246
+ "47": "sink",
247
+ "48": "skyscraper",
248
+ "49": "fireplace",
249
+ "50": "refrigerator, icebox",
250
+ "51": "grandstand, covered stand",
251
+ "52": "path",
252
+ "53": "stairs",
253
+ "54": "runway",
254
+ "55": "case, display case, showcase, vitrine",
255
+ "56": "pool table, billiard table, snooker table",
256
+ "57": "pillow",
257
+ "58": "screen door, screen",
258
+ "59": "stairway, staircase",
259
+ "60": "river",
260
+ "61": "bridge, span",
261
+ "62": "bookcase",
262
+ "63": "blind, screen",
263
+ "64": "coffee table",
264
+ "65": "toilet, can, commode, crapper, pot, potty, stool, throne",
265
+ "66": "flower",
266
+ "67": "book",
267
+ "68": "hill",
268
+ "69": "bench",
269
+ "70": "countertop",
270
+ "71": "stove",
271
+ "72": "palm, palm tree",
272
+ "73": "kitchen island",
273
+ "74": "computer",
274
+ "75": "swivel chair",
275
+ "76": "boat",
276
+ "77": "bar",
277
+ "78": "arcade machine",
278
+ "79": "hovel, hut, hutch, shack, shanty",
279
+ "80": "bus",
280
+ "81": "towel",
281
+ "82": "light",
282
+ "83": "truck",
283
+ "84": "tower",
284
+ "85": "chandelier",
285
+ "86": "awning, sunshade, sunblind",
286
+ "87": "street lamp",
287
+ "88": "booth",
288
+ "89": "tv",
289
+ "90": "plane",
290
+ "91": "dirt track",
291
+ "92": "clothes",
292
+ "93": "pole",
293
+ "94": "land, ground, soil",
294
+ "95": "bannister, banister, balustrade, balusters, handrail",
295
+ "96": "escalator, moving staircase, moving stairway",
296
+ "97": "ottoman, pouf, pouffe, puff, hassock",
297
+ "98": "bottle",
298
+ "99": "buffet, counter, sideboard",
299
+ "100": "poster, posting, placard, notice, bill, card",
300
+ "101": "stage",
301
+ "102": "van",
302
+ "103": "ship",
303
+ "104": "fountain",
304
+ "105": "conveyer belt, conveyor belt, conveyer, conveyor, transporter",
305
+ "106": "canopy",
306
+ "107": "washer, automatic washer, washing machine",
307
+ "108": "plaything, toy",
308
+ "109": "pool",
309
+ "110": "stool",
310
+ "111": "barrel, cask",
311
+ "112": "basket, handbasket",
312
+ "113": "falls",
313
+ "114": "tent",
314
+ "115": "bag",
315
+ "116": "minibike, motorbike",
316
+ "117": "cradle",
317
+ "118": "oven",
318
+ "119": "ball",
319
+ "120": "food, solid food",
320
+ "121": "step, stair",
321
+ "122": "tank, storage tank",
322
+ "123": "trade name",
323
+ "124": "microwave",
324
+ "125": "pot",
325
+ "126": "animal",
326
+ "127": "bicycle",
327
+ "128": "lake",
328
+ "129": "dishwasher",
329
+ "130": "screen",
330
+ "131": "blanket, cover",
331
+ "132": "sculpture",
332
+ "133": "hood, exhaust hood",
333
+ "134": "sconce",
334
+ "135": "vase",
335
+ "136": "traffic light",
336
+ "137": "tray",
337
+ "138": "trash can",
338
+ "139": "fan",
339
+ "140": "pier",
340
+ "141": "crt screen",
341
+ "142": "plate",
342
+ "143": "monitor",
343
+ "144": "bulletin board",
344
+ "145": "shower",
345
+ "146": "radiator",
346
+ "147": "glass, drinking glass",
347
+ "148": "clock",
348
+ "149": "flag"
349
+ },
350
+ "init_std": 0.02,
351
+ "init_xavier_std": 1.0,
352
+ "label2id": {
353
+ "animal": 126,
354
+ "arcade machine": 78,
355
+ "armchair": 30,
356
+ "awning, sunshade, sunblind": 86,
357
+ "bag": 115,
358
+ "ball": 119,
359
+ "bannister, banister, balustrade, balusters, handrail": 95,
360
+ "bar": 77,
361
+ "barrel, cask": 111,
362
+ "base, pedestal, stand": 40,
363
+ "basket, handbasket": 112,
364
+ "bed": 7,
365
+ "bench": 69,
366
+ "bicycle": 127,
367
+ "blanket, cover": 131,
368
+ "blind, screen": 63,
369
+ "boat": 76,
370
+ "book": 67,
371
+ "bookcase": 62,
372
+ "booth": 88,
373
+ "bottle": 98,
374
+ "box": 41,
375
+ "bridge, span": 61,
376
+ "buffet, counter, sideboard": 99,
377
+ "building": 1,
378
+ "bulletin board": 144,
379
+ "bus": 80,
380
+ "cabinet": 10,
381
+ "canopy": 106,
382
+ "car": 20,
383
+ "case, display case, showcase, vitrine": 55,
384
+ "ceiling": 5,
385
+ "chair": 19,
386
+ "chandelier": 85,
387
+ "chest of drawers, chest, bureau, dresser": 44,
388
+ "clock": 148,
389
+ "clothes": 92,
390
+ "coffee table": 64,
391
+ "column, pillar": 42,
392
+ "computer": 74,
393
+ "conveyer belt, conveyor belt, conveyer, conveyor, transporter": 105,
394
+ "counter": 45,
395
+ "countertop": 70,
396
+ "cradle": 117,
397
+ "crt screen": 141,
398
+ "curtain": 18,
399
+ "cushion": 39,
400
+ "desk": 33,
401
+ "dirt track": 91,
402
+ "dishwasher": 129,
403
+ "door": 14,
404
+ "earth, ground": 13,
405
+ "escalator, moving staircase, moving stairway": 96,
406
+ "falls": 113,
407
+ "fan": 139,
408
+ "fence": 32,
409
+ "field": 29,
410
+ "fireplace": 49,
411
+ "flag": 149,
412
+ "floor": 3,
413
+ "flower": 66,
414
+ "food, solid food": 120,
415
+ "fountain": 104,
416
+ "glass, drinking glass": 147,
417
+ "grandstand, covered stand": 51,
418
+ "grass": 9,
419
+ "hill": 68,
420
+ "hood, exhaust hood": 133,
421
+ "house": 25,
422
+ "hovel, hut, hutch, shack, shanty": 79,
423
+ "kitchen island": 73,
424
+ "lake": 128,
425
+ "lamp": 36,
426
+ "land, ground, soil": 94,
427
+ "light": 82,
428
+ "microwave": 124,
429
+ "minibike, motorbike": 116,
430
+ "mirror": 27,
431
+ "monitor": 143,
432
+ "mountain, mount": 16,
433
+ "ottoman, pouf, pouffe, puff, hassock": 97,
434
+ "oven": 118,
435
+ "painting, picture": 22,
436
+ "palm, palm tree": 72,
437
+ "path": 52,
438
+ "person": 12,
439
+ "pier": 140,
440
+ "pillow": 57,
441
+ "plane": 90,
442
+ "plant": 17,
443
+ "plate": 142,
444
+ "plaything, toy": 108,
445
+ "pole": 93,
446
+ "pool": 109,
447
+ "pool table, billiard table, snooker table": 56,
448
+ "poster, posting, placard, notice, bill, card": 100,
449
+ "pot": 125,
450
+ "radiator": 146,
451
+ "rail": 38,
452
+ "refrigerator, icebox": 50,
453
+ "river": 60,
454
+ "road, route": 6,
455
+ "rock, stone": 34,
456
+ "rug": 28,
457
+ "runway": 54,
458
+ "sand": 46,
459
+ "sconce": 134,
460
+ "screen": 130,
461
+ "screen door, screen": 58,
462
+ "sculpture": 132,
463
+ "sea": 26,
464
+ "seat": 31,
465
+ "shelf": 24,
466
+ "ship": 103,
467
+ "shower": 145,
468
+ "sidewalk, pavement": 11,
469
+ "signboard, sign": 43,
470
+ "sink": 47,
471
+ "sky": 2,
472
+ "skyscraper": 48,
473
+ "sofa": 23,
474
+ "stage": 101,
475
+ "stairs": 53,
476
+ "stairway, staircase": 59,
477
+ "step, stair": 121,
478
+ "stool": 110,
479
+ "stove": 71,
480
+ "street lamp": 87,
481
+ "swivel chair": 75,
482
+ "table": 15,
483
+ "tank, storage tank": 122,
484
+ "tent": 114,
485
+ "toilet, can, commode, crapper, pot, potty, stool, throne": 65,
486
+ "towel": 81,
487
+ "tower": 84,
488
+ "trade name": 123,
489
+ "traffic light": 136,
490
+ "trash can": 138,
491
+ "tray": 137,
492
+ "tree": 4,
493
+ "truck": 83,
494
+ "tub": 37,
495
+ "tv": 89,
496
+ "van": 102,
497
+ "vase": 135,
498
+ "wall": 0,
499
+ "wardrobe, closet, press": 35,
500
+ "washer, automatic washer, washing machine": 107,
501
+ "water": 21,
502
+ "window ": 8
503
+ },
504
+ "mask_feature_size": 256,
505
+ "mask_weight": 20.0,
506
+ "model_type": "maskformer",
507
+ "no_object_weight": 0.1,
508
+ "num_attention_heads": 8,
509
+ "num_hidden_layers": 6,
510
+ "num_queries": 100,
511
+ "output_auxiliary_logits": null,
512
+ "torch_dtype": "float32",
513
+ "transformers_version": null,
514
+ "use_auxiliary_loss": false
515
+ }
preprocessor_config.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_max_size": 2048,
3
+ "do_normalize": true,
4
+ "do_reduce_labels": false,
5
+ "do_rescale": true,
6
+ "do_resize": true,
7
+ "ignore_index": 255,
8
+ "image_mean": [0.48500001430511475, 0.4560000002384186, 0.4059999883174896],
9
+ "image_processor_type": "MaskFormerImageProcessor",
10
+ "image_std": [0.2290000021457672, 0.2239999920129776, 0.22499999403953552],
11
+ "resample": 2,
12
+ "rescale_factor": 0.00392156862745098,
13
+ "size": {
14
+ "longest_edge": 2048,
15
+ "shortest_edge": 512
16
+ },
17
+ "size_divisor": 32
18
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5b1d58082fb676c8471d46d2ac88a0391b88c683463a0020295914697db03126
3
+ size 167419213