标题:改进YOLOv7的带钢表面缺陷检测算法 作者:孙卫波 丁卫 作者单位:上海大学机电工程与自动化学院,上海 200444 关键字:带钢;缺陷检测;YOLOv7;PConv;SimAM;BiFPN 摘要:针对带钢缺陷检测中存在的计算复杂度高、对小目标检测效果差等问题,提出了一种基于改进YOLOv7的小目标检测算法,用于区分带钢在生产过程中产生的表面缺陷。在YOLOv7的基础上,首先,将loss部分的损失函数CIOU替换为MPDiou,可以更好地处理目标框之间的交叉、遮挡等情况;接着,使用PConv(Partial Convolution)替换YOLOv7中的Backbone部分的卷积,可以减少算法的冗余计算量;最后,在YOLOv7的Head部分引入SimAM(Simple Attention Mechanism)注意力机制、动态蛇形卷积(Dynamic Snake Convolution)和BiFPN特征融合模块,以增强卷积神经网络的特征表达能力。 |