peter881122 commited on
Commit
42c3ca2
·
1 Parent(s): 16b075e

Upload DetrForObjectDetection

Browse files
Files changed (2) hide show
  1. config.json +40 -40
  2. pytorch_model.bin +2 -2
config.json CHANGED
@@ -119,51 +119,51 @@
119
  "giou_cost": 2,
120
  "giou_loss_coefficient": 2,
121
  "id2label": {
122
- "0": "LABEL_0",
123
- "1": "LABEL_1",
124
- "2": "LABEL_2",
125
- "3": "LABEL_3",
126
- "4": "LABEL_4",
127
- "5": "LABEL_5",
128
- "6": "LABEL_6",
129
- "7": "LABEL_7",
130
- "8": "LABEL_8",
131
- "9": "LABEL_9",
132
- "10": "LABEL_10",
133
- "11": "LABEL_11",
134
- "12": "LABEL_12",
135
- "13": "LABEL_13",
136
- "14": "LABEL_14",
137
- "15": "LABEL_15",
138
- "16": "LABEL_16",
139
- "17": "LABEL_17",
140
- "18": "LABEL_18",
141
- "19": "LABEL_19"
142
  },
143
  "init_std": 0.02,
144
  "init_xavier_std": 1.0,
145
  "is_encoder_decoder": true,
146
  "label2id": {
147
- "LABEL_0": 0,
148
- "LABEL_1": 1,
149
- "LABEL_10": 10,
150
- "LABEL_11": 11,
151
- "LABEL_12": 12,
152
- "LABEL_13": 13,
153
- "LABEL_14": 14,
154
- "LABEL_15": 15,
155
- "LABEL_16": 16,
156
- "LABEL_17": 17,
157
- "LABEL_18": 18,
158
- "LABEL_19": 19,
159
- "LABEL_2": 2,
160
- "LABEL_3": 3,
161
- "LABEL_4": 4,
162
- "LABEL_5": 5,
163
- "LABEL_6": 6,
164
- "LABEL_7": 7,
165
- "LABEL_8": 8,
166
- "LABEL_9": 9
167
  },
168
  "mask_loss_coefficient": 1,
169
  "model_type": "detr",
 
119
  "giou_cost": 2,
120
  "giou_loss_coefficient": 2,
121
  "id2label": {
122
+ "1": "WASTE_1",
123
+ "2": "WASTE_2",
124
+ "3": "WASTE_3",
125
+ "4": "WASTE_4",
126
+ "5": "WASTE_5",
127
+ "6": "WASTE_6",
128
+ "7": "WASTE_7",
129
+ "8": "WASTE_8",
130
+ "9": "WASTE_9",
131
+ "10": "WASTE_10",
132
+ "11": "WASTE_11",
133
+ "12": "WASTE_12",
134
+ "13": "WASTE_13",
135
+ "14": "WASTE_14",
136
+ "15": "WASTE_15",
137
+ "16": "WASTE_16",
138
+ "17": "WASTE_17",
139
+ "18": "WASTE_18",
140
+ "19": "WASTE_19",
141
+ "20": "WASTE_20"
142
  },
143
  "init_std": 0.02,
144
  "init_xavier_std": 1.0,
145
  "is_encoder_decoder": true,
146
  "label2id": {
147
+ "WASTE_1": 1,
148
+ "WASTE_10": 10,
149
+ "WASTE_11": 11,
150
+ "WASTE_12": 12,
151
+ "WASTE_13": 13,
152
+ "WASTE_14": 14,
153
+ "WASTE_15": 15,
154
+ "WASTE_16": 16,
155
+ "WASTE_17": 17,
156
+ "WASTE_18": 18,
157
+ "WASTE_19": 19,
158
+ "WASTE_2": 2,
159
+ "WASTE_20": 20,
160
+ "WASTE_3": 3,
161
+ "WASTE_4": 4,
162
+ "WASTE_5": 5,
163
+ "WASTE_6": 6,
164
+ "WASTE_7": 7,
165
+ "WASTE_8": 8,
166
+ "WASTE_9": 9
167
  },
168
  "mask_loss_coefficient": 1,
169
  "model_type": "detr",
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a05c59a194d797522ac85badd0b22f2b71a98f1b61745227beb221f256650515
3
- size 166635669
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:836201fc7491b2650d82a6e33a365ed6fd594cbb3d4b8f50f98b0b7305f73abc
3
+ size 166643093