YOLOV7改进-添加基于注意力机制的目标检测头(DYHEAD)

DYHEAD
复制到这:
YOLOV7改进-添加基于注意力机制的目标检测头(DYHEAD)_第1张图片
1、models下新建文件
2、yolo.py中import一下
在这里插入图片描述
3、改IDetect这里
YOLOV7改进-添加基于注意力机制的目标检测头(DYHEAD)_第2张图片

YOLOV7改进-添加基于注意力机制的目标检测头(DYHEAD)_第3张图片
YOLOV7改进-添加基于注意力机制的目标检测头(DYHEAD)_第4张图片
YOLOV7改进-添加基于注意力机制的目标检测头(DYHEAD)_第5张图片
4、论文中说6的效果最好,但参数量不少,做一下工作量
在这里插入图片描述
5、在进入IDetect之前,会对RepConv做卷积
YOLOV7改进-添加基于注意力机制的目标检测头(DYHEAD)_第6张图片
5、因为DYHEAD需要三个层输入的特征层一致,所以需要改为一样的。256或者512都行
YOLOV7改进-添加基于注意力机制的目标检测头(DYHEAD)_第7张图片
6、提前安装的命令
在这里插入图片描述

你可能感兴趣的:(yolov7,YOLO,目标检测,人工智能)