hussamturjman commited on
Commit
050f23e
1 Parent(s): abb1a46

Upload DetrForObjectDetection

Browse files
Files changed (2) hide show
  1. config.json +236 -183
  2. pytorch_model.bin +2 -2
config.json CHANGED
@@ -1,5 +1,4 @@
1
  {
2
- "_name_or_path": "facebook/detr-resnet-50",
3
  "activation_dropout": 0.0,
4
  "activation_function": "relu",
5
  "architectures": [
@@ -27,193 +26,247 @@
27
  "giou_cost": 2,
28
  "giou_loss_coefficient": 2,
29
  "id2label": {
30
- "0": "N/A",
31
- "1": "person",
32
- "2": "bicycle",
33
- "3": "car",
34
- "4": "motorcycle",
35
- "5": "airplane",
36
- "6": "bus",
37
- "7": "train",
38
- "8": "truck",
39
- "9": "boat",
40
- "10": "traffic light",
41
- "11": "fire hydrant",
42
- "12": "street sign",
43
- "13": "stop sign",
44
- "14": "parking meter",
45
- "15": "bench",
46
- "16": "bird",
47
- "17": "cat",
48
- "18": "dog",
49
- "19": "horse",
50
- "20": "sheep",
51
- "21": "cow",
52
- "22": "elephant",
53
- "23": "bear",
54
- "24": "zebra",
55
- "25": "giraffe",
56
- "26": "hat",
57
- "27": "backpack",
58
- "28": "umbrella",
59
- "29": "shoe",
60
- "30": "eye glasses",
61
- "31": "handbag",
62
- "32": "tie",
63
- "33": "suitcase",
64
- "34": "frisbee",
65
- "35": "skis",
66
- "36": "snowboard",
67
- "37": "sports ball",
68
- "38": "kite",
69
- "39": "baseball bat",
70
- "40": "baseball glove",
71
- "41": "skateboard",
72
- "42": "surfboard",
73
- "43": "tennis racket",
74
- "44": "bottle",
75
- "45": "plate",
76
- "46": "wine glass",
77
- "47": "cup",
78
- "48": "fork",
79
- "49": "knife",
80
- "50": "spoon",
81
- "51": "bowl",
82
- "52": "banana",
83
- "53": "apple",
84
- "54": "sandwich",
85
- "55": "orange",
86
- "56": "broccoli",
87
- "57": "carrot",
88
- "58": "hot dog",
89
- "59": "pizza",
90
- "60": "donut",
91
- "61": "cake",
92
- "62": "chair",
93
- "63": "couch",
94
- "64": "potted plant",
95
- "65": "bed",
96
- "66": "mirror",
97
- "67": "dining table",
98
- "68": "window",
99
- "69": "desk",
100
- "70": "toilet",
101
- "71": "door",
102
- "72": "tv",
103
- "73": "laptop",
104
- "74": "mouse",
105
- "75": "remote",
106
- "76": "keyboard",
107
- "77": "cell phone",
108
- "78": "microwave",
109
- "79": "oven",
110
- "80": "toaster",
111
- "81": "sink",
112
- "82": "refrigerator",
113
- "83": "blender",
114
- "84": "book",
115
- "85": "clock",
116
- "86": "vase",
117
- "87": "scissors",
118
- "88": "teddy bear",
119
- "89": "hair drier",
120
- "90": "toothbrush"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
121
  },
122
  "init_std": 0.02,
123
  "init_xavier_std": 1.0,
124
  "is_encoder_decoder": true,
125
  "label2id": {
126
- "N/A": 0,
127
- "airplane": 5,
128
- "apple": 53,
129
- "backpack": 27,
130
- "banana": 52,
131
- "baseball bat": 39,
132
- "baseball glove": 40,
133
- "bear": 23,
134
- "bed": 65,
135
- "bench": 15,
136
- "bicycle": 2,
137
- "bird": 16,
138
- "blender": 83,
139
- "boat": 9,
140
- "book": 84,
141
- "bottle": 44,
142
- "bowl": 51,
143
- "broccoli": 56,
144
- "bus": 6,
145
- "cake": 61,
146
- "car": 3,
147
- "carrot": 57,
148
- "cat": 17,
149
- "cell phone": 77,
150
- "chair": 62,
151
- "clock": 85,
152
- "couch": 63,
153
- "cow": 21,
154
- "cup": 47,
155
- "desk": 69,
156
- "dining table": 67,
157
- "dog": 18,
158
- "donut": 60,
159
- "door": 71,
160
- "elephant": 22,
161
- "eye glasses": 30,
162
- "fire hydrant": 11,
163
- "fork": 48,
164
- "frisbee": 34,
165
- "giraffe": 25,
166
- "hair drier": 89,
167
- "handbag": 31,
168
- "hat": 26,
169
- "horse": 19,
170
- "hot dog": 58,
171
- "keyboard": 76,
172
- "kite": 38,
173
- "knife": 49,
174
- "laptop": 73,
175
- "microwave": 78,
176
- "mirror": 66,
177
- "motorcycle": 4,
178
- "mouse": 74,
179
- "orange": 55,
180
- "oven": 79,
181
- "parking meter": 14,
182
- "person": 1,
183
- "pizza": 59,
184
- "plate": 45,
185
- "potted plant": 64,
186
- "refrigerator": 82,
187
- "remote": 75,
188
- "sandwich": 54,
189
- "scissors": 87,
190
- "sheep": 20,
191
- "shoe": 29,
192
- "sink": 81,
193
- "skateboard": 41,
194
- "skis": 35,
195
- "snowboard": 36,
196
- "spoon": 50,
197
- "sports ball": 37,
198
- "stop sign": 13,
199
- "street sign": 12,
200
- "suitcase": 33,
201
- "surfboard": 42,
202
- "teddy bear": 88,
203
- "tennis racket": 43,
204
- "tie": 32,
205
- "toaster": 80,
206
- "toilet": 70,
207
- "toothbrush": 90,
208
- "traffic light": 10,
209
- "train": 7,
210
- "truck": 8,
211
- "tv": 72,
212
- "umbrella": 28,
213
- "vase": 86,
214
- "window": 68,
215
- "wine glass": 46,
216
- "zebra": 24
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
217
  },
218
  "mask_loss_coefficient": 1,
219
  "max_position_embeddings": 1024,
 
1
  {
 
2
  "activation_dropout": 0.0,
3
  "activation_function": "relu",
4
  "architectures": [
 
26
  "giou_cost": 2,
27
  "giou_loss_coefficient": 2,
28
  "id2label": {
29
+ "0": "bag",
30
+ "1": "bin",
31
+ "2": "bottle",
32
+ "3": "carrot",
33
+ "4": "carrots",
34
+ "5": "container",
35
+ "6": "courgette",
36
+ "7": "cup",
37
+ "8": "cupboard",
38
+ "9": "cutlery",
39
+ "10": "cutting board",
40
+ "11": "door",
41
+ "12": "drawer",
42
+ "13": "fork",
43
+ "14": "fridge",
44
+ "15": "glass",
45
+ "16": "hob",
46
+ "17": "kettle",
47
+ "18": "knife",
48
+ "19": "lid",
49
+ "20": "oil",
50
+ "21": "onion",
51
+ "22": "pan",
52
+ "23": "plate",
53
+ "24": "rice",
54
+ "25": "salt",
55
+ "26": "sauce pan",
56
+ "27": "saucepan",
57
+ "28": "spatula",
58
+ "29": "spoon",
59
+ "30": "table",
60
+ "31": "tap",
61
+ "32": "vegetables",
62
+ "33": "cloth",
63
+ "34": "colander",
64
+ "35": "cooker",
65
+ "36": "counter",
66
+ "37": "liquid",
67
+ "38": "rag",
68
+ "39": "sink",
69
+ "40": "sponge",
70
+ "41": "washing",
71
+ "42": "box",
72
+ "43": "bowl",
73
+ "44": "bread",
74
+ "45": "food",
75
+ "46": "frying pan",
76
+ "47": "ladle",
77
+ "48": "mushroom",
78
+ "49": "plastic",
79
+ "50": "potato",
80
+ "51": "potatoes",
81
+ "52": "sauce",
82
+ "53": "sausage",
83
+ "54": "skin",
84
+ "55": "aubergine",
85
+ "56": "dough",
86
+ "57": "flour",
87
+ "58": "leaves",
88
+ "59": "olives",
89
+ "60": "onions",
90
+ "61": "oven",
91
+ "62": "pizza",
92
+ "63": "plates",
93
+ "64": "scissors",
94
+ "65": "tray",
95
+ "66": "lettuce",
96
+ "67": "salad",
97
+ "68": "tomatoes",
98
+ "69": "pepper",
99
+ "70": "squash",
100
+ "71": "coffee",
101
+ "72": "coffee pot",
102
+ "73": "filter",
103
+ "74": "milk",
104
+ "75": "mug",
105
+ "76": "pot",
106
+ "77": "cheese",
107
+ "78": "chicken",
108
+ "79": "chopping board",
109
+ "80": "microwave",
110
+ "81": "pasta",
111
+ "82": "pesto",
112
+ "83": "stove",
113
+ "84": "tupperware",
114
+ "85": "jar",
115
+ "86": "cucumber",
116
+ "87": "packaging",
117
+ "88": "tomato",
118
+ "89": "garlic",
119
+ "90": "meat",
120
+ "91": "mixture",
121
+ "92": "peel",
122
+ "93": "spice",
123
+ "94": "drying rack",
124
+ "95": "dishwasher",
125
+ "96": "egg",
126
+ "97": "eggs",
127
+ "98": "patty",
128
+ "99": "paneer",
129
+ "100": "ginger",
130
+ "101": "spices",
131
+ "102": "curry",
132
+ "103": "pack",
133
+ "104": "package",
134
+ "105": "teaspoon",
135
+ "106": "ingredients",
136
+ "107": "bacon",
137
+ "108": "chilli",
138
+ "109": "dish",
139
+ "110": "tortilla",
140
+ "111": "kitchen",
141
+ "112": "salad bowl",
142
+ "113": "surface",
143
+ "114": "cabinet",
144
+ "115": "fish",
145
+ "116": "tortillas",
146
+ "117": "peach"
147
  },
148
  "init_std": 0.02,
149
  "init_xavier_std": 1.0,
150
  "is_encoder_decoder": true,
151
  "label2id": {
152
+ "aubergine": 55,
153
+ "bacon": 107,
154
+ "bag": 0,
155
+ "bin": 1,
156
+ "bottle": 2,
157
+ "bowl": 43,
158
+ "box": 42,
159
+ "bread": 44,
160
+ "cabinet": 114,
161
+ "carrot": 3,
162
+ "carrots": 4,
163
+ "cheese": 77,
164
+ "chicken": 78,
165
+ "chilli": 108,
166
+ "chopping board": 79,
167
+ "cloth": 33,
168
+ "coffee": 71,
169
+ "coffee pot": 72,
170
+ "colander": 34,
171
+ "container": 5,
172
+ "cooker": 35,
173
+ "counter": 36,
174
+ "courgette": 6,
175
+ "cucumber": 86,
176
+ "cup": 7,
177
+ "cupboard": 8,
178
+ "curry": 102,
179
+ "cutlery": 9,
180
+ "cutting board": 10,
181
+ "dish": 109,
182
+ "dishwasher": 95,
183
+ "door": 11,
184
+ "dough": 56,
185
+ "drawer": 12,
186
+ "drying rack": 94,
187
+ "egg": 96,
188
+ "eggs": 97,
189
+ "filter": 73,
190
+ "fish": 115,
191
+ "flour": 57,
192
+ "food": 45,
193
+ "fork": 13,
194
+ "fridge": 14,
195
+ "frying pan": 46,
196
+ "garlic": 89,
197
+ "ginger": 100,
198
+ "glass": 15,
199
+ "hob": 16,
200
+ "ingredients": 106,
201
+ "jar": 85,
202
+ "kettle": 17,
203
+ "kitchen": 111,
204
+ "knife": 18,
205
+ "ladle": 47,
206
+ "leaves": 58,
207
+ "lettuce": 66,
208
+ "lid": 19,
209
+ "liquid": 37,
210
+ "meat": 90,
211
+ "microwave": 80,
212
+ "milk": 74,
213
+ "mixture": 91,
214
+ "mug": 75,
215
+ "mushroom": 48,
216
+ "oil": 20,
217
+ "olives": 59,
218
+ "onion": 21,
219
+ "onions": 60,
220
+ "oven": 61,
221
+ "pack": 103,
222
+ "package": 104,
223
+ "packaging": 87,
224
+ "pan": 22,
225
+ "paneer": 99,
226
+ "pasta": 81,
227
+ "patty": 98,
228
+ "peach": 117,
229
+ "peel": 92,
230
+ "pepper": 69,
231
+ "pesto": 82,
232
+ "pizza": 62,
233
+ "plastic": 49,
234
+ "plate": 23,
235
+ "plates": 63,
236
+ "pot": 76,
237
+ "potato": 50,
238
+ "potatoes": 51,
239
+ "rag": 38,
240
+ "rice": 24,
241
+ "salad": 67,
242
+ "salad bowl": 112,
243
+ "salt": 25,
244
+ "sauce": 52,
245
+ "sauce pan": 26,
246
+ "saucepan": 27,
247
+ "sausage": 53,
248
+ "scissors": 64,
249
+ "sink": 39,
250
+ "skin": 54,
251
+ "spatula": 28,
252
+ "spice": 93,
253
+ "spices": 101,
254
+ "sponge": 40,
255
+ "spoon": 29,
256
+ "squash": 70,
257
+ "stove": 83,
258
+ "surface": 113,
259
+ "table": 30,
260
+ "tap": 31,
261
+ "teaspoon": 105,
262
+ "tomato": 88,
263
+ "tomatoes": 68,
264
+ "tortilla": 110,
265
+ "tortillas": 116,
266
+ "tray": 65,
267
+ "tupperware": 84,
268
+ "vegetables": 32,
269
+ "washing": 41
270
  },
271
  "mask_loss_coefficient": 1,
272
  "max_position_embeddings": 1024,
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fb3a73f2a88ee327b1ee665fddc27bc0bd0a111ae5d0d11ce408c06dbd95a8bf
3
- size 166697301
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1947e33fa00c52f52d6b3e7e3cf5f46618aa2fb85104cd07251d86922a27089d
3
+ size 166725077