Journal of Jilin University Science Edition ›› 2023, Vol. 61 ›› Issue (3): 612-622.
Previous Articles Next Articles
LIU Haohan, SUN Cheng, HE Huaiqing, HUI Kanghua
Received:
Online:
Published:
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:
LIU Haohan, SUN Cheng, HE Huaiqing, HUI Kanghua. Metal Surface Defect Detection Method YOLOv3I[J].Journal of Jilin University Science Edition, 2023, 61(3): 612-622.
0 / / Recommend
Add to citation manager EndNote|Reference Manager|ProCite|BibTeX|RefWorks
URL: http://xuebao.jlu.edu.cn/lxb/EN/
http://xuebao.jlu.edu.cn/lxb/EN/Y2023/V61/I3/612
Cited