Laser & Optoelectronics Progress, Volume. 58, Issue 24, 2415007(2021)

Surface Defect Detection Algorithm of Aluminum Profile Based on AM-YOLOv3 Model

Lianshan Sun1, Jingxue Wei1、*, Dengming Zhu2, and Min Shi3
Author Affiliations
  • 1School of Electronic Information and Artificial Intelligence, Shaanxi University of Science and Technology, Xi'an, Shaanxi 710021, China
  • 2Foresight Research Laboratory, Institute of Computing Technology, Chinese Academy of Sciences, Beijing 100080, China
  • 3School of Control and Computer Engineering, North China Electric Power University, Beijing 102206, China
  • show less

    In order to solve the problem that the accuracy of defect detection algorithm is not high and it is difficult to meet the actual requirements due to the complex surface texture of aluminum profile and the large difference in defect size, an improved object detection network AM-YOLOv3 (attention-guided multi-scale fusion YOLOv3) is proposed. The attention guide module and four prediction scales are designed to realize the multi-scale feature extraction of aluminum profile surface defects. A bottom-up feature transmission path is constructed, which is combined with the original feature pyramid network to form a twin-towers structure, and the multi-scale feature fusion is realized. K-medians algorithm is used for anchor box clustering, which more accurately characterizes the distribution law of anchor frame size and improves the convergence speed of the network. The performance of the proposed algorithm is verified by experiments on the public aluminum profile dataset. The experimental results show that the mAP (mean average precision) of the proposed algorithm reaches 99.05%, which is 6.8% higher than that of the YOLOv3 model, and the frame rate reaches 43.94 frame/s.

    Tools

    Get Citation

    Copy Citation Text

    Lianshan Sun, Jingxue Wei, Dengming Zhu, Min Shi. Surface Defect Detection Algorithm of Aluminum Profile Based on AM-YOLOv3 Model[J]. Laser & Optoelectronics Progress, 2021, 58(24): 2415007

    Download Citation

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

    Category: Machine Vision

    Received: Mar. 29, 2021

    Accepted: Jun. 2, 2021

    Published Online: Dec. 1, 2021

    The Author Email: Wei Jingxue (1779664651@qq.com)

    DOI:10.3788/LOP202158.2415007

    Topics