Back to Search Start Over

Multiclass small target detection algorithm for surface defects of chemicals special steel.

Authors :
Wang, Yuanyuan
Yan, Shaofeng
Abdullahi, Hauwa Suleiman
Gao, Shangbing
Zhang, Haiyan
Chen, Xiuchuan
Zhao, Hu
Liu, Haotian
Iqbal, Imran
Source :
Frontiers in Physics; 2024, p1-17, 17p
Publication Year :
2024

Abstract

Introduction: Chemical special steels are widely used in chemical equipment manufacturing and other fields, and small defects on its surface (such as cracks and punches) are easy to cause serious accidents in harsh environments. Methods: In order to solve this problem, this paper proposes an improved defect detection algorithm for chemical special steel based on YOLOv8. Firstly, in order to effectively capture local and global information, a ParC2Net (Parallel-C2f) structure is proposed for feature extraction, which can accurately capture the subtle features of steel defects. Secondly, the loss function is adjusted to MPD-IOU, and its dynamic non-monotonic focusing characteristics are used to effectively solve the overfitting problem of the bounding box of low-quality targets. In addition, RepGFPN is used to fuse multi-scale features, deepen the interaction between semantics and spatial information, and significantly improve the efficiency of cross-layer information transmission. Finally, the RexSE-Head (ResNeXt-Squeeze-Excitation) design is adopted to enhance the positioning accuracy of small defect targets. Results and discussion: The experimental results show that the mAP@0.5 of the improved model reaches 93.5%, and the number of parameters is only 3.29M, which realizes the high precision and high response performance of the detection of small defects in chemical special steels, and highlights the practical application value of the model. The code is available at https://github.com/improvment/prs-yolo. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
2296424X
Database :
Complementary Index
Journal :
Frontiers in Physics
Publication Type :
Academic Journal
Accession number :
179659153
Full Text :
https://doi.org/10.3389/fphy.2024.1451165