Surface Defect Detection of Metal Workpiece Based on Improved YOLOv5

WANG Yi, GONG Xiao-jie, CHENG Jia, SU Hao

Packaging Engineering ›› 2022 ›› Issue (15) : 54-60.

PDF(1751 KB)
PDF(1751 KB)
Packaging Engineering ›› 2022 ›› Issue (15) : 54-60. DOI: 10.19554/j.cnki.1001-3563.2022.15.006

Surface Defect Detection of Metal Workpiece Based on Improved YOLOv5

  • WANG Yi1, GONG Xiao-jie2, CHENG Jia2, SU Hao3
Author information +
History +

Abstract

The work aims to propose a surface defect detection method based on YOLOv5 network by combining attention mechanism and Ghost convolution to solve problem of low detection accuracy of small size defects on metal workpiece surface. First, the SE channel attention module was added to the original network. The weight of the defect-related information was increased and the interference of useless features was reduced to improve the detection accuracy of the target. Then, the maxpool module of the spatial pyramid pooling module in the network was replaced with Softpool so as to retain more feature information in the down sampling activation map and obtain a better classification accuracy. Finally, Ghost convolutional blocks were used to replace the conventional convolutional modules in the backbone network to extract rich and redundant features and improve the efficiency of the model. The mean average accuracy of the improved network reached 0.997 8, increased by 7.07% over the original network. The proposed network significantly improves the accuracy of surface defect detection in metal workpieces.

Cite this article

Download Citations
WANG Yi, GONG Xiao-jie, CHENG Jia, SU Hao. Surface Defect Detection of Metal Workpiece Based on Improved YOLOv5[J]. Packaging Engineering. 2022(15): 54-60 https://doi.org/10.19554/j.cnki.1001-3563.2022.15.006
PDF(1751 KB)

Accesses

Citation

Detail

Sections
Recommended

/