YOLOv5、YOLOv8改进-BIFPN
如果需要在YOLOv5l.yaml等网络结构进行修改的话,不可直接用以上的yaml文件或者就简单修改depth_multiple为1.0,而是 需要修改Concat_bifpn, [256,256]中的通道数为对应网络实际通道数。修改yaml文件(以修改官方YOLOv5s.yaml为例),需要修改head(特征融合网络)common.py构建Concat_BIFPN模块。yolo.py中注册Concat_BIFPNt模块。加权双向金字塔结构,提升不同尺度的检测效果。
复制链接