卷积(计算机科学)
计算机科学
人工神经网络
人工智能
卷积神经网络
材料科学
模式识别(心理学)
作者
Haijian Wang,Mo Han,Shun Lu,Xuemei Zhao
标识
DOI:10.1016/j.jksuci.2024.101935
摘要
The existing methods for detecting surface defects in electrolytic capacitors are typically based on conventional machine vision, with limited feature extraction capabilities, poor versatility, slow detection speed, and the inability to achieve accurate and real-time defect detection. In this study, a real-time object detection algorithm based on an improved single shot multibox detector (SSD) is proposed to achieve omnidirectional surface defect detection of electrolytic capacitors. First, an electrolytic capacitor surface image acquisition device was established to capture omnidirectional surface images of the capacitors, and an electrolytic capacitor surface defect dataset was created. Next, the visual geometry group (VGG)-16 network structure was replaced with the MobileNetv2 network structure, effectively reducing the model’s parameter count and improving inference speed. Moreover, the Multibox Loss function was replaced with the Focal Loss function to increase the model’s attention to difficult-to-classify samples and improve model accuracy. Additionally, a transfer learning network model was designed to apply the model to electrolytic capacitors of different colors using small sample learning. Finally, the performance of the improved network model was tested on a dataset of electrolytic capacitor surface defects. The experimental results demonstrate that the parameters quantity of improved model is 3.50M, the mAP value reaches 92.67%, which is improved by 2.54%, and the Macro-F1 value reaches 92.15%, which is 11.32% higher than that before improvement. Thus, the proposed improved SSD model provides a theoretical basis and technical prerequisites for automated and intelligent surface defect detection in electrolytic capacitors.
科研通智能强力驱动
Strongly Powered by AbleSci AI