基于改进YOLOv4的电机端盖缺陷检测
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家自然科学基金(51405229); 江苏省自然科学基金(BK20151470)


Defect Detection of Motor Cover Based on Improved YOLOv4
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 增强出版
  • |
  • 文章评论
    摘要:

    在基于机器视觉实现电机端盖裂纹缺陷检测过程中, 针对复杂背景下目标特征不明显的问题, 使用限制对比度的自适应直方图均衡化的方法加强目标特征. 针对机器视觉系统中训练数据量少且训练图片背景单一导致模型泛化性低的问题, 对比了Mosaic和CutMix数据增强方法, 并结合多种数据增强策略, 提出了系统的数据集构建方案. 针对使用YOLOv4进行单类检测和小目标检测时正负样本不平衡导致检测率低的问题, 提出了自适应多尺度焦点损失+CIoU损失的加权融合损失函数, 并通过实验得到最优超参数. 最后使用K-means算法初始化锚点框, 使模型更适应线状目标的预测. 结果表明, 该方法对于裂纹类别的检测达到了95.8%的平均精度(Average Precision, AP), 相较于改进前提升9.7%, 单张检测时间48 ms, 具有一定的工程应用价值.

    Abstract:

    Adaptive histogram equalization with limited contrast is applied to strengthening the target feature to solve the problem of unclear targets in a complex background during crack detection of motor covers based on machine vision. A systematic dataset construction scheme is proposed by comparing Mosaic and CutMix data augmentation and combining with a variety of data enhancement techniques to address the low generalization of the model induced by the small volume of training data in the machine vision system and single background of training pictures. Besides, a weighted fusion loss function combined with adaptive multi-scale focus loss and CIoU loss is proposed to deal with the low detection rate caused by unbalanced numbers of positive and negative samples in the single class detection and small target detection of YOLOv4, and the optimal hyper parameters are obtained through experiments. Finally, the anchor box is initialized by the K-means algorithm to make the model more suitable for predicting linear targets. Results demonstrate that this method achieves an Average Precision (AP) of 95.8% for detecting crack types, which is 9.7% higher than before, and the single-sheet detection time is 48 ms, presenting the potential for engineering application.

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

万卓,叶明,刘凯.基于改进YOLOv4的电机端盖缺陷检测.计算机系统应用,2021,30(3):79-87

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2020-07-06
  • 最后修改日期:2020-08-11
  • 录用日期:
  • 在线发布日期: 2021-03-06
  • 出版日期:
您是第位访问者
版权所有:中国科学院软件研究所 京ICP备05046678号-3
地址:北京海淀区中关村南四街4号 中科院软件园区 7号楼305房间,邮政编码:100190
电话:010-62661041 传真: Email:csa (a) iscas.ac.cn
技术支持:北京勤云科技发展有限公司

京公网安备 11040202500063号