PO-YOLOv5: A defect detection model for solenoid connector based on YOLOv5

Solenoid connectors play important role in electronic stability system design, with the features of small size, low cost, fast response time and high reliability. The main production process challenge for solenoid connectors is the accurate detection of defects, which is closely related to safe driv...

Full description

Saved in:
Bibliographic Details
Published in:PloS one 2024-01, Vol.19 (1), p.e0297059
Main Authors: Chen, Ming, Liu, Yuqing, Wei, Xing, Zhang, Zichen, Gaidai, Oleg, Sui, Hengshou, Li, Bin
Format: Article
Language:English
Subjects:
Accuracy
Algorithms
Aluminum
Classification
Connectors
Convolution
Deep learning
Defects
Electronics
Kalman filters
Learning
Management
Manufacturing
Manufacturing industry
Methods
Object recognition
Proposals
Reaction Time
Reproducibility of Results
Semantics
Sensors
Solenoids
Support vector machines
Systems design
Vision systems
Citations: Items that this one cites
Online Access:Get full text
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:Solenoid connectors play important role in electronic stability system design, with the features of small size, low cost, fast response time and high reliability. The main production process challenge for solenoid connectors is the accurate detection of defects, which is closely related to safe driving. Both faultless and defective products have similar color and shape at the defect location, making proper inspection challenging. To address these issues, we proposed a defect detection model called PO-YOLOv5 to achieve accurate defect detection for solenoid connectors. First, an additional prediction head was added to enable the model to acquire more semantic information to detect larger-scale defective features. Second, we introduced dynamic convolution to learn complementary connections between the four dimensions of the convolution kernel by utilizing its multidimensional attention mechanism. Replacing conventional convolution with dynamic convolution enhances the detection accuracy of the model and reduces the inference time. Finally, we validated PO-YOLOv5 versus the state-of-the-art object detection methods on the same solenoid connectors dataset. Experiments revealed that our proposed approach exhibited higher accuracy. The mAP (mean Average Precision) result of PO-YOLOv5 was found to be about 90.1%. Compared with the original YOLOv5, PO-YOLOv5 exhibited improved precision by about 3%.
ISSN:1932-6203
1932-6203
DOI:10.1371/journal.pone.0297059