基于轻量化YOLOv3的带钢表面缺陷检测方法
DOI:
CSTR:
作者:
作者单位:

(1.天津工业大学 机械工程学院,天津 300387; 2.天津市现代机电装备技术重点实验室,天津 300387)

作者简介:

刘国华 (1969-),男,硕士,教授,硕士生导师,主要从事计算机视觉与图像处理方面的研究.

通讯作者:

中图分类号:

基金项目:

天津市科技计划项目(21YFFCYS00080)资助项目


Surface defect detection method of strip steel based on lightweight YOLOv3
Author:
Affiliation:

(1.School of Mechanical Engineering, Tiangong University, Tianjin 300387, China;2.Advanced Mechatronics Equipment Technology Tianjin Major Laboratory, Tianjin 300387, China)

Fund Project:

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

    缺陷检测是带钢生产过程中不可缺少的工序,现有检测方法普遍存在检测精度较低、实时性差等问题。为解决上述问题,本文提出了一种基于轻量化YOLOv3的快速缺陷检测方法。MobileNetv2作为主干网络并用两个尺度的特征图进行输出,保证了网络模型的轻量化;将改进后的注意力模块融合进特征金字塔网络(feature pyramid network,FPN),同时结合空间金字塔池化模块(spatial pyramid pooling,SPP),以提高算法对缺陷的学习能力;使用K均值聚类算法获得更优的先验框,并且使用CIoU(complete-intersection over union)对损失函数进行优化,进一步提升网络性能。提出的方法在带钢缺陷数据集上检测速度为70.8 FPS;模型参数量为7.1 MB,仅为YOLOv3的3.02%。实验结果表明本文所提方法能够在保证精度的同时实现对缺陷的快速检测,具有良好的生产线部署能力。

    Abstract:

    Defect detection is an indispensable process in the strip steel production process,and existing inspection methods generally have problems such as low detection accuracy and poor real-time performance.To solve the above problems,a fast defect detection method based on lightweight YOLOv3 is proposed in this paper.MobileNetv2 is used as the backbone network and output with two scales of feature maps,so that the lightweight of the network model is guaranteed;the improved attention module is fused into the feature pyramid network (FPN) and the network is combined with the spatial pyramid pooling (SPP) to improve the learning ability of the algorithm for defects;the K-means mean clustering algorithm is used to obtain a better anchor box,and the complete-intersection over union (CIoU) is used to optimize the loss function to further improve the network performance.The proposed method has a detection speed of 70.8 FPS on the strip steel defect dataset; the number of model parameters is 7.1 MB,which is only 3.02% of YOLOv3.Experiments show that the proposed method can achieve rapid detection of defects while ensuring accuracy,and has good production line deployment capabilities.

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

马千文,刘国华.基于轻量化YOLOv3的带钢表面缺陷检测方法[J].光电子激光,2023,34(7):743~751

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