基于改进YOLOv5的金属焊缝缺陷检测
DOI:
CSTR:
作者:
作者单位:

青岛科技大学自动化与电子工程学院 青岛市 266061

作者简介:

通讯作者:

中图分类号:

TP391.41; TG441.7

基金项目:

山东省重大科技创新工程(2021SFGC0601)


Metal weld defect detection based on improved YOLOv5
Author:
Affiliation:

College of automation and electronic engineering, Qingdao University of science and technology, Qingdao 266061, China

Fund Project:

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

    为提高工业上焊缝缺陷自动检测与处理的效率,基于深度学习提出一种改进的YOLOv5焊缝缺陷检测方法。针对焊缝样本数据不足的问题,提出一种Mosaic+Mixup的数据增强策略,同时为减少网络的计算量和网络参数量,引入轻量型的GhostNet网络代替YOLOv5主干网络中CSP1模块中的残差模块,并且采用CIOU_Loss作为坐标位置损失提高算法的收敛速度与准确率。最后使用测试集进行焊缝缺陷检测,改进的YOLOv5的平均精度均值(mean Average Precision,mAP)达到96.88%,单张图片检测时间不超过50毫秒,优于传统机器学习算法,能够满足实际工程中对焊缝缺陷的实时性检测要求。

    Abstract:

    An improved YOLOv5 weld defect detection method based on deep learning was proposed in response to improve the efficiency of automatic detection and processing of weld defects in industry. Aiming at the insufficient weld sample data, a mosaic+mixup data augmentation strategy was proposed. At the same time, in order to reduce the amount of calculation and parameters of network, a lightweight GhostNet network was introduced to replace the residual module in CSP1 module in YOLOv5 backbone network, and CIOU_Loss was used as the coordinate position loss to improve the convergence rate and accuracy of the algorithm. Finally, the testing set was used for weld defect detection. The improved YOLOv5 has a mean average precision (mAP) of 96.88%, and the detection time of a single image is no more than 50 milliseconds, which is better than the traditional machine learning algorithms, could meet the real-time detection requirements of weld defects in practical engineering.

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

李衍照,于镭,田金文.基于改进YOLOv5的金属焊缝缺陷检测[J].电子测量技术,2022,45(19):70-75

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