Commit
•
20db466
1
Parent(s):
45f9a0a
Upload DetrForObjectDetection
Browse files- config.json +18 -4
- model.safetensors +2 -2
config.json
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "
|
3 |
"activation_dropout": 0.0,
|
4 |
"activation_function": "relu",
|
5 |
"architectures": [
|
6 |
-
"
|
7 |
],
|
8 |
"attention_dropout": 0.0,
|
9 |
"auxiliary_loss": false,
|
@@ -122,13 +122,27 @@
|
|
122 |
"giou_cost": 2,
|
123 |
"giou_loss_coefficient": 2,
|
124 |
"id2label": {
|
125 |
-
"0": "
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
126 |
},
|
127 |
"init_std": 0.02,
|
128 |
"init_xavier_std": 1.0,
|
129 |
"is_encoder_decoder": true,
|
130 |
"label2id": {
|
131 |
-
"LABEL_0": 0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
132 |
},
|
133 |
"mask_loss_coefficient": 1,
|
134 |
"model_type": "detr",
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "facebook/detr-resnet-50",
|
3 |
"activation_dropout": 0.0,
|
4 |
"activation_function": "relu",
|
5 |
"architectures": [
|
6 |
+
"DetrForObjectDetection"
|
7 |
],
|
8 |
"attention_dropout": 0.0,
|
9 |
"auxiliary_loss": false,
|
|
|
122 |
"giou_cost": 2,
|
123 |
"giou_loss_coefficient": 2,
|
124 |
"id2label": {
|
125 |
+
"0": "LABEL_0",
|
126 |
+
"1": "LABEL_1",
|
127 |
+
"2": "LABEL_2",
|
128 |
+
"3": "LABEL_3",
|
129 |
+
"4": "LABEL_4",
|
130 |
+
"5": "LABEL_5",
|
131 |
+
"6": "LABEL_6",
|
132 |
+
"7": "LABEL_7"
|
133 |
},
|
134 |
"init_std": 0.02,
|
135 |
"init_xavier_std": 1.0,
|
136 |
"is_encoder_decoder": true,
|
137 |
"label2id": {
|
138 |
+
"LABEL_0": 0,
|
139 |
+
"LABEL_1": 1,
|
140 |
+
"LABEL_2": 2,
|
141 |
+
"LABEL_3": 3,
|
142 |
+
"LABEL_4": 4,
|
143 |
+
"LABEL_5": 5,
|
144 |
+
"LABEL_6": 6,
|
145 |
+
"LABEL_7": 7
|
146 |
},
|
147 |
"mask_loss_coefficient": 1,
|
148 |
"model_type": "detr",
|
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:5cae7e0d5114cfd0cd2b2a218efb348a76e3753d0076ec0b198b925e31176bbb
|
3 |
+
size 166510924
|