AUTHOR=Wang Yuanyuan , Yan Shaofeng , Abdullahi Hauwa Suleiman , Gao Shangbing , Zhang Haiyan , Chen Xiuchuan , Zhao Hu
TITLE=Multiclass small target detection algorithm for surface defects of chemicals special steel
JOURNAL=Frontiers in Physics
VOLUME=12
YEAR=2024
URL=https://www.frontiersin.org/journals/physics/articles/10.3389/fphy.2024.1451165
DOI=10.3389/fphy.2024.1451165
ISSN=2296-424X
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.