注意:注意力机制可以加在Backbone、Neck、Head等部分,常见的有两种:一种是在主干的SPPF前面添加一层;二是将Backbone中的C3全部替换。不同的位置效果可能不同,需要我们去反复测试。 步骤5 验证是否添加成功 在yolo.py文件里,将配置改为我们刚才自定义的yolov5s_CBAM.yaml 然后运行yolo.py,得到结果 步骤6 修改trai...
1、添加GAM注意力机制; 添加部位为backbone: 2、添加小目标检测头; 新增检测4X4以上目标的检测头,提高对小目标的检测能力 3、替换为Wise_IoU损失函数 本项目中已将YOLOv8内置的CIoU替换为Wise-IoU 实验数据集 yolo格式的人体头部数据集(主要由教室等场所的摄像头拍摄获得) train:3475 val:868 历次有效提升的实...