rafaelpadilla commited on
Commit
8cf0fc1
1 Parent(s): 9919f51

Update config.json

Browse files

Fixing and reducing ids

Files changed (1) hide show
  1. config.json +161 -173
config.json CHANGED
@@ -35,97 +35,86 @@
35
  "freeze_batch_norm_2d": true,
36
  "hidden_dim": 256,
37
  "id2label": {
38
- "0": "N/A",
39
- "1": "person",
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
- "2": "bicycle",
51
- "20": "sheep",
52
- "21": "cow",
53
- "22": "elephant",
54
- "23": "bear",
55
- "24": "zebra",
56
- "25": "giraffe",
57
- "26": "hat",
58
- "27": "backpack",
59
- "28": "umbrella",
60
- "29": "shoe",
61
- "3": "car",
62
- "30": "eye glasses",
63
- "31": "handbag",
64
- "32": "tie",
65
- "33": "suitcase",
66
- "34": "frisbee",
67
- "35": "skis",
68
- "36": "snowboard",
69
- "37": "sports ball",
70
- "38": "kite",
71
- "39": "baseball bat",
72
- "4": "motorcycle",
73
- "40": "baseball glove",
74
- "41": "skateboard",
75
- "42": "surfboard",
76
- "43": "tennis racket",
77
- "44": "bottle",
78
- "45": "plate",
79
- "46": "wine glass",
80
- "47": "cup",
81
- "48": "fork",
82
- "49": "knife",
83
- "5": "airplane",
84
- "50": "spoon",
85
- "51": "bowl",
86
- "52": "banana",
87
- "53": "apple",
88
- "54": "sandwich",
89
- "55": "orange",
90
- "56": "broccoli",
91
- "57": "carrot",
92
- "58": "hot dog",
93
- "59": "pizza",
94
- "6": "bus",
95
- "60": "donut",
96
- "61": "cake",
97
- "62": "chair",
98
- "63": "couch",
99
- "64": "potted plant",
100
- "65": "bed",
101
- "66": "mirror",
102
- "67": "dining table",
103
- "68": "window",
104
- "69": "desk",
105
- "7": "train",
106
- "70": "toilet",
107
- "71": "door",
108
- "72": "tv",
109
- "73": "laptop",
110
- "74": "mouse",
111
- "75": "remote",
112
- "76": "keyboard",
113
- "77": "cell phone",
114
- "78": "microwave",
115
- "79": "oven",
116
- "8": "truck",
117
- "80": "toaster",
118
- "81": "sink",
119
- "82": "refrigerator",
120
- "83": "blender",
121
- "84": "book",
122
- "85": "clock",
123
- "86": "vase",
124
- "87": "scissors",
125
- "88": "teddy bear",
126
- "89": "hair drier",
127
- "9": "boat",
128
- "90": "toothbrush"
129
  },
130
  "in_channels": [
131
  512,
@@ -133,88 +122,87 @@
133
  2048
134
  ],
135
  "initializer_range": 0.02,
136
- "label2id": {
137
- "N/A": 83,
138
- "airplane": 5,
139
- "apple": 53,
140
- "backpack": 27,
141
- "banana": 52,
142
- "baseball bat": 39,
143
- "baseball glove": 40,
144
- "bear": 23,
145
- "bed": 65,
146
- "bench": 15,
147
- "bicycle": 2,
148
- "bird": 16,
149
- "boat": 9,
150
- "book": 84,
151
- "bottle": 44,
152
- "bowl": 51,
153
- "broccoli": 56,
154
- "bus": 6,
155
- "cake": 61,
156
- "car": 3,
157
- "carrot": 57,
158
- "cat": 17,
159
- "cell phone": 77,
160
- "chair": 62,
161
- "clock": 85,
162
- "couch": 63,
163
- "cow": 21,
164
- "cup": 47,
165
- "dining table": 67,
166
- "dog": 18,
167
- "donut": 60,
168
- "elephant": 22,
169
- "fire hydrant": 11,
170
- "fork": 48,
171
- "frisbee": 34,
172
- "giraffe": 25,
173
- "hair drier": 89,
174
- "handbag": 31,
175
- "horse": 19,
176
- "hot dog": 58,
177
- "keyboard": 76,
178
- "kite": 38,
179
- "knife": 49,
180
- "laptop": 73,
181
- "microwave": 78,
182
- "motorcycle": 4,
183
- "mouse": 74,
184
- "orange": 55,
185
- "oven": 79,
186
- "parking meter": 14,
187
- "person": 1,
188
- "pizza": 59,
189
- "potted plant": 64,
190
- "refrigerator": 82,
191
- "remote": 75,
192
- "sandwich": 54,
193
- "scissors": 87,
194
- "sheep": 20,
195
- "sink": 81,
196
- "skateboard": 41,
197
- "skis": 35,
198
- "snowboard": 36,
199
- "spoon": 50,
200
- "sports ball": 37,
201
- "stop sign": 13,
202
- "suitcase": 33,
203
- "surfboard": 42,
204
- "teddy bear": 88,
205
- "tennis racket": 43,
206
- "tie": 32,
207
- "toaster": 80,
208
- "toilet": 70,
209
- "toothbrush": 90,
210
- "traffic light": 10,
211
- "train": 7,
212
- "truck": 8,
213
- "tv": 72,
214
- "umbrella": 28,
215
- "vase": 86,
216
- "wine glass": 46,
217
- "zebra": 24
218
  },
219
  "label_noise_ratio": 0.5,
220
  "learnt_init_query": false,
 
35
  "freeze_batch_norm_2d": true,
