remove files
Browse files- config.json +9 -506
- detr_256_6_6_torchvision.yaml +0 -45
- detr_segm_256_6_6_torchvision.yaml +0 -46
- preprocessor_config.json +2 -2
- pytorch_model.bin +0 -3
config.json
CHANGED
@@ -1,8 +1,9 @@
|
|
1 |
{
|
|
|
2 |
"activation_dropout": 0.0,
|
3 |
"activation_function": "relu",
|
4 |
"architectures": [
|
5 |
-
"
|
6 |
],
|
7 |
"attention_dropout": 0.0,
|
8 |
"auxiliary_loss": false,
|
@@ -10,7 +11,6 @@
|
|
10 |
"bbox_cost": 5,
|
11 |
"bbox_loss_coefficient": 5,
|
12 |
"class_cost": 1,
|
13 |
-
"classifier_dropout": 0.0,
|
14 |
"d_model": 256,
|
15 |
"decoder_attention_heads": 8,
|
16 |
"decoder_ffn_dim": 2048,
|
@@ -27,512 +27,14 @@
|
|
27 |
"giou_cost": 2,
|
28 |
"giou_loss_coefficient": 2,
|
29 |
"id2label": {
|
30 |
-
"0": "
|
31 |
-
|
32 |
-
"10": "traffic light",
|
33 |
-
"100": "cardboard",
|
34 |
-
"101": "carpet",
|
35 |
-
"102": "ceiling-other",
|
36 |
-
"103": "ceiling-tile",
|
37 |
-
"104": "cloth",
|
38 |
-
"105": "clothes",
|
39 |
-
"106": "clouds",
|
40 |
-
"107": "counter",
|
41 |
-
"108": "cupboard",
|
42 |
-
"109": "curtain",
|
43 |
-
"11": "fire hydrant",
|
44 |
-
"110": "desk-stuff",
|
45 |
-
"111": "dirt",
|
46 |
-
"112": "door-stuff",
|
47 |
-
"113": "fence",
|
48 |
-
"114": "floor-marble",
|
49 |
-
"115": "floor-other",
|
50 |
-
"116": "floor-stone",
|
51 |
-
"117": "floor-tile",
|
52 |
-
"118": "floor-wood",
|
53 |
-
"119": "flower",
|
54 |
-
"12": "street sign",
|
55 |
-
"120": "fog",
|
56 |
-
"121": "food-other",
|
57 |
-
"122": "fruit",
|
58 |
-
"123": "furniture-other",
|
59 |
-
"124": "grass",
|
60 |
-
"125": "gravel",
|
61 |
-
"126": "ground-other",
|
62 |
-
"127": "hill",
|
63 |
-
"128": "house",
|
64 |
-
"129": "leaves",
|
65 |
-
"13": "stop sign",
|
66 |
-
"130": "light",
|
67 |
-
"131": "mat",
|
68 |
-
"132": "metal",
|
69 |
-
"133": "mirror-stuff",
|
70 |
-
"134": "moss",
|
71 |
-
"135": "mountain",
|
72 |
-
"136": "mud",
|
73 |
-
"137": "napkin",
|
74 |
-
"138": "net",
|
75 |
-
"139": "paper",
|
76 |
-
"14": "parking meter",
|
77 |
-
"140": "pavement",
|
78 |
-
"141": "pillow",
|
79 |
-
"142": "plant-other",
|
80 |
-
"143": "plastic",
|
81 |
-
"144": "platform",
|
82 |
-
"145": "playingfield",
|
83 |
-
"146": "railing",
|
84 |
-
"147": "railroad",
|
85 |
-
"148": "river",
|
86 |
-
"149": "road",
|
87 |
-
"15": "bench",
|
88 |
-
"150": "rock",
|
89 |
-
"151": "roof",
|
90 |
-
"152": "rug",
|
91 |
-
"153": "salad",
|
92 |
-
"154": "sand",
|
93 |
-
"155": "sea",
|
94 |
-
"156": "shelf",
|
95 |
-
"157": "sky-other",
|
96 |
-
"158": "skyscraper",
|
97 |
-
"159": "snow",
|
98 |
-
"16": "bird",
|
99 |
-
"160": "solid-other",
|
100 |
-
"161": "stairs",
|
101 |
-
"162": "stone",
|
102 |
-
"163": "straw",
|
103 |
-
"164": "structural-other",
|
104 |
-
"165": "table",
|
105 |
-
"166": "tent",
|
106 |
-
"167": "textile-other",
|
107 |
-
"168": "towel",
|
108 |
-
"169": "tree",
|
109 |
-
"17": "cat",
|
110 |
-
"170": "vegetable",
|
111 |
-
"171": "wall-brick",
|
112 |
-
"172": "wall-concrete",
|
113 |
-
"173": "wall-other",
|
114 |
-
"174": "wall-panel",
|
115 |
-
"175": "wall-stone",
|
116 |
-
"176": "wall-tile",
|
117 |
-
"177": "wall-wood",
|
118 |
-
"178": "water-other",
|
119 |
-
"179": "waterdrops",
|
120 |
-
"18": "dog",
|
121 |
-
"180": "window-blind",
|
122 |
-
"181": "window-other",
|
123 |
-
"182": "wood",
|
124 |
-
"183": "LABEL_183",
|
125 |
-
"184": "LABEL_184",
|
126 |
-
"185": "LABEL_185",
|
127 |
-
"186": "LABEL_186",
|
128 |
-
"187": "LABEL_187",
|
129 |
-
"188": "LABEL_188",
|
130 |
-
"189": "LABEL_189",
|
131 |
-
"19": "horse",
|
132 |
-
"190": "LABEL_190",
|
133 |
-
"191": "LABEL_191",
|
134 |
-
"192": "LABEL_192",
|
135 |
-
"193": "LABEL_193",
|
136 |
-
"194": "LABEL_194",
|
137 |
-
"195": "LABEL_195",
|
138 |
-
"196": "LABEL_196",
|
139 |
-
"197": "LABEL_197",
|
140 |
-
"198": "LABEL_198",
|
141 |
-
"199": "LABEL_199",
|
142 |
-
"2": "bicycle",
|
143 |
-
"20": "sheep",
|
144 |
-
"200": "LABEL_200",
|
145 |
-
"201": "LABEL_201",
|
146 |
-
"202": "LABEL_202",
|
147 |
-
"203": "LABEL_203",
|
148 |
-
"204": "LABEL_204",
|
149 |
-
"205": "LABEL_205",
|
150 |
-
"206": "LABEL_206",
|
151 |
-
"207": "LABEL_207",
|
152 |
-
"208": "LABEL_208",
|
153 |
-
"209": "LABEL_209",
|
154 |
-
"21": "cow",
|
155 |
-
"210": "LABEL_210",
|
156 |
-
"211": "LABEL_211",
|
157 |
-
"212": "LABEL_212",
|
158 |
-
"213": "LABEL_213",
|
159 |
-
"214": "LABEL_214",
|
160 |
-
"215": "LABEL_215",
|
161 |
-
"216": "LABEL_216",
|
162 |
-
"217": "LABEL_217",
|
163 |
-
"218": "LABEL_218",
|
164 |
-
"219": "LABEL_219",
|
165 |
-
"22": "elephant",
|
166 |
-
"220": "LABEL_220",
|
167 |
-
"221": "LABEL_221",
|
168 |
-
"222": "LABEL_222",
|
169 |
-
"223": "LABEL_223",
|
170 |
-
"224": "LABEL_224",
|
171 |
-
"225": "LABEL_225",
|
172 |
-
"226": "LABEL_226",
|
173 |
-
"227": "LABEL_227",
|
174 |
-
"228": "LABEL_228",
|
175 |
-
"229": "LABEL_229",
|
176 |
-
"23": "bear",
|
177 |
-
"230": "LABEL_230",
|
178 |
-
"231": "LABEL_231",
|
179 |
-
"232": "LABEL_232",
|
180 |
-
"233": "LABEL_233",
|
181 |
-
"234": "LABEL_234",
|
182 |
-
"235": "LABEL_235",
|
183 |
-
"236": "LABEL_236",
|
184 |
-
"237": "LABEL_237",
|
185 |
-
"238": "LABEL_238",
|
186 |
-
"239": "LABEL_239",
|
187 |
-
"24": "zebra",
|
188 |
-
"240": "LABEL_240",
|
189 |
-
"241": "LABEL_241",
|
190 |
-
"242": "LABEL_242",
|
191 |
-
"243": "LABEL_243",
|
192 |
-
"244": "LABEL_244",
|
193 |
-
"245": "LABEL_245",
|
194 |
-
"246": "LABEL_246",
|
195 |
-
"247": "LABEL_247",
|
196 |
-
"248": "LABEL_248",
|
197 |
-
"249": "LABEL_249",
|
198 |
-
"25": "giraffe",
|
199 |
-
"26": "hat",
|
200 |
-
"27": "backpack",
|
201 |
-
"28": "umbrella",
|
202 |
-
"29": "shoe",
|
203 |
-
"3": "car",
|
204 |
-
"30": "eye glasses",
|
205 |
-
"31": "handbag",
|
206 |
-
"32": "tie",
|
207 |
-
"33": "suitcase",
|
208 |
-
"34": "frisbee",
|
209 |
-
"35": "skis",
|
210 |
-
"36": "snowboard",
|
211 |
-
"37": "sports ball",
|
212 |
-
"38": "kite",
|
213 |
-
"39": "baseball bat",
|
214 |
-
"4": "motorcycle",
|
215 |
-
"40": "baseball glove",
|
216 |
-
"41": "skateboard",
|
217 |
-
"42": "surfboard",
|
218 |
-
"43": "tennis racket",
|
219 |
-
"44": "bottle",
|
220 |
-
"45": "plate",
|
221 |
-
"46": "wine glass",
|
222 |
-
"47": "cup",
|
223 |
-
"48": "fork",
|
224 |
-
"49": "knife",
|
225 |
-
"5": "airplane",
|
226 |
-
"50": "spoon",
|
227 |
-
"51": "bowl",
|
228 |
-
"52": "banana",
|
229 |
-
"53": "apple",
|
230 |
-
"54": "sandwich",
|
231 |
-
"55": "orange",
|
232 |
-
"56": "broccoli",
|
233 |
-
"57": "carrot",
|
234 |
-
"58": "hot dog",
|
235 |
-
"59": "pizza",
|
236 |
-
"6": "bus",
|
237 |
-
"60": "donut",
|
238 |
-
"61": "cake",
|
239 |
-
"62": "chair",
|
240 |
-
"63": "couch",
|
241 |
-
"64": "potted plant",
|
242 |
-
"65": "bed",
|
243 |
-
"66": "mirror",
|
244 |
-
"67": "dining table",
|
245 |
-
"68": "window",
|
246 |
-
"69": "desk",
|
247 |
-
"7": "train",
|
248 |
-
"70": "toilet",
|
249 |
-
"71": "door",
|
250 |
-
"72": "tv",
|
251 |
-
"73": "laptop",
|
252 |
-
"74": "mouse",
|
253 |
-
"75": "remote",
|
254 |
-
"76": "keyboard",
|
255 |
-
"77": "cell phone",
|
256 |
-
"78": "microwave",
|
257 |
-
"79": "oven",
|
258 |
-
"8": "truck",
|
259 |
-
"80": "toaster",
|
260 |
-
"81": "sink",
|
261 |
-
"82": "refrigerator",
|
262 |
-
"83": "blender",
|
263 |
-
"84": "book",
|
264 |
-
"85": "clock",
|
265 |
-
"86": "vase",
|
266 |
-
"87": "scissors",
|
267 |
-
"88": "teddy bear",
|
268 |
-
"89": "hair drier",
|
269 |
-
"9": "boat",
|
270 |
-
"90": "toothbrush",
|
271 |
-
"91": "hair brush",
|
272 |
-
"92": "banner",
|
273 |
-
"93": "blanket",
|
274 |
-
"94": "branch",
|
275 |
-
"95": "bridge",
|
276 |
-
"96": "building-other",
|
277 |
-
"97": "bush",
|
278 |
-
"98": "cabinet",
|
279 |
-
"99": "cage"
|
280 |
-
},
|
281 |
"init_std": 0.02,
|
282 |
"init_xavier_std": 1.0,
|
283 |
"is_encoder_decoder": true,
|
284 |
"label2id": {
|
285 |
-
"
|
286 |
-
|
287 |
-
"LABEL_185": 185,
|
288 |
-
"LABEL_186": 186,
|
289 |
-
"LABEL_187": 187,
|
290 |
-
"LABEL_188": 188,
|
291 |
-
"LABEL_189": 189,
|
292 |
-
"LABEL_190": 190,
|
293 |
-
"LABEL_191": 191,
|
294 |
-
"LABEL_192": 192,
|
295 |
-
"LABEL_193": 193,
|
296 |
-
"LABEL_194": 194,
|
297 |
-
"LABEL_195": 195,
|
298 |
-
"LABEL_196": 196,
|
299 |
-
"LABEL_197": 197,
|
300 |
-
"LABEL_198": 198,
|
301 |
-
"LABEL_199": 199,
|
302 |
-
"LABEL_200": 200,
|
303 |
-
"LABEL_201": 201,
|
304 |
-
"LABEL_202": 202,
|
305 |
-
"LABEL_203": 203,
|
306 |
-
"LABEL_204": 204,
|
307 |
-
"LABEL_205": 205,
|
308 |
-
"LABEL_206": 206,
|
309 |
-
"LABEL_207": 207,
|
310 |
-
"LABEL_208": 208,
|
311 |
-
"LABEL_209": 209,
|
312 |
-
"LABEL_210": 210,
|
313 |
-
"LABEL_211": 211,
|
314 |
-
"LABEL_212": 212,
|
315 |
-
"LABEL_213": 213,
|
316 |
-
"LABEL_214": 214,
|
317 |
-
"LABEL_215": 215,
|
318 |
-
"LABEL_216": 216,
|
319 |
-
"LABEL_217": 217,
|
320 |
-
"LABEL_218": 218,
|
321 |
-
"LABEL_219": 219,
|
322 |
-
"LABEL_220": 220,
|
323 |
-
"LABEL_221": 221,
|
324 |
-
"LABEL_222": 222,
|
325 |
-
"LABEL_223": 223,
|
326 |
-
"LABEL_224": 224,
|
327 |
-
"LABEL_225": 225,
|
328 |
-
"LABEL_226": 226,
|
329 |
-
"LABEL_227": 227,
|
330 |
-
"LABEL_228": 228,
|
331 |
-
"LABEL_229": 229,
|
332 |
-
"LABEL_230": 230,
|
333 |
-
"LABEL_231": 231,
|
334 |
-
"LABEL_232": 232,
|
335 |
-
"LABEL_233": 233,
|
336 |
-
"LABEL_234": 234,
|
337 |
-
"LABEL_235": 235,
|
338 |
-
"LABEL_236": 236,
|
339 |
-
"LABEL_237": 237,
|
340 |
-
"LABEL_238": 238,
|
341 |
-
"LABEL_239": 239,
|
342 |
-
"LABEL_240": 240,
|
343 |
-
"LABEL_241": 241,
|
344 |
-
"LABEL_242": 242,
|
345 |
-
"LABEL_243": 243,
|
346 |
-
"LABEL_244": 244,
|
347 |
-
"LABEL_245": 245,
|
348 |
-
"LABEL_246": 246,
|
349 |
-
"LABEL_247": 247,
|
350 |
-
"LABEL_248": 248,
|
351 |
-
"LABEL_249": 249,
|
352 |
-
"N/A": 0,
|
353 |
-
"airplane": 5,
|
354 |
-
"apple": 53,
|
355 |
-
"backpack": 27,
|
356 |
-
"banana": 52,
|
357 |
-
"banner": 92,
|
358 |
-
"baseball bat": 39,
|
359 |
-
"baseball glove": 40,
|
360 |
-
"bear": 23,
|
361 |
-
"bed": 65,
|
362 |
-
"bench": 15,
|
363 |
-
"bicycle": 2,
|
364 |
-
"bird": 16,
|
365 |
-
"blanket": 93,
|
366 |
-
"blender": 83,
|
367 |
-
"boat": 9,
|
368 |
-
"book": 84,
|
369 |
-
"bottle": 44,
|
370 |
-
"bowl": 51,
|
371 |
-
"branch": 94,
|
372 |
-
"bridge": 95,
|
373 |
-
"broccoli": 56,
|
374 |
-
"building-other": 96,
|
375 |
-
"bus": 6,
|
376 |
-
"bush": 97,
|
377 |
-
"cabinet": 98,
|
378 |
-
"cage": 99,
|
379 |
-
"cake": 61,
|
380 |
-
"car": 3,
|
381 |
-
"cardboard": 100,
|
382 |
-
"carpet": 101,
|
383 |
-
"carrot": 57,
|
384 |
-
"cat": 17,
|
385 |
-
"ceiling-other": 102,
|
386 |
-
"ceiling-tile": 103,
|
387 |
-
"cell phone": 77,
|
388 |
-
"chair": 62,
|
389 |
-
"clock": 85,
|
390 |
-
"cloth": 104,
|
391 |
-
"clothes": 105,
|
392 |
-
"clouds": 106,
|
393 |
-
"couch": 63,
|
394 |
-
"counter": 107,
|
395 |
-
"cow": 21,
|
396 |
-
"cup": 47,
|
397 |
-
"cupboard": 108,
|
398 |
-
"curtain": 109,
|
399 |
-
"desk": 69,
|
400 |
-
"desk-stuff": 110,
|
401 |
-
"dining table": 67,
|
402 |
-
"dirt": 111,
|
403 |
-
"dog": 18,
|
404 |
-
"donut": 60,
|
405 |
-
"door": 71,
|
406 |
-
"door-stuff": 112,
|
407 |
-
"elephant": 22,
|
408 |
-
"eye glasses": 30,
|
409 |
-
"fence": 113,
|
410 |
-
"fire hydrant": 11,
|
411 |
-
"floor-marble": 114,
|
412 |
-
"floor-other": 115,
|
413 |
-
"floor-stone": 116,
|
414 |
-
"floor-tile": 117,
|
415 |
-
"floor-wood": 118,
|
416 |
-
"flower": 119,
|
417 |
-
"fog": 120,
|
418 |
-
"food-other": 121,
|
419 |
-
"fork": 48,
|
420 |
-
"frisbee": 34,
|
421 |
-
"fruit": 122,
|
422 |
-
"furniture-other": 123,
|
423 |
-
"giraffe": 25,
|
424 |
-
"grass": 124,
|
425 |
-
"gravel": 125,
|
426 |
-
"ground-other": 126,
|
427 |
-
"hair brush": 91,
|
428 |
-
"hair drier": 89,
|
429 |
-
"handbag": 31,
|
430 |
-
"hat": 26,
|
431 |
-
"hill": 127,
|
432 |
-
"horse": 19,
|
433 |
-
"hot dog": 58,
|
434 |
-
"house": 128,
|
435 |
-
"keyboard": 76,
|
436 |
-
"kite": 38,
|
437 |
-
"knife": 49,
|
438 |
-
"laptop": 73,
|
439 |
-
"leaves": 129,
|
440 |
-
"light": 130,
|
441 |
-
"mat": 131,
|
442 |
-
"metal": 132,
|
443 |
-
"microwave": 78,
|
444 |
-
"mirror": 66,
|
445 |
-
"mirror-stuff": 133,
|
446 |
-
"moss": 134,
|
447 |
-
"motorcycle": 4,
|
448 |
-
"mountain": 135,
|
449 |
-
"mouse": 74,
|
450 |
-
"mud": 136,
|
451 |
-
"napkin": 137,
|
452 |
-
"net": 138,
|
453 |
-
"orange": 55,
|
454 |
-
"oven": 79,
|
455 |
-
"paper": 139,
|
456 |
-
"parking meter": 14,
|
457 |
-
"pavement": 140,
|
458 |
-
"person": 1,
|
459 |
-
"pillow": 141,
|
460 |
-
"pizza": 59,
|
461 |
-
"plant-other": 142,
|
462 |
-
"plastic": 143,
|
463 |
-
"plate": 45,
|
464 |
-
"platform": 144,
|
465 |
-
"playingfield": 145,
|
466 |
-
"potted plant": 64,
|
467 |
-
"railing": 146,
|
468 |
-
"railroad": 147,
|
469 |
-
"refrigerator": 82,
|
470 |
-
"remote": 75,
|
471 |
-
"river": 148,
|
472 |
-
"road": 149,
|
473 |
-
"rock": 150,
|
474 |
-
"roof": 151,
|
475 |
-
"rug": 152,
|
476 |
-
"salad": 153,
|
477 |
-
"sand": 154,
|
478 |
-
"sandwich": 54,
|
479 |
-
"scissors": 87,
|
480 |
-
"sea": 155,
|
481 |
-
"sheep": 20,
|
482 |
-
"shelf": 156,
|
483 |
-
"shoe": 29,
|
484 |
-
"sink": 81,
|
485 |
-
"skateboard": 41,
|
486 |
-
"skis": 35,
|
487 |
-
"sky-other": 157,
|
488 |
-
"skyscraper": 158,
|
489 |
-
"snow": 159,
|
490 |
-
"snowboard": 36,
|
491 |
-
"solid-other": 160,
|
492 |
-
"spoon": 50,
|
493 |
-
"sports ball": 37,
|
494 |
-
"stairs": 161,
|
495 |
-
"stone": 162,
|
496 |
-
"stop sign": 13,
|
497 |
-
"straw": 163,
|
498 |
-
"street sign": 12,
|
499 |
-
"structural-other": 164,
|
500 |
-
"suitcase": 33,
|
501 |
-
"surfboard": 42,
|
502 |
-
"table": 165,
|
503 |
-
"teddy bear": 88,
|
504 |
-
"tennis racket": 43,
|
505 |
-
"tent": 166,
|
506 |
-
"textile-other": 167,
|
507 |
-
"tie": 32,
|
508 |
-
"toaster": 80,
|
509 |
-
"toilet": 70,
|
510 |
-
"toothbrush": 90,
|
511 |
-
"towel": 168,
|
512 |
-
"traffic light": 10,
|
513 |
-
"train": 7,
|
514 |
-
"tree": 169,
|
515 |
-
"truck": 8,
|
516 |
-
"tv": 72,
|
517 |
-
"umbrella": 28,
|
518 |
-
"vase": 86,
|
519 |
-
"vegetable": 170,
|
520 |
-
"wall-brick": 171,
|
521 |
-
"wall-concrete": 172,
|
522 |
-
"wall-other": 173,
|
523 |
-
"wall-panel": 174,
|
524 |
-
"wall-stone": 175,
|
525 |
-
"wall-tile": 176,
|
526 |
-
"wall-wood": 177,
|
527 |
-
"water-other": 178,
|
528 |
-
"waterdrops": 179,
|
529 |
-
"window": 68,
|
530 |
-
"window-blind": 180,
|
531 |
-
"window-other": 181,
|
532 |
-
"wine glass": 46,
|
533 |
-
"wood": 182,
|
534 |
-
"zebra": 24
|
535 |
-
},
|
536 |
"mask_loss_coefficient": 1,
|
537 |
"max_position_embeddings": 1024,
|
538 |
"model_type": "detr",
|
@@ -540,5 +42,6 @@
|
|
540 |
"num_queries": 100,
|
541 |
"position_embedding_type": "sine",
|
542 |
"scale_embedding": false,
|
543 |
-
"
|
544 |
-
|
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "facebook/detr-resnet-50",
|
3 |
"activation_dropout": 0.0,
|
4 |
"activation_function": "relu",
|
5 |
"architectures": [
|
6 |
+
"DetrForObjectDetection"
|
7 |
],
|
8 |
"attention_dropout": 0.0,
|
9 |
"auxiliary_loss": false,
|
|
|
11 |
"bbox_cost": 5,
|
12 |
"bbox_loss_coefficient": 5,
|
13 |
"class_cost": 1,
|
|
|
14 |
"d_model": 256,
|
15 |
"decoder_attention_heads": 8,
|
16 |
"decoder_ffn_dim": 2048,
|
|
|
27 |
"giou_cost": 2,
|
28 |
"giou_loss_coefficient": 2,
|
29 |
"id2label": {
|
30 |
+
"0": "face",
|
31 |
+
},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
32 |
"init_std": 0.02,
|
33 |
"init_xavier_std": 1.0,
|
34 |
"is_encoder_decoder": true,
|
35 |
"label2id": {
|
36 |
+
"face": 1,
|
37 |
+
},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
38 |
"mask_loss_coefficient": 1,
|
39 |
"max_position_embeddings": 1024,
|
40 |
"model_type": "detr",
|
|
|
42 |
"num_queries": 100,
|
43 |
"position_embedding_type": "sine",
|
44 |
"scale_embedding": false,
|
45 |
+
"torch_dtype": "float32",
|
46 |
+
"transformers_version": "4.17.0"
|
47 |
+
}
|
detr_256_6_6_torchvision.yaml
DELETED
@@ -1,45 +0,0 @@
|
|
1 |
-
MODEL:
|
2 |
-
META_ARCHITECTURE: "Detr"
|
3 |
-
WEIGHTS: "detectron2://ImageNetPretrained/torchvision/R-50.pkl"
|
4 |
-
PIXEL_MEAN: [123.675, 116.280, 103.530]
|
5 |
-
PIXEL_STD: [58.395, 57.120, 57.375]
|
6 |
-
MASK_ON: False
|
7 |
-
RESNETS:
|
8 |
-
DEPTH: 50
|
9 |
-
STRIDE_IN_1X1: False
|
10 |
-
OUT_FEATURES: ["res2", "res3", "res4", "res5"]
|
11 |
-
DETR:
|
12 |
-
GIOU_WEIGHT: 2.0
|
13 |
-
L1_WEIGHT: 5.0
|
14 |
-
NUM_OBJECT_QUERIES: 100
|
15 |
-
DATASETS:
|
16 |
-
TRAIN: ("coco_2017_train",)
|
17 |
-
TEST: ("coco_2017_val",)
|
18 |
-
SOLVER:
|
19 |
-
IMS_PER_BATCH: 64
|
20 |
-
BASE_LR: 0.0001
|
21 |
-
STEPS: (369600,)
|
22 |
-
MAX_ITER: 554400
|
23 |
-
WARMUP_FACTOR: 1.0
|
24 |
-
WARMUP_ITERS: 10
|
25 |
-
WEIGHT_DECAY: 0.0001
|
26 |
-
OPTIMIZER: "ADAMW"
|
27 |
-
BACKBONE_MULTIPLIER: 0.1
|
28 |
-
CLIP_GRADIENTS:
|
29 |
-
ENABLED: True
|
30 |
-
CLIP_TYPE: "full_model"
|
31 |
-
CLIP_VALUE: 0.01
|
32 |
-
NORM_TYPE: 2.0
|
33 |
-
INPUT:
|
34 |
-
MIN_SIZE_TRAIN: (480, 512, 544, 576, 608, 640, 672, 704, 736, 768, 800)
|
35 |
-
CROP:
|
36 |
-
ENABLED: True
|
37 |
-
TYPE: "absolute_range"
|
38 |
-
SIZE: (384, 600)
|
39 |
-
FORMAT: "RGB"
|
40 |
-
TEST:
|
41 |
-
EVAL_PERIOD: 4000
|
42 |
-
DATALOADER:
|
43 |
-
FILTER_EMPTY_ANNOTATIONS: False
|
44 |
-
NUM_WORKERS: 4
|
45 |
-
VERSION: 2
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
detr_segm_256_6_6_torchvision.yaml
DELETED
@@ -1,46 +0,0 @@
|
|
1 |
-
MODEL:
|
2 |
-
META_ARCHITECTURE: "Detr"
|
3 |
-
# WEIGHTS: "detectron2://ImageNetPretrained/torchvision/R-50.pkl"
|
4 |
-
PIXEL_MEAN: [123.675, 116.280, 103.530]
|
5 |
-
PIXEL_STD: [58.395, 57.120, 57.375]
|
6 |
-
MASK_ON: True
|
7 |
-
RESNETS:
|
8 |
-
DEPTH: 50
|
9 |
-
STRIDE_IN_1X1: False
|
10 |
-
OUT_FEATURES: ["res2", "res3", "res4", "res5"]
|
11 |
-
DETR:
|
12 |
-
GIOU_WEIGHT: 2.0
|
13 |
-
L1_WEIGHT: 5.0
|
14 |
-
NUM_OBJECT_QUERIES: 100
|
15 |
-
FROZEN_WEIGHTS: ''
|
16 |
-
DATASETS:
|
17 |
-
TRAIN: ("coco_2017_train",)
|
18 |
-
TEST: ("coco_2017_val",)
|
19 |
-
SOLVER:
|
20 |
-
IMS_PER_BATCH: 64
|
21 |
-
BASE_LR: 0.0001
|
22 |
-
STEPS: (55440,)
|
23 |
-
MAX_ITER: 92400
|
24 |
-
WARMUP_FACTOR: 1.0
|
25 |
-
WARMUP_ITERS: 10
|
26 |
-
WEIGHT_DECAY: 0.0001
|
27 |
-
OPTIMIZER: "ADAMW"
|
28 |
-
BACKBONE_MULTIPLIER: 0.1
|
29 |
-
CLIP_GRADIENTS:
|
30 |
-
ENABLED: True
|
31 |
-
CLIP_TYPE: "full_model"
|
32 |
-
CLIP_VALUE: 0.01
|
33 |
-
NORM_TYPE: 2.0
|
34 |
-
INPUT:
|
35 |
-
MIN_SIZE_TRAIN: (480, 512, 544, 576, 608, 640, 672, 704, 736, 768, 800)
|
36 |
-
CROP:
|
37 |
-
ENABLED: True
|
38 |
-
TYPE: "absolute_range"
|
39 |
-
SIZE: (384, 600)
|
40 |
-
FORMAT: "RGB"
|
41 |
-
TEST:
|
42 |
-
EVAL_PERIOD: 4000
|
43 |
-
DATALOADER:
|
44 |
-
FILTER_EMPTY_ANNOTATIONS: False
|
45 |
-
NUM_WORKERS: 4
|
46 |
-
VERSION: 2
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
preprocessor_config.json
CHANGED
@@ -2,7 +2,7 @@
|
|
2 |
"do_normalize": true,
|
3 |
"do_resize": true,
|
4 |
"feature_extractor_type": "DetrFeatureExtractor",
|
5 |
-
"format": "
|
6 |
"image_mean": [
|
7 |
0.485,
|
8 |
0.456,
|
@@ -15,4 +15,4 @@
|
|
15 |
],
|
16 |
"max_size": 1333,
|
17 |
"size": 800
|
18 |
-
}
|
|
|
2 |
"do_normalize": true,
|
3 |
"do_resize": true,
|
4 |
"feature_extractor_type": "DetrFeatureExtractor",
|
5 |
+
"format": "coco_detection",
|
6 |
"image_mean": [
|
7 |
0.485,
|
8 |
0.456,
|
|
|
15 |
],
|
16 |
"max_size": 1333,
|
17 |
"size": 800
|
18 |
+
}
|
pytorch_model.bin
DELETED
@@ -1,3 +0,0 @@
|
|
1 |
-
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:3f8024c4744402adf5ebba3be495c91e5af388d73fb32f1e0a93c053767abeca
|
3 |
-
size 172245279
|
|
|
|
|
|
|
|