Laser & Optoelectronics Progress, Volume. 58, Issue 14, 1415004(2021)

Surface Defect Detection of Aeroengine Components Based on Improved YOLOv4 Algorithm

Bin Li, Cheng Wang*, Jing Wu, Jichao Liu, Lijia Tong, and Zhenping Guo
Author Affiliations
  • Fundamentals Department, Air Force Engineering University, Xi’an, Shaanxi 710038, China
  • show less

    Aiming at solving the problems of low accuracy and slow speed in the surface defects detection of aeroengine components using traditional methods, an improved YOLOv4 algorithm is proposed herein. First, an aeroengine component surface defects dataset was developed and the K-means clustering algorithm was suggested to cluster the defect samples for obtaining the priori anchor’s parameters of different sizes. Second, the improved parameter-adjustment algorithm was used to scale the priori anchor’s sizes and increase the difference in sizes to improve the matching rate between priori anchors and feature maps. Finally, convolution layers were added after the different feature layers of the backbone feature extraction network output and spatial pyramid pooling structure to improve the ability of network to extract defect features. Experimental results show that the mean average precision (mAP) value of the improved YOLOv4 algorithm in the test dataset is as high as 82.67%, which is 4.55 percent point greater than that of the original YOLOv4 algorithm. The average detection time of a single image is 0.1240 s, which is basically the same as that of the original algorithm. Moreover, the detection performance is better than Faster R-CNN and YOLOv3.

    Tools

    Get Citation

    Copy Citation Text

    Bin Li, Cheng Wang, Jing Wu, Jichao Liu, Lijia Tong, Zhenping Guo. Surface Defect Detection of Aeroengine Components Based on Improved YOLOv4 Algorithm[J]. Laser & Optoelectronics Progress, 2021, 58(14): 1415004

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Machine Vision

    Received: Nov. 5, 2020

    Accepted: Nov. 18, 2020

    Published Online: Jul. 8, 2021

    The Author Email: Wang Cheng (warrant_74@163.com)

    DOI:10.3788/LOP202158.1415004

    Topics