Journal of Jilin University Science Edition ›› 2023, Vol. 61 ›› Issue (3): 612-622.

Previous Articles     Next Articles

Metal Surface Defect Detection Method YOLOv3I

LIU Haohan, SUN Cheng, HE Huaiqing, HUI Kanghua   

  1. College of Computer Science and Technology, Civil Aviation University of China, Tianjin 300300, China
  • Received:2022-04-06 Online:2023-05-26 Published:2023-05-26

Abstract: We proposed an improved model of metal surface defect detection method. Firstly,  based on the YOLOv3(you only look once v3) object detection model, a multi-scale convolution parallel structure was used to extract and fuse multi-scale features. Secondly, efficient downsampling was used to maintain the feature information and reduce the computation caused by feature dimension raising. Finally, spatial separable convolution was used to  increase the width and depth of the model while keeping the receptive field unchanged, so that an  improved model YOLOv3I (you only look once v3 inception) with  reduced the amount of model parameters and improved  the performance of the model was obtained. The improved model improved the feature  extraction ability for  complex defects and further reduced the requirements for hardware configuration. The experimental results show that the improved model has significantly improved both accuracy  and calculation efficiency, with an  average accuracy  increase of  about 5% on the public dataset, and about 3% on the bearing dataset provided by the enterprise. The amount of model parameters decreases by more than 20%, and the  floating point computation of the model  reduces by 1.6×109 and 1.2×1010 times on both two datasets respectively.

Key words: defect detection, feature extraction, multi-scale convolution parallel structure, spatial separable convolution, downsampling

CLC Number: 

  • TP391.41