nielsr HF staff commited on
Commit
b4b0f10
1 Parent(s): f6a5daf

Upload DetaForObjectDetection

Browse files
Files changed (2) hide show
  1. config.json +316 -0
  2. pytorch_model.bin +3 -0
config.json ADDED
@@ -0,0 +1,316 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_commit_hash": null,
3
+ "activation_dropout": 0.0,
4
+ "activation_function": "relu",
5
+ "architectures": [
6
+ "DetaForObjectDetection"
7
+ ],
8
+ "assign_first_stage": true,
9
+ "attention_dropout": 0.0,
10
+ "auxiliary_loss": false,
11
+ "backbone": "resnet50",
12
+ "backbone_config": {
13
+ "_name_or_path": "",
14
+ "add_cross_attention": false,
15
+ "architectures": null,
16
+ "bad_words_ids": null,
17
+ "begin_suppress_tokens": null,
18
+ "bos_token_id": null,
19
+ "chunk_size_feed_forward": 0,
20
+ "cross_attention_hidden_size": null,
21
+ "decoder_start_token_id": null,
22
+ "depths": [
23
+ 3,
24
+ 4,
25
+ 6,
26
+ 3
27
+ ],
28
+ "diversity_penalty": 0.0,
29
+ "do_sample": false,
30
+ "downsample_in_first_stage": false,
31
+ "early_stopping": false,
32
+ "embedding_size": 64,
33
+ "encoder_no_repeat_ngram_size": 0,
34
+ "eos_token_id": null,
35
+ "exponential_decay_length_penalty": null,
36
+ "finetuning_task": null,
37
+ "forced_bos_token_id": null,
38
+ "forced_eos_token_id": null,
39
+ "hidden_act": "relu",
40
+ "hidden_sizes": [
41
+ 256,
42
+ 512,
43
+ 1024,
44
+ 2048
45
+ ],
46
+ "id2label": {
47
+ "0": "LABEL_0",
48
+ "1": "LABEL_1"
49
+ },
50
+ "is_decoder": false,
51
+ "is_encoder_decoder": false,
52
+ "label2id": {
53
+ "LABEL_0": 0,
54
+ "LABEL_1": 1
55
+ },
56
+ "layer_type": "bottleneck",
57
+ "length_penalty": 1.0,
58
+ "max_length": 20,
59
+ "min_length": 0,
60
+ "model_type": "resnet",
61
+ "no_repeat_ngram_size": 0,
62
+ "num_beam_groups": 1,
63
+ "num_beams": 1,
64
+ "num_channels": 3,
65
+ "num_return_sequences": 1,
66
+ "out_features": [
67
+ "stage2",
68
+ "stage3",
69
+ "stage4"
70
+ ],
71
+ "output_attentions": false,
72
+ "output_hidden_states": false,
73
+ "output_scores": false,
74
+ "pad_token_id": null,
75
+ "prefix": null,
76
+ "problem_type": null,
77
+ "pruned_heads": {},
78
+ "remove_invalid_values": false,
79
+ "repetition_penalty": 1.0,
80
+ "return_dict": true,
81
+ "return_dict_in_generate": false,
82
+ "sep_token_id": null,
83
+ "stage_names": [
84
+ "stem",
85
+ "stage1",
86
+ "stage2",
87
+ "stage3",
88
+ "stage4"
89
+ ],
90
+ "suppress_tokens": null,
91
+ "task_specific_params": null,
92
+ "temperature": 1.0,
93
+ "tf_legacy_loss": false,
94
+ "tie_encoder_decoder": false,
95
+ "tie_word_embeddings": true,
96
+ "tokenizer_class": null,
97
+ "top_k": 50,
98
+ "top_p": 1.0,
99
+ "torch_dtype": null,
100
+ "torchscript": false,
101
+ "transformers_version": "4.26.0.dev0",
102
+ "typical_p": 1.0,
103
+ "use_bfloat16": false
104
+ },
105
+ "bbox_cost": 5,
106
+ "bbox_loss_coefficient": 5,
107
+ "class_cost": 1,
108
+ "d_model": 256,
109
+ "decoder_attention_heads": 8,
110
+ "decoder_ffn_dim": 2048,
111
+ "decoder_layerdrop": 0.0,
112
+ "decoder_layers": 6,
113
+ "decoder_n_points": 4,
114
+ "dice_loss_coefficient": 1,
115
+ "dilation": false,
116
+ "dropout": 0.1,
117
+ "encoder_attention_heads": 8,
118
+ "encoder_ffn_dim": 2048,
119
+ "encoder_layerdrop": 0.0,
120
+ "encoder_layers": 6,
121
+ "encoder_n_points": 4,
122
+ "eos_coefficient": 0.1,
123
+ "focal_alpha": 0.25,
124
+ "giou_cost": 2,
125
+ "giou_loss_coefficient": 2,
126
+ "id2label": {
127
+ "0": "N/A",
128
+ "1": "person",
129
+ "2": "bicycle",
130
+ "3": "car",
131
+ "4": "motorcycle",
132
+ "5": "airplane",
133
+ "6": "bus",
134
+ "7": "train",
135
+ "8": "truck",
136
+ "9": "boat",
137
+ "10": "traffic light",
138
+ "11": "fire hydrant",
139
+ "12": "N/A",
140
+ "13": "stop sign",
141
+ "14": "parking meter",
142
+ "15": "bench",
143
+ "16": "bird",
144
+ "17": "cat",
145
+ "18": "dog",
146
+ "19": "horse",
147
+ "20": "sheep",
148
+ "21": "cow",
149
+ "22": "elephant",
150
+ "23": "bear",
151
+ "24": "zebra",
152
+ "25": "giraffe",
153
+ "26": "N/A",
154
+ "27": "backpack",
155
+ "28": "umbrella",
156
+ "29": "N/A",
157
+ "30": "N/A",
158
+ "31": "handbag",
159
+ "32": "tie",
160
+ "33": "suitcase",
161
+ "34": "frisbee",
162
+ "35": "skis",
163
+ "36": "snowboard",
164
+ "37": "sports ball",
165
+ "38": "kite",
166
+ "39": "baseball bat",
167
+ "40": "baseball glove",
168
+ "41": "skateboard",
169
+ "42": "surfboard",
170
+ "43": "tennis racket",
171
+ "44": "bottle",
172
+ "45": "N/A",
173
+ "46": "wine glass",
174
+ "47": "cup",
175
+ "48": "fork",
176
+ "49": "knife",
177
+ "50": "spoon",
178
+ "51": "bowl",
179
+ "52": "banana",
180
+ "53": "apple",
181
+ "54": "sandwich",
182
+ "55": "orange",
183
+ "56": "broccoli",
184
+ "57": "carrot",
185
+ "58": "hot dog",
186
+ "59": "pizza",
187
+ "60": "donut",
188
+ "61": "cake",
189
+ "62": "chair",
190
+ "63": "couch",
191
+ "64": "potted plant",
192
+ "65": "bed",
193
+ "66": "N/A",
194
+ "67": "dining table",
195
+ "68": "N/A",
196
+ "69": "N/A",
197
+ "70": "toilet",
198
+ "71": "N/A",
199
+ "72": "tv",
200
+ "73": "laptop",
201
+ "74": "mouse",
202
+ "75": "remote",
203
+ "76": "keyboard",
204
+ "77": "cell phone",
205
+ "78": "microwave",
206
+ "79": "oven",
207
+ "80": "toaster",
208
+ "81": "sink",
209
+ "82": "refrigerator",
210
+ "83": "N/A",
211
+ "84": "book",
212
+ "85": "clock",
213
+ "86": "vase",
214
+ "87": "scissors",
215
+ "88": "teddy bear",
216
+ "89": "hair drier",
217
+ "90": "toothbrush"
218
+ },
219
+ "init_std": 0.02,
220
+ "init_xavier_std": 1.0,
221
+ "is_encoder_decoder": true,
222
+ "label2id": {
223
+ "N/A": 83,
224
+ "airplane": 5,
225
+ "apple": 53,
226
+ "backpack": 27,
227
+ "banana": 52,
228
+ "baseball bat": 39,
229
+ "baseball glove": 40,
230
+ "bear": 23,
231
+ "bed": 65,
232
+ "bench": 15,
233
+ "bicycle": 2,
234
+ "bird": 16,
235
+ "boat": 9,
236
+ "book": 84,
237
+ "bottle": 44,
238
+ "bowl": 51,
239
+ "broccoli": 56,
240
+ "bus": 6,
241
+ "cake": 61,
242
+ "car": 3,
243
+ "carrot": 57,
244
+ "cat": 17,
245
+ "cell phone": 77,
246
+ "chair": 62,
247
+ "clock": 85,
248
+ "couch": 63,
249
+ "cow": 21,
250
+ "cup": 47,
251
+ "dining table": 67,
252
+ "dog": 18,
253
+ "donut": 60,
254
+ "elephant": 22,
255
+ "fire hydrant": 11,
256
+ "fork": 48,
257
+ "frisbee": 34,
258
+ "giraffe": 25,
259
+ "hair drier": 89,
260
+ "handbag": 31,
261
+ "horse": 19,
262
+ "hot dog": 58,
263
+ "keyboard": 76,
264
+ "kite": 38,
265
+ "knife": 49,
266
+ "laptop": 73,
267
+ "microwave": 78,
268
+ "motorcycle": 4,
269
+ "mouse": 74,
270
+ "orange": 55,
271
+ "oven": 79,
272
+ "parking meter": 14,
273
+ "person": 1,
274
+ "pizza": 59,
275
+ "potted plant": 64,
276
+ "refrigerator": 82,
277
+ "remote": 75,
278
+ "sandwich": 54,
279
+ "scissors": 87,
280
+ "sheep": 20,
281
+ "sink": 81,
282
+ "skateboard": 41,
283
+ "skis": 35,
284
+ "snowboard": 36,
285
+ "spoon": 50,
286
+ "sports ball": 37,
287
+ "stop sign": 13,
288
+ "suitcase": 33,
289
+ "surfboard": 42,
290
+ "teddy bear": 88,
291
+ "tennis racket": 43,
292
+ "tie": 32,
293
+ "toaster": 80,
294
+ "toilet": 70,
295
+ "toothbrush": 90,
296
+ "traffic light": 10,
297
+ "train": 7,
298
+ "truck": 8,
299
+ "tv": 72,
300
+ "umbrella": 28,
301
+ "vase": 86,
302
+ "wine glass": 46,
303
+ "zebra": 24
304
+ },
305
+ "mask_loss_coefficient": 1,
306
+ "max_position_embeddings": 1024,
307
+ "model_type": "deta",
308
+ "num_feature_levels": 5,
309
+ "num_queries": 900,
310
+ "position_embedding_type": "sine",
311
+ "torch_dtype": "float32",
312
+ "transformers_version": null,
313
+ "two_stage": true,
314
+ "two_stage_num_proposals": 300,
315
+ "with_box_refine": true
316
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fce0009f21714d10c8830d1a2e7e2dd0d17754cc3fce1ad13902fb2d03a0480b
3
+ size 194414363