高级检索

基于改进U-Net的皮革表面缺陷精细化分割方法

Refined Segmentation Network for Leather Surface Defect Detection Based on Improved U-Net

  • 摘要: 针对皮革缺陷形态多变、局部相似程度高导致细节信息难以提取、类型错分的问题, 提出一种基于改进U-Net结构的表面缺陷精细化分割方法.编码端, 在保留原始图像细节特征的同时嵌入级联扩张卷积模块获取全局特征, 同时在跳跃连接中添加特征融合模块, 改善因高低特征张量直接拼接造成的局部信息丢失; 解码端使用基于通道注意力机制的解码模块代替原始卷积层, 自适应地指导网络关注缺陷区域; 为进一步整合高层信息, 还嵌入全局平均池化模块, 将输出结果作为解码端的语义指导增强网络对相似缺陷的分辨能力.在包含7种缺陷的皮革数据集上进行实验的结果表明, 所提方法在PA, MPA, FWIoU和MIoU上分别达到99.17%, 93.27%, 98.39%和88.88%, 对比U-Net分别提升0.28, 2.78, 0.53和4.03个百分点; 定性分析和定量分析结果表明, 该方法对于皮革缺陷能得到更加精细的分割结果.

     

    Abstract: Leather defects with variable morphology and high local similarity are of difficulty in extracting features comprehensively and accurately. In this work, a refined surface defect segmentation method based on improved U-Net network is proposed. On the encoder side, a cascaded dilated convolution module is embedded to obtain the global features while preserving the detail information of the original image, and a feature fusion module is added to the jump connection to reduce local features loss caused by directly splicing of the high-level and low-level feature tensor; on the decoder side, a decoding module based on the channel attention mechanism,which can guide the network to adaptively focus on defective regions, is used to replace the original convolutional layer; to further integrate high-level information, a global average pooling module is embedded as the semantic guide to improve the discrimination capability of the network from similar defects at the decoding end. The experimental results conducted on a leather dataset containing 7 kinds of defects show that the proposed method achieves 99.17%, 93.27%, 98.39%, and 88.88% in PA, MPA, FWIoU, and MIoU, which is 0.28, 2.78, 0.53, and 4.03 percentage points better than that of U-Net. The qualitative and quantitative analysis results demonstrate that the algorithm proposed has remarkable ability to refine the segmentation in leather defect recognition.

     

/

返回文章
返回