• Laser & Optoelectronics Progress
  • Vol. 58, Issue 24, 2415007 (2021)
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
    DOI: 10.3788/LOP202158.2415007 Cite this Article Set citation alerts
    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 Copy Citation Text show less

    Abstract

    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.
    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