Development of a Cross-scale Connection Network With Gather-and-distribute Structure for Steel Surface Defect Detection
摘要
Steel defects can significantly diminish the corrosion resistance, wear resistance, and load-bearing capacity of the steel, leading to substantial financial losses. To effectively identify and locate steel surface defects, we propose a cross-scale feature fusion network. The process begins with the pre-processing of the input image through gray transformation and histogram equalization, followed by feature extraction using an enhanced backbone feature extraction network. Subsequently, a feature fusion network incorporating a gather-and-distribute (GD) structure is introduced to merge multi-scale feature maps, improving the robustness of information fusion across different scales. In the final stage, three detection heads of varying sizes undergo processing by a convolution module with a coordinate attention mechanism. The efficacy of the proposed method is validated using the Northeastern University surface defect database (NEU-DET) dataset, with experimental results demonstrating that the network achieves an 84.7% mean average precision (mAP) at an intersection over union (IoU) threshold of 0.5. Noteworthy contributors to the mAP of the proposed network include the image pre-processing module, the improved feature extraction network, the gather-and-distribute feature fusion network, and the detection network, contributing 4.1%, 2.9%, 2.6%, and 0.2%, respectively. The comparative experiments based on the attention mechanisms illustrate that the Squeeze-and-Excitation (SE) mechanism is the most suitable mechanism for the model proposed in this paper compared to other mainstream attention mechanisms. In comparison with other deep learning networks, our network demonstrates a significant enhancement in detection capability, showcasing superior performance in the identification of steel surface defects.