stevenbucaille commited on
Commit
65d4129
·
verified ·
1 Parent(s): 6d1bc6d

Add lwdetr_large_60e_coco model

Browse files
Files changed (2) hide show
  1. config.json +184 -183
  2. model.safetensors +2 -2
config.json CHANGED
@@ -65,6 +65,7 @@
65
  "batch_norm_eps": 1e-05,
66
  "bbox_cost": 5,
67
  "bbox_loss_coefficient": 5,
 
68
  "class_cost": 2,
69
  "d_model": 384,
70
  "decoder_activation_function": "relu",
@@ -84,191 +85,191 @@
84
  "group_detr": 13,
85
  "hidden_expansion": 0.5,
86
  "id2label": {
87
- "0": "N/A",
88
- "1": "person",
89
- "10": "traffic light",
90
- "11": "fire hydrant",
91
- "12": "street sign",
92
- "13": "stop sign",
93
- "14": "parking meter",
94
- "15": "bench",
95
- "16": "bird",
96
- "17": "cat",
97
- "18": "dog",
98
- "19": "horse",
99
- "2": "bicycle",
100
- "20": "sheep",
101
- "21": "cow",
102
- "22": "elephant",
103
- "23": "bear",
104
- "24": "zebra",
105
- "25": "giraffe",
106
- "26": "hat",
107
- "27": "backpack",
108
- "28": "umbrella",
109
- "29": "shoe",
110
- "3": "car",
111
- "30": "eye glasses",
112
- "31": "handbag",
113
- "32": "tie",
114
- "33": "suitcase",
115
- "34": "frisbee",
116
- "35": "skis",
117
- "36": "snowboard",
118
- "37": "sports ball",
119
- "38": "kite",
120
- "39": "baseball bat",
121
- "4": "motorcycle",
122
- "40": "baseball glove",
123
- "41": "skateboard",
124
- "42": "surfboard",
125
- "43": "tennis racket",
126
- "44": "bottle",
127
- "45": "plate",
128
- "46": "wine glass",
129
- "47": "cup",
130
- "48": "fork",
131
- "49": "knife",
132
- "5": "airplane",
133
- "50": "spoon",
134
- "51": "bowl",
135
- "52": "banana",
136
- "53": "apple",
137
- "54": "sandwich",
138
- "55": "orange",
139
- "56": "broccoli",
140
- "57": "carrot",
141
- "58": "hot dog",
142
- "59": "pizza",
143
- "6": "bus",
144
- "60": "donut",
145
- "61": "cake",
146
- "62": "chair",
147
- "63": "couch",
148
- "64": "potted plant",
149
- "65": "bed",
150
- "66": "mirror",
151
- "67": "dining table",
152
- "68": "window",
153
- "69": "desk",
154
- "7": "train",
155
- "70": "toilet",
156
- "71": "door",
157
- "72": "tv",
158
- "73": "laptop",
159
- "74": "mouse",
160
- "75": "remote",
161
- "76": "keyboard",
162
- "77": "cell phone",
163
- "78": "microwave",
164
- "79": "oven",
165
- "8": "truck",
166
- "80": "toaster",
167
- "81": "sink",
168
- "82": "refrigerator",
169
- "83": "blender",
170
- "84": "book",
171
- "85": "clock",
172
- "86": "vase",
173
- "87": "scissors",
174
- "88": "teddy bear",
175
- "89": "hair drier",
176
- "9": "boat",
177
- "90": "toothbrush"
178
  },
179
  "init_std": 0.02,
180
  "label2id": {
181
- "N/A": 0,
182
- "airplane": 5,
183
- "apple": 53,
184
- "backpack": 27,
185
- "banana": 52,
186
- "baseball bat": 39,
187
- "baseball glove": 40,
188
- "bear": 23,
189
- "bed": 65,
190
- "bench": 15,
191
- "bicycle": 2,
192
- "bird": 16,
193
- "blender": 83,
194
- "boat": 9,
195
- "book": 84,
196
- "bottle": 44,
197
- "bowl": 51,
198
- "broccoli": 56,
199
- "bus": 6,
200
- "cake": 61,
201
- "car": 3,
202
- "carrot": 57,
203
- "cat": 17,
204
- "cell phone": 77,
205
- "chair": 62,
206
- "clock": 85,
207
- "couch": 63,
208
- "cow": 21,
209
- "cup": 47,
210
- "desk": 69,
211
- "dining table": 67,
212
- "dog": 18,
213
- "donut": 60,
214
- "door": 71,
215
- "elephant": 22,
216
- "eye glasses": 30,
217
- "fire hydrant": 11,
218
- "fork": 48,
219
- "frisbee": 34,
220
- "giraffe": 25,
221
- "hair drier": 89,
222
- "handbag": 31,
223
- "hat": 26,
224
- "horse": 19,
225
- "hot dog": 58,
226
- "keyboard": 76,
227
- "kite": 38,
228
- "knife": 49,
229
- "laptop": 73,
230
- "microwave": 78,
231
- "mirror": 66,
232
- "motorcycle": 4,
233
- "mouse": 74,
234
- "orange": 55,
235
- "oven": 79,
236
- "parking meter": 14,
237
- "person": 1,
238
- "pizza": 59,
239
- "plate": 45,
240
- "potted plant": 64,
241
- "refrigerator": 82,
242
- "remote": 75,
243
- "sandwich": 54,
244
- "scissors": 87,
245
- "sheep": 20,
246
- "shoe": 29,
247
- "sink": 81,
248
- "skateboard": 41,
249
- "skis": 35,
250
- "snowboard": 36,
251
- "spoon": 50,
252
- "sports ball": 37,
253
- "stop sign": 13,
254
- "street sign": 12,
255
- "suitcase": 33,
256
- "surfboard": 42,
257
- "teddy bear": 88,
258
- "tennis racket": 43,
259
- "tie": 32,
260
- "toaster": 80,
261
- "toilet": 70,
262
- "toothbrush": 90,
263
- "traffic light": 10,
264
- "train": 7,
265
- "truck": 8,
266
- "tv": 72,
267
- "umbrella": 28,
268
- "vase": 86,
269
- "window": 68,
270
- "wine glass": 46,
271
- "zebra": 24
272
  },
