Back to Search Start Over

An Improved Feature Enhancement CenterNet Model for Small Object Defect Detection on Metal Surfaces.

An Improved Feature Enhancement CenterNet Model for Small Object Defect Detection on Metal Surfaces.

Authors :
Zhu, Xingfei
Wang, Qimeng
Zhang, Bufan
Sun, Zhaofei
Yu, Jinghu
Qian, Shanhua
Source :
Advanced Theory & Simulations. Aug2024, Vol. 7 Issue 8, p1-10. 10p.
Publication Year :
2024

Abstract

In defect detection on metal surfaces, there are many small defects with subtle features that are difficult to distinguish from the background environment using mainstream object detection methods. To alleviate this issue, this study proposes an improved CenterNet model for enhancing the features of small defects on metal surfaces, namely MSDD. In this work, we utilize an attention mechanism to reconstruct the basic feature extraction module in the metal defect feature extraction network, aiming to enhance the focus on features related to small defects. Additionally, we redesign an efficient deconvolution module to extract multi‐scale defect feature information, capturing details of small defects at different scales. Finally, leveraging the idea of reparameterization to enhance feature representation capabilities, we optimize the output of the detection head, thereby strengthening the model's ability to capture features of small objects on the metal surface. On the NEU‐DET dataset, compared to the CenterNet baseline, the improved feature enhancement network shows a 4.9% improvement. Furthermore, with an accuracy of 80.2% compared to mainstream object detection methods, it outperforms other state‐of‐the‐art (SOTA) level object detection methods, significantly enhancing the detection accuracy and performance of the network model. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
25130390
Volume :
7
Issue :
8
Database :
Academic Search Index
Journal :
Advanced Theory & Simulations
Publication Type :
Academic Journal
Accession number :
178973139
Full Text :
https://doi.org/10.1002/adts.202301230