基于改进YOLOv7的金属表面缺陷检测方法

打开文本图片集

摘  要： 针对金属表面缺陷检测中不同缺陷之间存在相似性以及小目标缺陷的问题，提出一种基于改进YOLOv7的金属表面缺陷检测算法。首先通过构建更大的高效层聚合网络（V-ELAN）模块来有效增强网络对不同缺陷间差异较小的学习能力。其次在Neck部分结合空间通道双重注意力机制，提升对图像中小目标特征提取能力，减少无用特征的干扰；采用Alpha IOU损失函数替换原来的CIOU损失函数，加速网络收敛，提高网络的鲁棒性。（剩余10709字）

网站仅支持在线阅读（不支持PDF下载），如需保存文章，可以选择【打印】保存。