Upload DetrForObjectDetection
Browse files- config.json +10 -0
- model.safetensors +2 -2
config.json
CHANGED
@@ -120,9 +120,19 @@
|
|
120 |
"eos_coefficient": 0.1,
|
121 |
"giou_cost": 2,
|
122 |
"giou_loss_coefficient": 2,
|
|
|
|
|
|
|
|
|
|
|
123 |
"init_std": 0.02,
|
124 |
"init_xavier_std": 1.0,
|
125 |
"is_encoder_decoder": true,
|
|
|
|
|
|
|
|
|
|
|
126 |
"mask_loss_coefficient": 1,
|
127 |
"model_type": "detr",
|
128 |
"num_channels": 3,
|
|
|
120 |
"eos_coefficient": 0.1,
|
121 |
"giou_cost": 2,
|
122 |
"giou_loss_coefficient": 2,
|
123 |
+
"id2label": {
|
124 |
+
"0": "LABEL_0",
|
125 |
+
"1": "LABEL_1",
|
126 |
+
"2": "LABEL_2"
|
127 |
+
},
|
128 |
"init_std": 0.02,
|
129 |
"init_xavier_std": 1.0,
|
130 |
"is_encoder_decoder": true,
|
131 |
+
"label2id": {
|
132 |
+
"LABEL_0": 0,
|
133 |
+
"LABEL_1": 1,
|
134 |
+
"LABEL_2": 2
|
135 |
+
},
|
136 |
"mask_loss_coefficient": 1,
|
137 |
"model_type": "detr",
|
138 |
"num_channels": 3,
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b62c391239608669450f6285162a0cb7d71a9e5483c3b95df1231cbdb023ab0b
|
3 |
+
size 166505784
|