Upload DetrForObjectDetection
Browse files- config.json +8 -3
- model.safetensors +2 -2
config.json
CHANGED
@@ -102,6 +102,7 @@
|
|
102 |
"typical_p": 1.0,
|
103 |
"use_bfloat16": false
|
104 |
},
|
|
|
105 |
"bbox_cost": 5,
|
106 |
"bbox_loss_coefficient": 5,
|
107 |
"class_cost": 1,
|
@@ -126,7 +127,9 @@
|
|
126 |
"2": "LABEL_2",
|
127 |
"3": "LABEL_3",
|
128 |
"4": "LABEL_4",
|
129 |
-
"5": "LABEL_5"
|
|
|
|
|
130 |
},
|
131 |
"init_std": 0.02,
|
132 |
"init_xavier_std": 1.0,
|
@@ -137,7 +140,9 @@
|
|
137 |
"LABEL_2": 2,
|
138 |
"LABEL_3": 3,
|
139 |
"LABEL_4": 4,
|
140 |
-
"LABEL_5": 5
|
|
|
|
|
141 |
},
|
142 |
"mask_loss_coefficient": 1,
|
143 |
"model_type": "detr",
|
@@ -147,7 +152,7 @@
|
|
147 |
"position_embedding_type": "sine",
|
148 |
"scale_embedding": false,
|
149 |
"torch_dtype": "float32",
|
150 |
-
"transformers_version": "4.
|
151 |
"use_pretrained_backbone": null,
|
152 |
"use_timm_backbone": false
|
153 |
}
|
|
|
102 |
"typical_p": 1.0,
|
103 |
"use_bfloat16": false
|
104 |
},
|
105 |
+
"backbone_kwargs": null,
|
106 |
"bbox_cost": 5,
|
107 |
"bbox_loss_coefficient": 5,
|
108 |
"class_cost": 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,
|
|
|
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",
|
|
|
152 |
"position_embedding_type": "sine",
|
153 |
"scale_embedding": false,
|
154 |
"torch_dtype": "float32",
|
155 |
+
"transformers_version": "4.38.2",
|
156 |
"use_pretrained_backbone": null,
|
157 |
"use_timm_backbone": false
|
158 |
}
|
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:61b6d7b6033a248a6412fe639d2722f27b45e80358a51c0d5b00faa43fc6e83e
|
3 |
+
size 166510924
|