Plastic Part Surface Defect Detection Based on Improved YOLOv7
To address the issues of low efficiency and poor accuracy in plastic part surface defect detection models, this paper proposes an improved YOLOv7-based detection algorithm, YOLOv7-TBC. First, a T-shaped convolutional module is introduced into the backbone network to enhance feature extraction capability and detection speed. Then, a weighted bidirectional feature pyramid network is applied at the neck to improve feature fusion efficiency and capacity. Finally, a convolutional block attention module is embedded to enhance the model's performance in detecting small objects. Experimental results show that the proposed algorithm achieves a mean average precision (mAP) of 97.4% on the plastic part dataset, representing a 1.5% improvement over the original model. Additionally, model parameters are reduced by 21.3%, and computational load is decreased by 17.4<sup xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">%</sup>, significantly improving detection speed and accuracy, effectively meeting industrial defect detection requirements.
Authors
- Yan Piao
- Jingjie Dong (ORCID: https://orcid.org/0009-0005-3655-4237)
- Yu Wang (ORCID: https://orcid.org/0000-0003-2883-1087)
Institutions
- Changchun University of Science and Technology (CN)
Publication Details
- Published
- 2024-09-27
- DOI
- https://doi.org/10.1109/eiecs63941.2024.10800512
- Primary Topic
- Industrial Vision Systems and Defect Detection
- Type
- article
- Field-Weighted Citation Impact
- 0.00