xiuqhou commited on
Commit
a062e87
1 Parent(s): 8a16e96

Upload RelationDetrForObjectDetection

Browse files
Files changed (1) hide show
  1. config.json +172 -182
config.json CHANGED
@@ -41,193 +41,183 @@
41
  "hybrid_assign": 6,
42
  "hybrid_queries": 1500,
43
  "id2label": {
44
- "0": "LABEL_0",
45
- "1": "LABEL_1",
46
- "2": "LABEL_2",
47
- "3": "LABEL_3",
48
- "4": "LABEL_4",
49
- "5": "LABEL_5",
50
- "6": "LABEL_6",
51
- "7": "LABEL_7",
52
- "8": "LABEL_8",
53
- "9": "LABEL_9",
54
- "10": "LABEL_10",
55
- "11": "LABEL_11",
56
- "12": "LABEL_12",
57
- "13": "LABEL_13",
58
- "14": "LABEL_14",
59
- "15": "LABEL_15",
60
- "16": "LABEL_16",
61
- "17": "LABEL_17",
62
- "18": "LABEL_18",
63
- "19": "LABEL_19",
64
- "20": "LABEL_20",
65
- "21": "LABEL_21",
66
- "22": "LABEL_22",
67
- "23": "LABEL_23",
68
- "24": "LABEL_24",
69
- "25": "LABEL_25",
70
- "26": "LABEL_26",
71
- "27": "LABEL_27",
72
- "28": "LABEL_28",
73
- "29": "LABEL_29",
74
- "30": "LABEL_30",
75
- "31": "LABEL_31",
76
- "32": "LABEL_32",
77
- "33": "LABEL_33",
78
- "34": "LABEL_34",
79
- "35": "LABEL_35",
80
- "36": "LABEL_36",
81
- "37": "LABEL_37",
82
- "38": "LABEL_38",
83
- "39": "LABEL_39",
84
- "40": "LABEL_40",
85
- "41": "LABEL_41",
86
- "42": "LABEL_42",
87
- "43": "LABEL_43",
88
- "44": "LABEL_44",
89
- "45": "LABEL_45",
90
- "46": "LABEL_46",
91
- "47": "LABEL_47",
92
- "48": "LABEL_48",
93
- "49": "LABEL_49",
94
- "50": "LABEL_50",
95
- "51": "LABEL_51",
96
- "52": "LABEL_52",
97
- "53": "LABEL_53",
98
- "54": "LABEL_54",
99
- "55": "LABEL_55",
100
- "56": "LABEL_56",
101
- "57": "LABEL_57",
102
- "58": "LABEL_58",
103
- "59": "LABEL_59",
104
- "60": "LABEL_60",
105
- "61": "LABEL_61",
106
- "62": "LABEL_62",
107
- "63": "LABEL_63",
108
- "64": "LABEL_64",
109
- "65": "LABEL_65",
110
- "66": "LABEL_66",
111
- "67": "LABEL_67",
112
- "68": "LABEL_68",
113
- "69": "LABEL_69",
114
- "70": "LABEL_70",
115
- "71": "LABEL_71",
116
- "72": "LABEL_72",
117
- "73": "LABEL_73",
118
- "74": "LABEL_74",
119
- "75": "LABEL_75",
120
- "76": "LABEL_76",
121
- "77": "LABEL_77",
122
- "78": "LABEL_78",
123
- "79": "LABEL_79",
124
- "80": "LABEL_80",
125
- "81": "LABEL_81",
126
- "82": "LABEL_82",
127
- "83": "LABEL_83",
128
- "84": "LABEL_84",
129
- "85": "LABEL_85",
130
- "86": "LABEL_86",
131
- "87": "LABEL_87",
132
- "88": "LABEL_88",
133
- "89": "LABEL_89",
134
- "90": "LABEL_90"
135
  },
136
  "init_std": 0.02,
137
  "init_xavier_std": 1.0,
138
  "is_encoder_decoder": true,