36
  "hidden_dim": 256,
37
  "id2label": {
38
+ "0": "person",
39
+ "1": "bicycle",
40
+ "2": "car",
41
+ "3": "motorcycle",
42
+ "4": "airplane",
43
+ "5": "bus",
44
+ "6": "train",
45
+ "7": "truck",
46
+ "8": "boat",
47
+ "9": "traffic light",
48
+ "10": "fire hydrant",
49
+ "11": "stop sign",
50
+ "12": "parking meter",
51
+ "13": "bench",
52
+ "14": "bird",
53
+ "15": "cat",
54
+ "16": "dog",
55
+ "17": "horse",
56
+ "18": "sheep",
57
+ "19": "cow",
58
+ "20": "elephant",
59
+ "21": "bear",
60
+ "22": "zebra",
61
+ "23": "giraffe",
62
+ "24": "backpack",
63
+ "25": "umbrella",
64
+ "26": "handbag",
65
+ "27": "tie",
66
+ "28": "suitcase",
67
+ "29": "frisbee",
68
+ "30": "skis",
69
+ "31": "snowboard",
70
+ "32": "sports ball",
71
+ "33": "kite",
72
+ "34": "baseball bat",
73
+ "35": "baseball glove",
74
+ "36": "skateboard",
75
+ "37": "surfboard",
76
+ "38": "tennis racket",
77
+ "39": "bottle",
78
+ "40": "wine glass",
79
+ "41": "cup",
80
+ "42": "fork",
81
+ "43": "knife",
82
+ "44": "spoon",
83
+ "45": "bowl",
84
+ "46": "banana",
85
+ "47": "apple",
86
+ "48": "sandwich",
87
+ "49": "orange",
88
+ "50": "broccoli",
89
+ "51": "carrot",
90
+ "52": "hot dog",
91
+ "53": "pizza",
92
+ "54": "donut",
93
+ "55": "cake",
94
+ "56": "chair",
95
+ "57": "couch",
96
+ "58": "potted plant",
97
+ "59": "bed",
98
+ "60": "dining table",
99
+ "61": "toilet",
100
+ "62": "tv",
101
+ "63": "laptop",
102
+ "64": "mouse",
103
+ "65": "remote",
104
+ "66": "keyboard",
105
+ "67": "cell phone",
106
+ "68": "microwave",
107
+ "69": "oven",
108
+ "70": "toaster",
109
+ "71": "sink",
110
+ "72": "refrigerator",
111
+ "73": "book",
112
+ "74": "clock",
113
+ "75": "vase",
114
+ "76": "scissors",
115
+ "77": "teddy bear",
116
+ "78": "hair drier",
117
+ "79": "toothbrush"
 
 
 
 
 
 
 
 
 
 
 
118
  },
119
  "in_channels": [
120
  512,
 
122
  2048
123
  ],
124
  "initializer_range": 0.02,
125
+ "label2id": {
126
+ "person: 0,
127
+ "bicycle: 1,
128
+ "car: 2,
129
+ "motorcycle: 3,
130
+ "airplane: 4,
131
+ "bus: 5,
132
+ "train: 6,
133
+ "truck: 7,
134
+ "boat: 8,
135
+ "traffic light: 9,
136
+ "fire hydrant: 10,
137
+ "stop sign: 11,
138
+ "parking meter: 12,
139
+ "bench: 13,
140
+ "bird: 14,
141
+ "cat: 15,
142
+ "dog: 16,
143
+ "horse: 17,
144
+ "sheep: 18,
145
+ "cow: 19,
146
+ "elephant: 20,
147
+ "bear: 21,
148
+ "zebra: 22,
149
+ "giraffe: 23,
150
+ "backpack: 24,
151
+ "umbrella: 25,
152
+ "handbag: 26,
153
+ "tie: 27,
154
+ "suitcase: 28,
155
+ "frisbee: 29,
156
+ "skis: 30,
157
+ "snowboard: 31,
158
+ "sports ball: 32,
159
+ "kite: 33,
160
+ "baseball bat: 34,
161
+ "baseball glove: 35,
162
+ "skateboard: 36,
163
+ "surfboard: 37,
164
+ "tennis racket: 38,
165
+ "bottle: 39,
166
+ "wine glass: 40,
167
+ "cup: 41,
168
+ "fork: 42,
169
+ "knife: 43,
170
+ "spoon: 44,
171
+ "bowl: 45,
172
+ "banana: 46,
173
+ "apple: 47,
174
+ "sandwich: 48,
175
+ "orange: 49,
176
+ "broccoli: 50,
177
+ "carrot: 51,
178
+ "hot dog: 52,
179
+ "pizza: 53,
180
+ "donut: 54,
181
+ "cake: 55,
182
+ "chair: 56,
183
+ "couch: 57,
184
+ "potted plant: 58,
185
+ "bed: 59,
186
+ "dining table: 60,
187
+ "toilet: 61,
188
+ "tv: 62,
189
+ "laptop: 63,
190
+ "mouse: 64,
191
+ "remote: 65,
192
+ "keyboard: 66,
193
+ "cell phone: 67,
194
+ "microwave: 68,
195
+ "oven: 69,
196
+ "toaster: 70,
197
+ "sink: 71,
198
+ "refrigerator: 72,
199
+ "book: 73,
200
+ "clock: 74,
201
+ "vase: 75,
202
+ "scissors: 76,
203
+ "teddy bear: 77,
204
+ "hair drier: 78,
205
+ "toothbrush: 79
 
206
  },
207
  "label_noise_ratio": 0.5,
208
  "learnt_init_query": false,