改进 YOLOv5 网络的带钢表面缺陷检测方法
DOI:
CSTR:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

TN98; TP391. 4

基金项目:

天津市科技计划项目(企业科技特派员项目)(21YDTPJC00050)、光电信息控制和安全技术重点实验室基金项目(2021JCJQLB055008)资助


Strip steel surface defect detection method by improved YOLOv5 network
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    带钢表面缺陷检测已成为保证带钢生产质量的重要环节之一。 针对当前带钢缺陷检测算法精度有待提高等问题,提出 了一种基于 YOLOv5 网络改进的算法模型 MT-YOLOv5。 首先在主干网络中引入 Transformer 自注意力机制,使主干网络更聚焦 于图像全局特征信息的提取;其次采用 T-BiFPN 网络结构,将 Transformer 层与 BiFPN 网络结构相结合,进一步增强了图像浅层 特征信息与深层特征信息的融合;然后引入改进后的轻量化网络 RepVGG 替换主干网络中的部分卷积层,增强主干网络的特征 提取能力;最后增加预测层,检测不同尺度的目标。 实验结果表明,MT-YOLOv5 算法在 NEU-DET 数据集上的均值平均精度 (mAP)达到了 82. 4%,较原 YOLOv5s 算法提高了 5. 3%,检测速度为 65. 4 fps,更好地均衡了检测速度与检测精度。

    Abstract:

    Strip steel surface defect detection has become one of the important links to guarantee the quality of strip steel production. Aiming at the problem of improving the detection accuracy of current strip steel defect detection algorithm, an improved MT-YOLOv5 algorithm based on YOLOv5 is proposed. Firstly, introducing Transformer self-attention mechanism in the backbone network to make the network more focused on the extraction of global image feature information. Secondly, combining the Transformer layer with the BiFPN structure, and the T-BiFPN network is used to further enhance the fusion of image shallow feature information and deep feature information. Then, an improved lightweight network RepVGG is introduced to replace part of the convolutional layers in the backbone network, which can enhance the feature extraction capability of backbone network. Finally, adding a prediction layer to detect objects of different scales. The experimental results show that the value of mean average precision (mAP) of the MT-YOLOv5 algorithm is 82. 4% on the NEU-DET dataset, which is 5. 3% higher than the original YOLOv5s algorithm, and the detection speed reaches 65. 4 fps, which achieves a better balance between detection speed and detection accuracy.

    参考文献
    相似文献
    引证文献
引用本文

马燕婷,赵红东,阎 超,封慧杰,于快快,刘 赫.改进 YOLOv5 网络的带钢表面缺陷检测方法[J].电子测量与仪器学报,2022,36(8):150-157

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2023-03-06
  • 出版日期:
文章二维码