Upload Yolov6ForObjectDetection
Browse files- config.json +5 -11
config.json
CHANGED
@@ -2,6 +2,7 @@
|
|
2 |
"architectures": [
|
3 |
"Yolov6ForObjectDetection"
|
4 |
],
|
|
|
5 |
"backbone_csp_e": 0.0,
|
6 |
"backbone_cspsppf": true,
|
7 |
"backbone_fuse_P2": true,
|
@@ -23,6 +24,7 @@
|
|
23 |
"block_type": "Yolov6RepVGGBlock",
|
24 |
"class_loss_coefficient": 1.0,
|
25 |
"dfl_loss_coefficient": 1.0,
|
|
|
26 |
"forward_fuse": false,
|
27 |
"head_anchors": 3,
|
28 |
"head_in_channels": [
|
@@ -36,7 +38,6 @@
|
|
36 |
16,
|
37 |
32
|
38 |
],
|
39 |
-
"hidden_size": 192,
|
40 |
"id2label": {
|
41 |
"0": "person",
|
42 |
"1": "bicycle",
|
@@ -119,13 +120,9 @@
|
|
119 |
"78": "hair drier",
|
120 |
"79": "toothbrush"
|
121 |
},
|
122 |
-
"image_size":
|
123 |
-
800,
|
124 |
-
1333
|
125 |
-
],
|
126 |
"in_channels": 3,
|
127 |
"initializer_range": 0.02,
|
128 |
-
"intermediate_size": 768,
|
129 |
"iou_loss_coefficient": 1.0,
|
130 |
"iou_type": "siou",
|
131 |
"label2id": {
|
@@ -227,12 +224,9 @@
|
|
227 |
128
|
228 |
],
|
229 |
"neck_stage_block_type": "Yolov6RepBlock",
|
230 |
-
"num_attention_heads": 3,
|
231 |
-
"num_hidden_layers": 12,
|
232 |
"reg_max": 0,
|
233 |
"reg_max_proj": 16,
|
234 |
"torch_dtype": "float32",
|
235 |
-
"transformers_version": "4.
|
236 |
-
"use_dfl": false
|
237 |
-
"use_mid_position_embeddings": false
|
238 |
}
|
|
|
2 |
"architectures": [
|
3 |
"Yolov6ForObjectDetection"
|
4 |
],
|
5 |
+
"atss_warmup_epoch": 0,
|
6 |
"backbone_csp_e": 0.0,
|
7 |
"backbone_cspsppf": true,
|
8 |
"backbone_fuse_P2": true,
|
|
|
24 |
"block_type": "Yolov6RepVGGBlock",
|
25 |
"class_loss_coefficient": 1.0,
|
26 |
"dfl_loss_coefficient": 1.0,
|
27 |
+
"export": false,
|
28 |
"forward_fuse": false,
|
29 |
"head_anchors": 3,
|
30 |
"head_in_channels": [
|
|
|
38 |
16,
|
39 |
32
|
40 |
],
|
|
|
41 |
"id2label": {
|
42 |
"0": "person",
|
43 |
"1": "bicycle",
|
|
|
120 |
"78": "hair drier",
|
121 |
"79": "toothbrush"
|
122 |
},
|
123 |
+
"image_size": 640,
|
|
|
|
|
|
|
124 |
"in_channels": 3,
|
125 |
"initializer_range": 0.02,
|
|
|
126 |
"iou_loss_coefficient": 1.0,
|
127 |
"iou_type": "siou",
|
128 |
"label2id": {
|
|
|
224 |
128
|
225 |
],
|
226 |
"neck_stage_block_type": "Yolov6RepBlock",
|
|
|
|
|
227 |
"reg_max": 0,
|
228 |
"reg_max_proj": 16,
|
229 |
"torch_dtype": "float32",
|
230 |
+
"transformers_version": "4.38.0.dev0",
|
231 |
+
"use_dfl": false
|
|
|
232 |
}
|