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

Lianshan Sun; Jingxue Wei; Dengming Zhu; Min Shi

doi:10.3788/LOP202158.2415007

[1] Tang B, Kong J Y, Wu S Q. Review of surface defect detection based on machine vision[J]. Journal of Image and Graphics, 22, 1640-1663(2017).

[2] Li C X. Research on surface defect detection algorithm of aluminum profile based on machine learning[D], 21-34(2020).

[3] Tao X, Hou W, Xu D. A survey of surface defect detection methods based on deep learning[J]. Acta Automatica Sinica, 47, 1017-1034(2021).

[4] Tabernik D, Šela S, Skvarč J et al. Segmentation-based deep-learning approach for surface-defect detection[J]. Journal of Intelligent Manufacturing, 31, 759-776(2020).

[5] Yan W X. Research on deep learning and its application on the casting defects automatic detection[D], 23-37(2016).

[6] Girshick R, Donahue J, Darrell T et al. Rich feature hierarchies for accurate object detection and semantic segmentation[C]. //2014 IEEE Conference on Computer Vision and Pattern Recognition, June 23-28, 2014, Columbus, OH, USA., 580-587(2014).

[7] Girshick R. Fast R-CNN[C]. //2015 IEEE International Conference on Computer Vision (ICCV), December 7-13, 2015, Santiago, Chile., 1440-1448(2015).

[8] Ren S Q, He K M, Girshick R et al. Faster R-CNN: towards real-time object detection with region proposal networks[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 39, 1137-1149(2017).

[9] Liu W, Anguelov D, Erhan D et al. SSD: single shot MultiBox detector[M]. //Leibe B, Matas J, Sebe N, et al. Computer vision-ECCV 2016. Lecture notes in computer science, 9905, 21-37(2016).

[10] Redmon J, Divvala S, Girshick R et al. You only look once: unified, real-time object detection[C]. //2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), June 27-30, 2016, Las Vegas, NV, USA, 779-788(2016).

[11] Redmon J, Farhadi A. YOLOv3: an incremental improvement[EB/OL]. (2018-04-08)[2021-03-20]. https://arxiv.org/abs/1804.02767

[12] Bochkovskiy A, Wang C Y, Liao HY Mark. YOLOv4: optimal speed and accuracy of object detection[EB/OL]. (2020-04-23)[2021-03-20]. https://arxiv.org/abs/2004.10934

[13] Glenn J, Alex S, Jirka B et al. ultralytics: yolov5[DB/OL]. (2020-08-13)[2021-03-20]. http://doi.org/10.5281/zenodo.3983579

[14] Law H, Deng J. CornerNet: detecting objects as paired keypoints[J]. International Journal of Computer Vision, 128, 642-656(2020).

[15] Zhou X Y, Wang D Q, Krähenbühl P. Objects as points[EB/OL]. (2019-04-16)[2021-03-20]. https: //arxiv.org/abs/1904.07850

[16] Tian Z, Shen C H, Chen H et al. FCOS: fully convolutional one-stage object detection[C]. //2019 IEEE/CVF International Conference on Computer Vision (ICCV), October 27-November 2, 2019, Seoul, Korea, 9626-9635(2019).

[17] Li D J, Li R H. Mug defect detection method based on improved Faster RCNN[J]. Laser & Optoelectronics Progress, 57, 041515(2020).

[18] Lin T Y, Dollár P, Girshick R et al. Feature pyramid networks for object detection[C]. //2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), July 21-26, 2017, Honolulu, HI, USA., 936-944(2017).

[19] Li Y T, Huang H S, Xie Q S et al. Research on a surface defect detection algorithm based on MobileNet-SSD[J]. Applied Sciences, 8, 1678(2018).

[20] Wen S P, Li C X. Surface defect detection of aluminum profile based on Gaussian-Yolov3[J]. Computer Measurement & Control, 28, 88-93(2020).

[21] Zhou T Y, Zhu Q B, Huang M et al. Defect detection of chip on carrier based on improved YOLOV3[J]. Laser & Optoelectronics Progress, 58, 1210002(2021).

[22] Wang X, Kang S, Zhu W D et al. Defect detection of laminated surface in the automated fiber placement process based on improved CenterNet[J]. Infrared and Laser Engineering, 50, 20210011(2021).

[23] Hu J, Shen L, Albanie S et al. Squeeze-and-Excitation networks[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 42, 2011-2023(2020).

[24] Yu F, Koltun V. Multi-scale context aggregation by dilated convolutions[EB/OL]. (2015-11-23)[2021-03-25]. https://arxiv.org/abs/1511.07122

[25] Shen X H, Li Z H, Li M et al. Aluminum surface-defect detection based on multi-task deep learning[J]. Laser & Optoelectronics Progress, 57, 101501(2020).

[27] Wu Y, Chen Y P, Yuan L et al. Rethinking classification and localization for object detection[C]. //2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), June 13-19, 2020, Seattle, WA, USA., 10183-10192(2020).

[28] Ghiasi G, Lin T Y, Le Q V et al. NAS-FPN: learning scalable feature pyramid architecture for object detection[C]. //2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition(CVPR), June 15-20, 2019, Long Beach, CA, USA, 7029-7038(2020).

[29] Dong Z W, Li G X, Liao Y et al. CentripetalNet: pursuing high-quality keypoint pairs for object detection[C]. //2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), June 13-19, 2020, Seattle, WA, USA, 10516-10525(2020).