基于改进YOLOv5算法的带钢表面缺陷检测

作者：程豪; 蒋占四; 陈晓鑫; 郑洪鑫; 蒋慧

来源：组合机床与自动化加工技术, 2023, 35(10): 141-144+149.

DOI：10.13462/j.cnki.mmtamt.2023.10.031

• 缺陷检测
• YOLOv5
• 注意力机制
• 特征融合

针对当前带钢在表面缺陷检测过程中存在检测算法精度有待提高等问题，提出了一种基于改进YOLOv5算法的带钢表面缺陷检测模型。首先，在检测端构建新的检测层，提高网络对不同尺寸目标的检测；其次，在主干网络结构中引入注意力模块，进一步加强网络提取特征的能力；然后，通过BiFPN＿Add来增强深浅层特征信息的融合；最后，构建新的CNeB模块来取代各检测层对应的C3模块，进而增强网络对特征的提取。实验结果表明，改进后的算法在NEU-DET数据集上均值平均精度达到了80.9%,较原有的算法提升了4.5%,同时检测速度与原模型保持基本不变，性能优于目前其他主流的检测方法。

• 单位
  广州航海学院; 机电工程学院; 桂林电子科技大学