273
  "model_type": "lw_detr",
274
  "num_feature_levels": 2,
@@ -285,4 +286,4 @@
285
  "transformers_version": "5.0.0.dev0",
286
  "use_pretrained_backbone": false,
287
  "use_timm_backbone": false
288
- }
 
65
  "batch_norm_eps": 1e-05,
66
  "bbox_cost": 5,
67
  "bbox_loss_coefficient": 5,
68
+ "c2f_num_blocks": 3,
69
  "class_cost": 2,
70
  "d_model": 384,
71
  "decoder_activation_function": "relu",
 
85
  "group_detr": 13,
86
  "hidden_expansion": 0.5,
87
  "id2label": {
88
+ "0": "LABEL_0",
89
+ "1": "LABEL_1",
90
+ "2": "LABEL_2",
91
+ "3": "LABEL_3",
92
+ "4": "LABEL_4",
93
+ "5": "LABEL_5",
94
+ "6": "LABEL_6",
95
+ "7": "LABEL_7",
96
+ "8": "LABEL_8",
97
+ "9": "LABEL_9",
98
+ "10": "LABEL_10",
99
+ "11": "LABEL_11",
100
+ "12": "LABEL_12",
101
+ "13": "LABEL_13",
102
+ "14": "LABEL_14",
103
+ "15": "LABEL_15",
104
+ "16": "LABEL_16",
105
+ "17": "LABEL_17",
106
+ "18": "LABEL_18",
107
+ "19": "LABEL_19",
108
+ "20": "LABEL_20",
109
+ "21": "LABEL_21",
110
+ "22": "LABEL_22",
111
+ "23": "LABEL_23",
112
+ "24": "LABEL_24",
113
+ "25": "LABEL_25",
114
+ "26": "LABEL_26",
115
+ "27": "LABEL_27",
116
+ "28": "LABEL_28",
117
+ "29": "LABEL_29",
118
+ "30": "LABEL_30",
119
+ "31": "LABEL_31",
120
+ "32": "LABEL_32",
121
+ "33": "LABEL_33",
122
+ "34": "LABEL_34",
123
+ "35": "LABEL_35",
124
+ "36": "LABEL_36",
125
+ "37": "LABEL_37",
126
+ "38": "LABEL_38",
127
+ "39": "LABEL_39",
128
+ "40": "LABEL_40",
129
+ "41": "LABEL_41",
130
+ "42": "LABEL_42",
131
+ "43": "LABEL_43",
132
+ "44": "LABEL_44",
133
+ "45": "LABEL_45",
134
+ "46": "LABEL_46",
135
+ "47": "LABEL_47",
136
+ "48": "LABEL_48",
137
+ "49": "LABEL_49",
138
+ "50": "LABEL_50",
139
+ "51": "LABEL_51",
140
+ "52": "LABEL_52",
141
+ "53": "LABEL_53",
142
+ "54": "LABEL_54",
143
+ "55": "LABEL_55",
144
+ "56": "LABEL_56",
145
+ "57": "LABEL_57",
146
+ "58": "LABEL_58",
147
+ "59": "LABEL_59",
148
+ "60": "LABEL_60",
149
+ "61": "LABEL_61",
150
+ "62": "LABEL_62",
151
+ "63": "LABEL_63",
152
+ "64": "LABEL_64",
153
+ "65": "LABEL_65",
154
+ "66": "LABEL_66",
155
+ "67": "LABEL_67",
156
+ "68": "LABEL_68",
157
+ "69": "LABEL_69",
158
+ "70": "LABEL_70",
159
+ "71": "LABEL_71",
160
+ "72": "LABEL_72",
161
+ "73": "LABEL_73",
162
+ "74": "LABEL_74",
163
+ "75": "LABEL_75",
164
+ "76": "LABEL_76",
165
+ "77": "LABEL_77",
166
+ "78": "LABEL_78",
167
+ "79": "LABEL_79",
168
+ "80": "LABEL_80",
169
+ "81": "LABEL_81",
170
+ "82": "LABEL_82",
171
+ "83": "LABEL_83",
172
+ "84": "LABEL_84",
173
+ "85": "LABEL_85",
174
+ "86": "LABEL_86",
175
+ "87": "LABEL_87",
176
+ "88": "LABEL_88",
177
+ "89": "LABEL_89",
178
+ "90": "LABEL_90"
179
  },
