MCD-YOLO: an efficient model for detecting surface defects of electroplated safety buckles
摘要
Aiming at the problems of low detection accuracy, large parameters, and slow model detection speed in the surface defect detection of electroplated safety buckles, in this paper proposes a MCD-YOLO lightweight model. First, we used the lightweight MobileNetV3 as the backbone network to reduce model parameters and computational complexity. SPPELAN-MDCR replaces the SPPF module, which can better maintain the sensitivity to defect features and improve the feature extraction ability of the model. Second, the DDFPN feature fusion network is designed to enhance the ability of feature fusion at different levels, and the ADDC module and down-sampling module are used to enhance the ability of the network to extract the feature information in the image, reduce the calculation of the network, and then improve the detection speed and accuracy of the model. Finally, we conducted experiments on the self-built DAQK dataset and the public NEU-DET dataset. Compared with the original model, the mAP increased by 6.8% and 3.2%, FPS increased by 14.4%, FLOPs decreased by 9.9% and 15.1%, and parameters decreased by 5.3% and 10.6%, respectively. The experimental results show the model realizes the balance between lightweight and detection accuracy, and has real-time and efficient practical value in surface defect detection.