An efficient steel defect detection model based on multi-scale information extraction

Abstract

Purpose

In the field of steel defect detection, the existing detection algorithms struggle to achieve a satisfactory balance between detection accuracy, computational cost and inference speed due to the interference from complex background information, the variety of defect types and significant variations in defect morphology. To solve this problem, this paper aims to propose an efficient detector based on multi-scale information extraction (MSI-YOLO), which uses YOLOv8s as the baseline model.

Design/methodology/approach

First, the authors introduce an efficient multi-scale convolution with different-sized convolution kernels, which enables the feature extraction network to accommodate significant variations in defect morphology. Furthermore, the authors introduce the channel prior convolutional attention mechanism, which allows the network to focus on defect areas and ignore complex background interference. Considering the lightweight design and accuracy improvement, the authors introduce a more lightweight feature fusion network (Slim-neck) to improve the fusion effect of feature maps.

Findings

MSI-YOLO achieves 79.9% mean average precision on the public data set Northeastern University (NEU)-DET, with a model size of only 19.0 MB and an frames per second of 62.5. Compared with other state-of-the-art detectors, MSI-YOLO greatly improves the recognition accuracy and has significant advantages in computational cost and inference speed. Additionally, the strong generalization ability of MSI-YOLO is verified on the collected industrial site steel data set.

Originality/value

This paper proposes an efficient steel defect detector with high accuracy, low computational cost, excellent detection speed and strong generalization ability, which is more valuable for practical applications in resource-limited industrial production.

Citation

Xu, W., Zhang, Y., Jiang, X., Lian, J. and Lin, Y. (2024), "An efficient steel defect detection model based on multi-scale information extraction", Robotic Intelligence and Automation, Vol. 44 No. 6, pp. 817-829. https://doi.org/10.1108/RIA-03-2024-0065