peter881122 commited on
Commit
7fa149e
·
1 Parent(s): 89cbff3

Upload DetrForObjectDetection

Browse files
Files changed (2) hide show
  1. config.json +20 -20
  2. pytorch_model.bin +1 -1
config.json CHANGED
@@ -144,26 +144,26 @@
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",
 
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",
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b6f874b68335cbdf40d8e32e937b4086c011ace4759db8d8d2980f4fb171f4e9
3
  size 166635669
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b488324a2e751bc0948ae745defe1b804bf33014bba5acb2297e52bf9e896f5f
3
  size 166635669