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

Xingfei Zhu,Qimeng Wang,Bufan Zhang,Zhaofei Sun,Jinghu Yu,Shanhua Qian
DOI: https://doi.org/10.1002/adts.202301230
2024-06-12
Advanced Theory and Simulations
Abstract:This study proposes the MSDD model, an improved version of CenterNet, featuring an attention mechanism for better feature extraction, a multi‐scale inverse convolution module for detecting small defects at various scales, and an optimized detection head using reparameterization to enhance detection performance on metal surfaces. 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.
multidisciplinary sciences
What problem does this paper attempt to address?