基于多尺度局部与全局上下文信息的钢材缺陷检测方法

Steel Surface Defect Detection Model Based on Multiscale Local and Global Context Information

  • 摘要: 钢材表面缺陷对许多工业产品的质量和性能有重大影响,会给生产带来巨大的经济损失。因此,对钢材表面进行实时监控,及时发现缺陷是非常有必要的。为提升对尺度差异较大、背景复杂的钢材表面缺陷的检测性能,提出一种基于多尺度局部与全局上下文信息的钢材缺陷检测模型。该模型使用卷积神经网络模型中带下采样机制的卷积操作,获取粗糙多尺度局部特征图,再使用自注意力机制分别在每一尺度作用于经过卷积提取的局部特征图,以获取像素间的长相关信息(如图像划痕、斑块、夹杂物等),增强缺陷类间判别能力;然后,采用特征金字塔结构,融合多尺度的特征图,以此提升对多尺度目标的检测能力;最后,引入通道与空间注意力模块与WIoU损失函数。实验结果表明,相比于Faster RCNN和EDDN等模型,该方法对于提升钢材表面缺陷检测性能行之有效。

     

    Abstract: Steel surface defects have a significant impact on the quality and performance of many industrial products, which will bring huge economic losses to production. Therefore, it is very necessary to detect the steel surface in real time and find defects in time. In order to improve the detection performance of steel surface defects with large scale differences and complex backgrounds, a steel surface defect detection model based on multiscale local and global context information was proposed. Convolution operation with down-sampling mechanism in the convolutional neural network model was used to obtain rough multi-scale local feature maps. Then, self-attention mechanism was used to act on the local feature map extracted by convolution at each scale, to obtain long-distance dependencies between pixels (such as scratches, patches, inclusions, etc.), thus to enhance the inter-class discrimination ability of defects. Afterwards, the feature pyramid structure was used to fuse multi-scale feature maps, to improve the detection ability of multi-scale objects. Finally, channel and spatial attention module and WIoU loss function were introduced. The experimental results showed that, compared with algorithms such as Faster RCNN and EDDN, the proposed method was effective in improving the detection performance of steel surface defects.

     

/

返回文章
返回