MTWD commited on
Commit
e416ec7
1 Parent(s): c40328d

Training in progress, step 100

Browse files
Files changed (4) hide show
  1. config.json +311 -0
  2. model.safetensors +3 -0
  3. preprocessor_config.json +45 -0
  4. training_args.bin +3 -0
config.json ADDED
@@ -0,0 +1,311 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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,
10
+ "backbone": "resnet50",
11
+ "backbone_config": null,
12
+ "backbone_kwargs": {
13
+ "in_chans": 3,
14
+ "out_indices": [
15
+ 1,
16
+ 2,
17
+ 3,
18
+ 4
19
+ ]
20
+ },
21
+ "bbox_cost": 5,
22
+ "bbox_loss_coefficient": 5,
23
+ "class_cost": 1,
24
+ "classifier_dropout": 0.0,
25
+ "d_model": 256,
26
+ "decoder_attention_heads": 8,
27
+ "decoder_ffn_dim": 2048,
28
+ "decoder_layerdrop": 0.0,
29
+ "decoder_layers": 6,
30
+ "dice_loss_coefficient": 1,
31
+ "dilation": false,
32
+ "dropout": 0.1,
33
+ "encoder_attention_heads": 8,
34
+ "encoder_ffn_dim": 2048,
35
+ "encoder_layerdrop": 0.0,
36
+ "encoder_layers": 6,
37
+ "eos_coefficient": 0.1,
38
+ "giou_cost": 2,
39
+ "giou_loss_coefficient": 2,
40
+ "id2label": {
41
+ "0": "grey and white light aircraft",
42
+ "1": "grey fighter jet",
43
+ "2": "grey helicopter",
44
+ "3": "orange and black fighter jet",
45
+ "4": "white commercial aircraft",
46
+ "5": "white and yellow commercial aircraft",
47
+ "6": "white light aircraft",
48
+ "7": "blue and green fighter plane",
49
+ "8": "green fighter plane",
50
+ "9": "white, blue, and red commercial aircraft",
51
+ "10": "white and blue fighter plane",
52
+ "11": "yellow and red light aircraft",
53
+ "12": "grey drone",
54
+ "13": "black camouflage fighter jet",
55
+ "14": "blue and red light aircraft",
56
+ "15": "black and white commercial aircraft",
57
+ "16": "blue commercial aircraft",
58
+ "17": "black cargo aircraft",
59
+ "18": "yellow, black, and red helicopter",
60
+ "19": "white, red, and green fighter plane",
61
+ "20": "white and blue fighter jet",
62
+ "21": "blue and grey fighter jet",
63
+ "22": "green and grey helicopter",
64
+ "23": "blue and red commercial aircraft",
65
+ "24": "yellow fighter plane",
66
+ "25": "yellow and black fighter plane",
67
+ "26": "green and black camouflage helicopter",
68
+ "27": "black helicopter",
69
+ "28": "red light aircraft",
70
+ "29": "green and brown camouflage helicopter",
71
+ "30": "black fighter plane",
72
+ "31": "grey and yellow fighter plane",
73
+ "32": "white and black cargo aircraft",
74
+ "33": "green light aircraft",
75
+ "34": "white and black fighter plane",
76
+ "35": "white and black light aircraft",
77
+ "36": "white drone",
78
+ "37": "yellow light aircraft",
79
+ "38": "white and red light aircraft",
80
+ "39": "yellow and green helicopter",
81
+ "40": "white and black drone",
82
+ "41": "white fighter jet",
83
+ "42": "grey, red, and blue commercial aircraft",
84
+ "43": "white cargo aircraft",
85
+ "44": "white, black, and grey missile",
86
+ "45": "red, white, and blue light aircraft",
87
+ "46": "red and black drone",
88
+ "47": "green and white fighter plane",
89
+ "48": "red fighter plane",
90
+ "49": "grey commercial aircraft",
91
+ "50": "grey and red fighter jet",
92
+ "51": "grey and black fighter plane",
93
+ "52": "grey camouflage fighter jet",
94
+ "53": "green camouflage helicopter",
95
+ "54": "white and red fighter plane",
96
+ "55": "blue and yellow fighter jet",
97
+ "56": "blue and white helicopter",
98
+ "57": "white, black, and red drone",
99
+ "58": "white and blue cargo aircraft",
100
+ "59": "green and yellow fighter plane",
101
+ "60": "red and white helicopter",
102
+ "61": "grey and green cargo aircraft",
103
+ "62": "yellow, red, and grey helicopter",
104
+ "63": "silver fighter plane",
105
+ "64": "blue, yellow, and green fighter plane",
106
+ "65": "red and white fighter plane",
107
+ "66": "grey missile",
108
+ "67": "white and black fighter jet",
109
+ "68": "blue missile",
110
+ "69": "grey light aircraft",
111
+ "70": "red fighter jet",
112
+ "71": "yellow missile",
113
+ "72": "white helicopter",
114
+ "73": "black drone",
115
+ "74": "yellow helicopter",
116
+ "75": "black fighter jet",
117
+ "76": "white and red helicopter",
118
+ "77": "grey and white fighter plane",
119
+ "78": "blue helicopter",
120
+ "79": "white and blue light aircraft",
121
+ "80": "grey and black helicopter",
122
+ "81": "blue and yellow helicopter",
123
+ "82": "white and blue commercial aircraft",
124
+ "83": "blue and white missile",
125
+ "84": "black and brown camouflage helicopter",
126
+ "85": "red and white fighter jet",
127
+ "86": "white and orange commercial aircraft",
128
+ "87": "black and white cargo aircraft",
129
+ "88": "white and red missile",
130
+ "89": "white and red commercial aircraft",
131
+ "90": "white missile",
132
+ "91": "white and black helicopter",
133
+ "92": "yellow, red, and blue fighter plane",
134
+ "93": "red helicopter",
135
+ "94": "black and white missile",
136
+ "95": "green helicopter",
137
+ "96": "green and brown camouflage fighter plane",
138
+ "97": "grey cargo aircraft",
139
+ "98": "green and brown camouflage fighter jet",
140
+ "99": "yellow commercial aircraft",
141
+ "100": "white and grey helicopter",
142
+ "101": "white fighter plane",
143
+ "102": "silver and blue fighter plane",
144
+ "103": "red and white missile",
145
+ "104": "green missile",
146
+ "105": "black and orange drone",
147
+ "106": "grey fighter plane",
148
+ "107": "black and yellow missile",
149
+ "108": "orange light aircraft",
150
+ "109": "white and blue helicopter",
151
+ "110": "red and white light aircraft",
152
+ "111": "grey and red missile",
153
+ "112": "black and yellow drone",
154
+ "113": "white and orange light aircraft",
155
+ "114": "blue, yellow, and white cargo aircraft",
156
+ "115": "yellow fighter jet",
157
+ "116": "blue camouflage fighter jet",
158
+ "117": "green and black missile",
159
+ "118": "blue, yellow, and black helicopter",
160
+ "119": "white and red fighter jet",
161
+ "120": "grey and red commercial aircraft",
162
+ "121": "red, white, and blue fighter jet",
163
+ "122": "white, red, and blue commercial aircraft",
164
+ "123": "blue and white commercial aircraft",
165
+ "124": "blue and white light aircraft",
166
+ "125": "red and grey missile"
167
+ },
168
+ "init_std": 0.02,
169
+ "init_xavier_std": 1.0,
170
+ "is_encoder_decoder": true,
171
+ "label2id": {
172
+ "black and brown camouflage helicopter": 84,
173
+ "black and orange drone": 105,
174
+ "black and white cargo aircraft": 87,
175
+ "black and white commercial aircraft": 15,
176
+ "black and white missile": 94,
177
+ "black and yellow drone": 112,
178
+ "black and yellow missile": 107,
179
+ "black camouflage fighter jet": 13,
180
+ "black cargo aircraft": 17,
181
+ "black drone": 73,
182
+ "black fighter jet": 75,
183
+ "black fighter plane": 30,
184
+ "black helicopter": 27,
185
+ "blue and green fighter plane": 7,
186
+ "blue and grey fighter jet": 21,
187
+ "blue and red commercial aircraft": 23,
188
+ "blue and red light aircraft": 14,
189
+ "blue and white commercial aircraft": 123,
190
+ "blue and white helicopter": 56,
191
+ "blue and white light aircraft": 124,
192
+ "blue and white missile": 83,
193
+ "blue and yellow fighter jet": 55,
194
+ "blue and yellow helicopter": 81,
195
+ "blue camouflage fighter jet": 116,
196
+ "blue commercial aircraft": 16,
197
+ "blue helicopter": 78,
198
+ "blue missile": 68,
199
+ "blue, yellow, and black helicopter": 118,
200
+ "blue, yellow, and green fighter plane": 64,
201
+ "blue, yellow, and white cargo aircraft": 114,
202
+ "green and black camouflage helicopter": 26,
203
+ "green and black missile": 117,
204
+ "green and brown camouflage fighter jet": 98,
205
+ "green and brown camouflage fighter plane": 96,
206
+ "green and brown camouflage helicopter": 29,
207
+ "green and grey helicopter": 22,
208
+ "green and white fighter plane": 47,
209
+ "green and yellow fighter plane": 59,
210
+ "green camouflage helicopter": 53,
211
+ "green fighter plane": 8,
212
+ "green helicopter": 95,
213
+ "green light aircraft": 33,
214
+ "green missile": 104,
215
+ "grey and black fighter plane": 51,
216
+ "grey and black helicopter": 80,
217
+ "grey and green cargo aircraft": 61,
218
+ "grey and red commercial aircraft": 120,
219
+ "grey and red fighter jet": 50,
220
+ "grey and red missile": 111,
221
+ "grey and white fighter plane": 77,
222
+ "grey and white light aircraft": 0,
223
+ "grey and yellow fighter plane": 31,
224
+ "grey camouflage fighter jet": 52,
225
+ "grey cargo aircraft": 97,
226
+ "grey commercial aircraft": 49,
227
+ "grey drone": 12,
228
+ "grey fighter jet": 1,
229
+ "grey fighter plane": 106,
230
+ "grey helicopter": 2,
231
+ "grey light aircraft": 69,
232
+ "grey missile": 66,
233
+ "grey, red, and blue commercial aircraft": 42,
234
+ "orange and black fighter jet": 3,
235
+ "orange light aircraft": 108,
236
+ "red and black drone": 46,
237
+ "red and grey missile": 125,
238
+ "red and white fighter jet": 85,
239
+ "red and white fighter plane": 65,
240
+ "red and white helicopter": 60,
241
+ "red and white light aircraft": 110,
242
+ "red and white missile": 103,
243
+ "red fighter jet": 70,
244
+ "red fighter plane": 48,
245
+ "red helicopter": 93,
246
+ "red light aircraft": 28,
247
+ "red, white, and blue fighter jet": 121,
248
+ "red, white, and blue light aircraft": 45,
249
+ "silver and blue fighter plane": 102,
250
+ "silver fighter plane": 63,
251
+ "white and black cargo aircraft": 32,
252
+ "white and black drone": 40,
253
+ "white and black fighter jet": 67,
254
+ "white and black fighter plane": 34,
255
+ "white and black helicopter": 91,
256
+ "white and black light aircraft": 35,
257
+ "white and blue cargo aircraft": 58,
258
+ "white and blue commercial aircraft": 82,
259
+ "white and blue fighter jet": 20,
260
+ "white and blue fighter plane": 10,
261
+ "white and blue helicopter": 109,
262
+ "white and blue light aircraft": 79,
263
+ "white and grey helicopter": 100,
264
+ "white and orange commercial aircraft": 86,
265
+ "white and orange light aircraft": 113,
266
+ "white and red commercial aircraft": 89,
267
+ "white and red fighter jet": 119,
268
+ "white and red fighter plane": 54,
269
+ "white and red helicopter": 76,
270
+ "white and red light aircraft": 38,
271
+ "white and red missile": 88,
272
+ "white and yellow commercial aircraft": 5,
273
+ "white cargo aircraft": 43,
274
+ "white commercial aircraft": 4,
275
+ "white drone": 36,
276
+ "white fighter jet": 41,
277
+ "white fighter plane": 101,
278
+ "white helicopter": 72,
279
+ "white light aircraft": 6,
280
+ "white missile": 90,
281
+ "white, black, and grey missile": 44,
282
+ "white, black, and red drone": 57,
283
+ "white, blue, and red commercial aircraft": 9,
284
+ "white, red, and blue commercial aircraft": 122,
285
+ "white, red, and green fighter plane": 19,
286
+ "yellow and black fighter plane": 25,
287
+ "yellow and green helicopter": 39,
288
+ "yellow and red light aircraft": 11,
289
+ "yellow commercial aircraft": 99,
290
+ "yellow fighter jet": 115,
291
+ "yellow fighter plane": 24,
292
+ "yellow helicopter": 74,
293
+ "yellow light aircraft": 37,
294
+ "yellow missile": 71,
295
+ "yellow, black, and red helicopter": 18,
296
+ "yellow, red, and blue fighter plane": 92,
297
+ "yellow, red, and grey helicopter": 62
298
+ },
299
+ "mask_loss_coefficient": 1,
300
+ "max_position_embeddings": 1024,
301
+ "model_type": "detr",
302
+ "num_channels": 3,
303
+ "num_hidden_layers": 6,
304
+ "num_queries": 100,
305
+ "position_embedding_type": "sine",
306
+ "scale_embedding": false,
307
+ "torch_dtype": "float32",
308
+ "transformers_version": "4.41.2",
309
+ "use_pretrained_backbone": true,
310
+ "use_timm_backbone": true
311
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2d9280137b59b6266f9a6e28b0d9e3191f395a4d36f5bfb0b0d33e0748e1f4f7
3
+ size 166623364
preprocessor_config.json ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_valid_processor_keys": [
3
+ "images",
4
+ "annotations",
5
+ "return_segmentation_masks",
6
+ "masks_path",
7
+ "do_resize",
8
+ "size",
9
+ "resample",
10
+ "do_rescale",
11
+ "rescale_factor",
12
+ "do_normalize",
13
+ "do_convert_annotations",
14
+ "image_mean",
15
+ "image_std",
16
+ "do_pad",
17
+ "format",
18
+ "return_tensors",
19
+ "data_format",
20
+ "input_data_format"
21
+ ],
22
+ "do_convert_annotations": true,
23
+ "do_normalize": true,
24
+ "do_pad": true,
25
+ "do_rescale": true,
26
+ "do_resize": true,
27
+ "format": "coco_detection",
28
+ "image_mean": [
29
+ 0.485,
30
+ 0.456,
31
+ 0.406
32
+ ],
33
+ "image_processor_type": "DetrImageProcessor",
34
+ "image_std": [
35
+ 0.229,
36
+ 0.224,
37
+ 0.225
38
+ ],
39
+ "resample": 2,
40
+ "rescale_factor": 0.00392156862745098,
41
+ "size": {
42
+ "longest_edge": 1333,
43
+ "shortest_edge": 800
44
+ }
45
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce9dc6108f522c7192b28bf05118f6fb0a50bd962b1bd7fc0255c1910092a221
3
+ size 5112