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

Clc Number:

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    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.

    Reference
    Related
    Cited by
Get Citation

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

Copy
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:July 06,2020
  • Revised:August 11,2020
  • Adopted:
  • Online: March 06,2021
  • Published:
Article QR Code
You are the firstVisitors
Copyright: Institute of Software, Chinese Academy of Sciences Beijing ICP No. 05046678-3
Address:4# South Fourth Street, Zhongguancun,Haidian, Beijing,Postal Code:100190
Phone:010-62661041 Fax: Email:csa (a) iscas.ac.cn
Technical Support:Beijing Qinyun Technology Development Co., Ltd.

Beijing Public Network Security No. 11040202500063