Abstract
Magnetic rings are widely used in automotive, home appliances, and consumer electronics. Due to the materials used, processing techniques, and other factors, there will be top cracks, internal cracks, adhesion, and other defects on individual magnetic rings during the manufacturing process. To find such defects, the most sophisticated YOLOv5 target identification algorithm is frequently utilized. However, it has problems such as high computation, sluggish detection, and a large model size. This work suggests an enhanced lightweight YOLOv5 (MR-YOLO) approach for the identification of magnetic ring surface defects to address these issues. To decrease the floating-point operation (FLOP) in the feature channel fusion process and enhance the performance of feature expression, the YOLOv5 neck network was added to the Mobilenetv3 module. To improve the robustness of the algorithm, a Mosaic data enhancement technique was applied. Moreover, in order to increase the network’s interest in minor defects, the SE attention module is inserted into the backbone network to replace the SPPF module with substantially more calculations. Finally, to further increase the new network’s accuracy and training speed, we substituted the original CIoU-Ioss for SIoU-Loss. According to the test, the FLOP and Params of the modified network model decreased by 59.4% and 47.9%, respectively; the reasoning speed increased by 16.6%, the model’s size decreased by 48.1%, and the mAP only lost by 0.3%. The effectiveness and superiority of this method are proved by an analysis and comparison of examples.
Funder
National key research and development program of China
Subject
Electrical and Electronic Engineering,Biochemistry,Instrumentation,Atomic and Molecular Physics, and Optics,Analytical Chemistry
Reference45 articles.
1. Study on surface defect extraction of magnetic ring based on masking technology;Li;J. Optoelectron.,2017
2. Posilovic, L., Medak, D., Subasic, M., Petkovic, T., Budimir, M., and Loncaric, S. (2019, January 23–25). Flaw Detection from Ultrasonic Images Using YOLO and SSD. Proceedings of the 2019 11th International Symposium on Image and Signal Processing and Analysis (ISPA), Dubrovnik, Croatia.
3. Suszyński, Z., Bednarek, M., and Kosikowski, M. (2010, January 6–8). Method and Measurement Setup for Detection of the Shape of Optical Power Pulse in Active Thermography. Proceedings of the 2010 16th International Workshop on Thermal Investigations of ICs and Systems (THERMINIC), Barcelona, Spain.
4. Machine Learning-Based Image Processing for on-Line Defect Recognition in Additive Manufacturing;Caggiano;Cirp Ann. -Manuf. Technol.,2019
5. Deep Features Based on a DCNN Model for Classifying Imbalanced Weld Flaw Types;Hou;Measurement,2019
Cited by
19 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献