139
  "label2id": {
140
- "LABEL_0": 0,
141
- "LABEL_1": 1,
142
- "LABEL_10": 10,
143
- "LABEL_11": 11,
144
- "LABEL_12": 12,
145
- "LABEL_13": 13,
146
- "LABEL_14": 14,
147
- "LABEL_15": 15,
148
- "LABEL_16": 16,
149
- "LABEL_17": 17,
150
- "LABEL_18": 18,
151
- "LABEL_19": 19,
152
- "LABEL_2": 2,
153
- "LABEL_20": 20,
154
- "LABEL_21": 21,
155
- "LABEL_22": 22,
156
- "LABEL_23": 23,
157
- "LABEL_24": 24,
158
- "LABEL_25": 25,
159
- "LABEL_26": 26,
160
- "LABEL_27": 27,
161
- "LABEL_28": 28,
162
- "LABEL_29": 29,
163
- "LABEL_3": 3,
164
- "LABEL_30": 30,
165
- "LABEL_31": 31,
166
- "LABEL_32": 32,
167
- "LABEL_33": 33,
168
- "LABEL_34": 34,
169
- "LABEL_35": 35,
170
- "LABEL_36": 36,
171
- "LABEL_37": 37,
172
- "LABEL_38": 38,
173
- "LABEL_39": 39,
174
- "LABEL_4": 4,
175
- "LABEL_40": 40,
176
- "LABEL_41": 41,
177
- "LABEL_42": 42,
178
- "LABEL_43": 43,
179
- "LABEL_44": 44,
180
- "LABEL_45": 45,
181
- "LABEL_46": 46,
182
- "LABEL_47": 47,
183
- "LABEL_48": 48,
184
- "LABEL_49": 49,
185
- "LABEL_5": 5,
186
- "LABEL_50": 50,
187
- "LABEL_51": 51,
188
- "LABEL_52": 52,
189
- "LABEL_53": 53,
190
- "LABEL_54": 54,
191
- "LABEL_55": 55,
192
- "LABEL_56": 56,
193
- "LABEL_57": 57,
194
- "LABEL_58": 58,
195
- "LABEL_59": 59,
196
- "LABEL_6": 6,
197
- "LABEL_60": 60,
198
- "LABEL_61": 61,
199
- "LABEL_62": 62,
200
- "LABEL_63": 63,
201
- "LABEL_64": 64,
202
- "LABEL_65": 65,
203
- "LABEL_66": 66,
204
- "LABEL_67": 67,
205
- "LABEL_68": 68,
206
- "LABEL_69": 69,
207
- "LABEL_7": 7,
208
- "LABEL_70": 70,
209
- "LABEL_71": 71,
210
- "LABEL_72": 72,
211
- "LABEL_73": 73,
212
- "LABEL_74": 74,
213
- "LABEL_75": 75,
214
- "LABEL_76": 76,
215
- "LABEL_77": 77,
216
- "LABEL_78": 78,
217
- "LABEL_79": 79,
218
- "LABEL_8": 8,
219
- "LABEL_80": 80,
220
- "LABEL_81": 81,
221
- "LABEL_82": 82,
222
- "LABEL_83": 83,
223
- "LABEL_84": 84,
224
- "LABEL_85": 85,
225
- "LABEL_86": 86,
226
- "LABEL_87": 87,
227
- "LABEL_88": 88,
228
- "LABEL_89": 89,
229
- "LABEL_9": 9,
230
- "LABEL_90": 90
231
  },
232
  "label_noise_ratio": 0.5,
233
  "max_position_embeddings": 2048,
 
41
  "hybrid_assign": 6,
42
  "hybrid_queries": 1500,
