search

Gumpest / YOLOv5-Multibackbone-Compression

YOLOv5 Series Multi-backbone(TPH-YOLOv5, Ghostnet, ShuffleNetv2, Mobilenetv3Small, EfficientNetLite, PP-LCNet, SwinTransformer YOLO), Module(CBAM, DCN), Pruning (EagleEye, Network Slimming), Quantization (MQBench) and Deployment (TensorRT, ncnn) Compression Tool Box.
994 stars 200 forks source link

您好，在您的readme中看到了这一段有点不太理解，想请问下为什么要这么做 #74

Open wyzwyzzzzz opened 2 years ago

wyzwyzzzzz commented 2 years ago

当YOLOv5x为主体与标准结构的区别是：（1）首先去掉14和19的CBAM模块（2）降低与P2关联的通道数（128）（3）在输出头之前会添加SPP模块，注意SPP的kernel随着P的像素减小而减小（4）在CBAM之后进行输出（5）只保留backbone以及最后一层输出的TransBlock（6）采用BiFPN作为neck

我看官方实现的TPH的代码中也和您对网络做了同样的更改，想请问下为什么不用论文中的那个网络结构图而要做出这些修改呢

Xjh-UCAS commented 1 year ago

我也想问