180
  "init_std": 0.02,
181
  "label2id": {
182
+ "LABEL_0": 0,
183
+ "LABEL_1": 1,
184
+ "LABEL_10": 10,
185
+ "LABEL_11": 11,
186
+ "LABEL_12": 12,
187
+ "LABEL_13": 13,
188
+ "LABEL_14": 14,
189
+ "LABEL_15": 15,
190
+ "LABEL_16": 16,
191
+ "LABEL_17": 17,
192
+ "LABEL_18": 18,
193
+ "LABEL_19": 19,
194
+ "LABEL_2": 2,
195
+ "LABEL_20": 20,
196
+ "LABEL_21": 21,
197
+ "LABEL_22": 22,
198
+ "LABEL_23": 23,
199
+ "LABEL_24": 24,
200
+ "LABEL_25": 25,
201
+ "LABEL_26": 26,
202
+ "LABEL_27": 27,
203
+ "LABEL_28": 28,
204
+ "LABEL_29": 29,
205
+ "LABEL_3": 3,
206
+ "LABEL_30": 30,
207
+ "LABEL_31": 31,
208
+ "LABEL_32": 32,
209
+ "LABEL_33": 33,
210
+ "LABEL_34": 34,
211
+ "LABEL_35": 35,
212
+ "LABEL_36": 36,
213
+ "LABEL_37": 37,
214
+ "LABEL_38": 38,
215
+ "LABEL_39": 39,
216
+ "LABEL_4": 4,
217
+ "LABEL_40": 40,
218
+ "LABEL_41": 41,
219
+ "LABEL_42": 42,
220
+ "LABEL_43": 43,
221
+ "LABEL_44": 44,
222
+ "LABEL_45": 45,
223
+ "LABEL_46": 46,
224
+ "LABEL_47": 47,
225
+ "LABEL_48": 48,
226
+ "LABEL_49": 49,
227
+ "LABEL_5": 5,
228
+ "LABEL_50": 50,
229
+ "LABEL_51": 51,
230
+ "LABEL_52": 52,
231
+ "LABEL_53": 53,
232
+ "LABEL_54": 54,
233
+ "LABEL_55": 55,
234
+ "LABEL_56": 56,
235
+ "LABEL_57": 57,
236
+ "LABEL_58": 58,
237
+ "LABEL_59": 59,
238
+ "LABEL_6": 6,
239
+ "LABEL_60": 60,
240
+ "LABEL_61": 61,
241
+ "LABEL_62": 62,
242
+ "LABEL_63": 63,
243
+ "LABEL_64": 64,
244
+ "LABEL_65": 65,
245
+ "LABEL_66": 66,
246
+ "LABEL_67": 67,
247
+ "LABEL_68": 68,
248
+ "LABEL_69": 69,
249
+ "LABEL_7": 7,
250
+ "LABEL_70": 70,
251
+ "LABEL_71": 71,
252
+ "LABEL_72": 72,
253
+ "LABEL_73": 73,
254
+ "LABEL_74": 74,
255
+ "LABEL_75": 75,
256
+ "LABEL_76": 76,
257
+ "LABEL_77": 77,
258
+ "LABEL_78": 78,
259
+ "LABEL_79": 79,
260
+ "LABEL_8": 8,
261
+ "LABEL_80": 80,
262
+ "LABEL_81": 81,
263
+ "LABEL_82": 82,
264
+ "LABEL_83": 83,
265
+ "LABEL_84": 84,
266
+ "LABEL_85": 85,
267
+ "LABEL_86": 86,
268
+ "LABEL_87": 87,
269
+ "LABEL_88": 88,
270
+ "LABEL_89": 89,
271
+ "LABEL_9": 9,
272
+ "LABEL_90": 90
273
  },
274
  "model_type": "lw_detr",
275
  "num_feature_levels": 2,
 
286
  "transformers_version": "5.0.0.dev0",
287
  "use_pretrained_backbone": false,
288
  "use_timm_backbone": false
289
+ }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:016454aa0f18e6507afead19cb5772a2328c26d18093018f18204a1560bd91ad
3
- size 187407528
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c9d4a1066e29df42e9bbe81cdaf0cd877c0d500e90c50d5cfbbf606f23f4ac5c
3
+ size 187408048