43
  "id2label": {
44
+ "0": "none",
45
+ "1": "person",
46
+ "2": "bicycle",
47
+ "3": "car",
48
+ "4": "motorcycle",
49
+ "5": "airplane",
50
+ "6": "bus",
51
+ "7": "train",
52
+ "8": "truck",
53
+ "9": "boat",
54
+ "10": "traffic light",
55
+ "11": "fire hydrant",
56
+ "12": "none",
57
+ "13": "stop sign",
58
+ "14": "parking meter",
59
+ "15": "bench",
60
+ "16": "bird",
61
+ "17": "cat",
62
+ "18": "dog",
63
+ "19": "horse",
64
+ "20": "sheep",
65
+ "21": "cow",
66
+ "22": "elephant",
67
+ "23": "bear",
68
+ "24": "zebra",
69
+ "25": "giraffe",
70
+ "26": "none",
71
+ "27": "backpack",
72
+ "28": "umbrella",
73
+ "29": "none",
74
+ "30": "none",
75
+ "31": "handbag",
76
+ "32": "tie",
77
+ "33": "suitcase",
78
+ "34": "frisbee",
79
+ "35": "skis",
80
+ "36": "snowboard",
81
+ "37": "sports ball",
82
+ "38": "kite",
83
+ "39": "baseball bat",
84
+ "40": "baseball glove",
85
+ "41": "skateboard",
86
+ "42": "surfboard",
87
+ "43": "tennis racket",
88
+ "44": "bottle",
89
+ "45": "none",
90
+ "46": "wine glass",
91
+ "47": "cup",
92
+ "48": "fork",
93
+ "49": "knife",
94
+ "50": "spoon",
95
+ "51": "bowl",
96
+ "52": "banana",
97
+ "53": "apple",
98
+ "54": "sandwich",
99
+ "55": "orange",
100
+ "56": "broccoli",
101
+ "57": "carrot",
102
+ "58": "hot dog",
103
+ "59": "pizza",
104
+ "60": "donut",
105
+ "61": "cake",
106
+ "62": "chair",
107
+ "63": "couch",
108
+ "64": "potted plant",
109
+ "65": "bed",
110
+ "66": "none",
111
+ "67": "dining table",
112
+ "68": "none",
113
+ "69": "none",
114
+ "70": "toilet",
115
+ "71": "none",
116
+ "72": "tv",
117
+ "73": "laptop",
118
+ "74": "mouse",
119
+ "75": "remote",
120
+ "76": "keyboard",
121
+ "77": "cell phone",
122
+ "78": "microwave",
123
+ "79": "oven",
124
+ "80": "toaster",
125
+ "81": "sink",
126
+ "82": "refrigerator",
127
+ "83": "none",
128
+ "84": "book",
129
+ "85": "clock",
130
+ "86": "vase",
131
+ "87": "scissors",
132
+ "88": "teddy bear",
133
+ "89": "hair drier",
134
+ "90": "toothbrush"
135
  },
136
  "init_std": 0.02,
137
  "init_xavier_std": 1.0,
138
  "is_encoder_decoder": true,
139
  "label2id": {
140
+ "airplane": 5,
141
+ "apple": 53,
142
+ "backpack": 27,
143
+ "banana": 52,
144
+ "baseball bat": 39,
145
+ "baseball glove": 40,
146
+ "bear": 23,
147
+ "bed": 65,
148
+ "bench": 15,
149
+ "bicycle": 2,
150
+ "bird": 16,
151
+ "boat": 9,
152
+ "book": 84,
153
+ "bottle": 44,
154
+ "bowl": 51,
155
+ "broccoli": 56,
156
+ "bus": 6,
157
+ "cake": 61,
158
+ "car": 3,
159
+ "carrot": 57,
160
+ "cat": 17,
161
+ "cell phone": 77,
162
+ "chair": 62,
163
+ "clock": 85,
164
+ "couch": 63,
165
+ "cow": 21,
166
+ "cup": 47,
167
+ "dining table": 67,
168
+ "dog": 18,
169
+ "donut": 60,
170
+ "elephant": 22,
171
+ "fire hydrant": 11,
172
+ "fork": 48,
173
+ "frisbee": 34,
174
+ "giraffe": 25,
175
+ "hair drier": 89,
176
+ "handbag": 31,
177
+ "horse": 19,
178
+ "hot dog": 58,
179
+ "keyboard": 76,
180
+ "kite": 38,
181
+ "knife": 49,
182
+ "laptop": 73,
183
+ "microwave": 78,
184
+ "motorcycle": 4,
185
+ "mouse": 74,
186
+ "none": 83,
187
+ "orange": 55,
188
+ "oven": 79,
189
+ "parking meter": 14,
190
+ "person": 1,
191
+ "pizza": 59,
192
+ "potted plant": 64,
193
+ "refrigerator": 82,
194
+ "remote": 75,
195
+ "sandwich": 54,
196
+ "scissors": 87,
197
+ "sheep": 20,
198
+ "sink": 81,
199
+ "skateboard": 41,
200
+ "skis": 35,
201
+ "snowboard": 36,
202
+ "spoon": 50,
203
+ "sports ball": 37,
204
+ "stop sign": 13,
205
+ "suitcase": 33,
206
+ "surfboard": 42,
207
+ "teddy bear": 88,
208
+ "tennis racket": 43,
209
+ "tie": 32,
210
+ "toaster": 80,
211
+ "toilet": 70,
212
+ "toothbrush": 90,
213
+ "traffic light": 10,
214
+ "train": 7,
215
+ "truck": 8,
216
+ "tv": 72,
217
+ "umbrella": 28,
218
+ "vase": 86,
219
+ "wine glass": 46,
220
+ "zebra": 24
 
 
 
 
 
 
 
 
 
 
221
  },
222
  "label_noise_ratio": 0.5,
223
  "max_position_embeddings": 2048,