基于改进YOLOv4算法的玻璃杯缺陷识别方法研究
DOI:
CSTR:
作者:
作者单位:

1.北京信息科技大学高动态导航技术北京市重点实验室 北京 100101; 2.沧州格锐特钻头有限公司 沧州 062450

作者简介:

通讯作者:

中图分类号:

TP399

基金项目:

国家自然科学基金青年项目(52104001)、北京信息科技大学重点研究培育项目(2121YJPY220)、北京市教育委员会科学研究计划项目(KM202111232004)、中国石油天然气集团有限公司-中国石油大学(北京)战略合作科技专项(ZLZX2020-03)资助


Research on glass defect recognition method based on improved YOLOv4
Author:
Affiliation:

1.Beijing Key Laboratory of High Dynamic Navigation Technology,Beijing Information Science & Technology University, Beijing 100101,China; 2.Cangzhou Geruite bit Co., Ltd.,Cangzhou 062450,China

Fund Project:

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

    当前,玻璃杯表面缺陷检测主要依赖人力劳动来完成,存在耗时长且准确率不高等问题。提出了一种将YOLOv4与MobileNetV3结合的改进算法模型YOLO-M来解决该问题。首先,利用MobileNetv3网络替换YOLOv4原本的主干网络CSPDarknet53,并修改激活函数,在减少模型大小和参数量的基础上提升运行速度。然后,对玻璃杯缺陷样本进行拍照采样,将缺陷分为磨损、气泡、划痕三种,建立玻璃杯缺陷数据集。最后利用YOLO-M、YOLOv4以及YOLOv4-tiny三种算法对玻璃杯缺陷数据集进行训练,将不同算法下的平均精度均值、帧率等评价指标进行对比。实验结果表明,YOLO-M算法在玻璃杯缺陷检测上的帧率达到57.72 f/s,平均精度均值达到91.95%,均为最高。YOLO-M算法在玻璃杯缺陷识别的速度和精度上有明显效果,可做为后续分拣研究,以及其他玻璃制品缺陷识别的重要参考。

    Abstract:

    At present, glass surface defect detection is mainly manual, which takes a long time and has low accuracy. An improved algorithm model YOLO-M combining YOLOv4 and mobilenetv3 is proposed to solve this problem. Firstly, the mobilenetv3 network is used to replace the original backbone network cspdarknet53 of YOLOv4, and the activation function is modified to improve the running speed on the basis of reducing the model size and parameters. Then, the glass defect samples were photographed and sampled. The defects were divided into wear, bubble and scratch, and the glass defect data set was established. Finally, the glass defect data set is trained by using YOLO-M, YOLOv4 and YOLOv4 tiny algorithms, and the evaluation indexes such as average precision and frame rate under different algorithms are compared. The experimental results show that the frame rate of YOLO-M algorithm in glass defect detection is 57.72 f/s, and the average accuracy is 91.95%. YOLO-M algorithm has obvious effect on the speed and accuracy of glass defect recognition, and can be used as an important reference for subsequent sorting research and other glass product defect recognition.

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

张涛,谢探阳,李玉梅,白俊华.基于改进YOLOv4算法的玻璃杯缺陷识别方法研究[J].电子测量技术,2023,46(2):46-51

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