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

DYHEAD

复制到这:

1、models下新建文件

2、yolo.py中import一下

3、改IDetect这里



4、论文中说6的效果最好,但参数量不少,做一下工作量

5、在进入IDetect之前,会对RepConv做卷积

5、因为DYHEAD需要三个层输入的特征层一致,所以需要改为一样的。256或者512都行

6、提前安装的命令

相关推荐
tellmewhoisi1 分钟前
深度学习13:反向传播4:输出层Softmax-with-Loss
人工智能·深度学习
hahaha60166 分钟前
Unsharp Masking算法综述
图像处理·人工智能·计算机视觉
武天旭7 分钟前
AI安全--AI大模型的模型窃取攻击
人工智能·安全
注塑机数据采集联网11 分钟前
中科晟达注塑机 MES 系统:从车间到数字化的智能制造实践
大数据·数据仓库·人工智能·物联网·制造·数据库架构
xierui12312314 分钟前
视频Agent的声音编排:原声、TTS、ASR与背景音乐怎样分工
人工智能·音视频
lucas_AI22 分钟前
花旗、安永、英伟达联手写的文档摄取系统报告:92% 的页面根本不需要 OCR
人工智能
七夜zippoe24 分钟前
Spring AI + Ollama 本地模型:数据不出内网的合规方案(落地全流程)
java·人工智能·spring·ai·ollama
Jsundoku27 分钟前
第三章 深度学习 -- 神经网络基础
人工智能·深度学习·神经网络
lucas_AI28 分钟前
PolyOCR-Venus:蚂蚁把 OCR 从『抄写员』练成『视觉智能体』,教师蒸馏还会自己下岗
人工智能