基于U型残差编解码网络的带钢缺陷检测算法

Strip Steel Defect Detection Algorithm Based on U-shaped Residual Codec Network

  • 摘要: 工业生产中带钢表面缺陷具有缺陷类型多变、背景杂乱和对比度低等特点, 现有的缺陷检测方法无法检测出完整的缺陷对象。针对上述问题, 提出一种基于U型残差编解码网络的带钢缺陷检测算法。在编码阶段, 算法利用全卷积神经网络提取丰富的多尺度缺陷特征, 并结合注意力机制加速模型收敛。在解码阶段, 使用所提出的U型残差解码网络恢复编码阶段编码的显著性信息。此外, 设计了一个残差细化网络, 用以进一步优化粗糙的显著图。实验结果表明, 所提出的算法具有较强的鲁棒性。

     

    Abstract: The surface defects of strip steel in industrial production have the characteristics of changeable defect types, cluttered background and low contrast. The existing defect detection methods cannot detect the complete defect objects. In order to solve the above problems, a strip defect detection algorithm based on U-shaped residual encoder-decoder network is proposed. In the encoding stage, the algorithm uses a fully convolutional neural network to extract rich multi-scale defect features, and combines an attention mechanism to accelerate model convergence. In the decoding stage, the saliency information encoded in the encoding stage is recovered using the proposed U-shaped residual decoding network. Furthermore, a residual refinement network is designed to further optimize the coarse saliency map. The experimental results show that the proposed algorithm has strong robustness.

     

/

返回文章
返回