|
{ |
|
"_name_or_path": "C:\\Users\\eatin\\yolo\\yolov7", |
|
"anchors": [ |
|
[ |
|
12, |
|
16, |
|
19, |
|
36, |
|
40, |
|
28 |
|
], |
|
[ |
|
36, |
|
75, |
|
76, |
|
55, |
|
72, |
|
146 |
|
], |
|
[ |
|
142, |
|
110, |
|
192, |
|
243, |
|
459, |
|
401 |
|
] |
|
], |
|
"architectures": [ |
|
"YolosForObjectDetection" |
|
], |
|
"attention_probs_dropout_prob": 0.0, |
|
"auxiliary_loss": false, |
|
"backbone": [ |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
32, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
64, |
|
3, |
|
2 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
64, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
3, |
|
2 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
64, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-2, |
|
1, |
|
"Conv", |
|
[ |
|
64, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
64, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
64, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
64, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
64, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
[ |
|
-1, |
|
-3, |
|
-5, |
|
-6 |
|
], |
|
1, |
|
"Concat", |
|
[ |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"MP", |
|
[] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-3, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
3, |
|
2 |
|
] |
|
], |
|
[ |
|
[ |
|
-1, |
|
-3 |
|
], |
|
1, |
|
"Concat", |
|
[ |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-2, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
[ |
|
-1, |
|
-3, |
|
-5, |
|
-6 |
|
], |
|
1, |
|
"Concat", |
|
[ |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
512, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"MP", |
|
[] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-3, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
3, |
|
2 |
|
] |
|
], |
|
[ |
|
[ |
|
-1, |
|
-3 |
|
], |
|
1, |
|
"Concat", |
|
[ |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-2, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
[ |
|
-1, |
|
-3, |
|
-5, |
|
-6 |
|
], |
|
1, |
|
"Concat", |
|
[ |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
1024, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"MP", |
|
[] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
512, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-3, |
|
1, |
|
"Conv", |
|
[ |
|
512, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
512, |
|
3, |
|
2 |
|
] |
|
], |
|
[ |
|
[ |
|
-1, |
|
-3 |
|
], |
|
1, |
|
"Concat", |
|
[ |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-2, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
[ |
|
-1, |
|
-3, |
|
-5, |
|
-6 |
|
], |
|
1, |
|
"Concat", |
|
[ |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
1024, |
|
1, |
|
1 |
|
] |
|
] |
|
], |
|
"bbox_cost": 5, |
|
"bbox_loss_coefficient": 5, |
|
"class_cost": 1, |
|
"conf_threshold": 0.1, |
|
"depth_multiple": 1.0, |
|
"eos_coefficient": 0.1, |
|
"giou_cost": 2, |
|
"giou_loss_coefficient": 2, |
|
"head": [ |
|
[ |
|
-1, |
|
1, |
|
"SPPCSPC", |
|
[ |
|
512 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"nn.Upsample", |
|
[ |
|
"None", |
|
2, |
|
"nearest" |
|
] |
|
], |
|
[ |
|
37, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
[ |
|
-1, |
|
-2 |
|
], |
|
1, |
|
"Concat", |
|
[ |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-2, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
[ |
|
-1, |
|
-2, |
|
-3, |
|
-4, |
|
-5, |
|
-6 |
|
], |
|
1, |
|
"Concat", |
|
[ |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"nn.Upsample", |
|
[ |
|
"None", |
|
2, |
|
"nearest" |
|
] |
|
], |
|
[ |
|
24, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
[ |
|
-1, |
|
-2 |
|
], |
|
1, |
|
"Concat", |
|
[ |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-2, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
64, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
64, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
64, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
64, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
[ |
|
-1, |
|
-2, |
|
-3, |
|
-4, |
|
-5, |
|
-6 |
|
], |
|
1, |
|
"Concat", |
|
[ |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"MP", |
|
[] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-3, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
3, |
|
2 |
|
] |
|
], |
|
[ |
|
[ |
|
-1, |
|
-3, |
|
63 |
|
], |
|
1, |
|
"Concat", |
|
[ |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-2, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
128, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
[ |
|
-1, |
|
-2, |
|
-3, |
|
-4, |
|
-5, |
|
-6 |
|
], |
|
1, |
|
"Concat", |
|
[ |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"MP", |
|
[] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-3, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
3, |
|
2 |
|
] |
|
], |
|
[ |
|
[ |
|
-1, |
|
-3, |
|
51 |
|
], |
|
1, |
|
"Concat", |
|
[ |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
512, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-2, |
|
1, |
|
"Conv", |
|
[ |
|
512, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
256, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
[ |
|
-1, |
|
-2, |
|
-3, |
|
-4, |
|
-5, |
|
-6 |
|
], |
|
1, |
|
"Concat", |
|
[ |
|
1 |
|
] |
|
], |
|
[ |
|
-1, |
|
1, |
|
"Conv", |
|
[ |
|
512, |
|
1, |
|
1 |
|
] |
|
], |
|
[ |
|
75, |
|
1, |
|
"RepConv", |
|
[ |
|
256, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
88, |
|
1, |
|
"RepConv", |
|
[ |
|
512, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
101, |
|
1, |
|
"RepConv", |
|
[ |
|
1024, |
|
3, |
|
1 |
|
] |
|
], |
|
[ |
|
[ |
|
102, |
|
103, |
|
104 |
|
], |
|
1, |
|
"IDetect", |
|
[ |
|
"nc", |
|
"anchors" |
|
] |
|
] |
|
], |
|
"hidden_act": "gelu", |
|
"hidden_dropout_prob": 0.0, |
|
"hidden_size": 768, |
|
"image_size": [ |
|
512, |
|
864 |
|
], |
|
"initializer_range": 0.02, |
|
"input_size": [ |
|
640, |
|
640 |
|
], |
|
"intermediate_size": 3072, |
|
"iou_threshold": 0.1, |
|
"layer_norm_eps": 1e-12, |
|
"model_type": "yolos", |
|
"names": [ |
|
"0", |
|
"1" |
|
], |
|
"nc": 2, |
|
"nms_threshold": 0.1, |
|
"num_attention_heads": 12, |
|
"num_channels": 3, |
|
"num_detection_tokens": 100, |
|
"num_hidden_layers": 12, |
|
"patch_size": 16, |
|
"qkv_bias": true, |
|
"save_txt": true, |
|
"torch_dtype": "float32", |
|
"transformers_version": "4.36.2", |
|
"use_mid_position_embeddings": true, |
|
"weights": "yolov7.pt", |
|
"width_multiple": 1.0 |
|
} |
|
|