基于改进U2-Net网络的金属涂层剥落与腐蚀图像分割方法

Image segmentation method for metal coating peeling and corrosion based on improved U2-Net network

  • 摘要: 针对金属涂层缺陷图像分割中存在特征提取能力弱和分割精度低的问题,提出了一种改进的U2-Net分割模型。首先,在U型残差块(RSU)中嵌入改进的增大感受野模块(receptive field block light,RFB_l),组成新的特征提取层,增强对细节特征的学习能力,解决了网络由于感受野受限造成分割精度低的问题;其次,在U2-Net分割模型的解码阶段引入有效的边缘增强注意力机制(contour enhanced attention,CEA),抑制网络中的冗余特征,获取具有详细位置信息的特征注意力图,增强了边界与背景信息的差异性,从而达到更精确的分割效果。实验结果表明,该模型在两个金属涂层剥落与腐蚀数据集上的平均交并比、准确率、查准率、召回率和F1-measure分别达到80.36%、96.29%、87.43%、84.61%和86.00%,相比于常用的SegNet、U-Net以及U2-Net分割网络的性能都有较大提升。

     

    Abstract: An improved U2-Net segmentation model was proposed to address the issues of weak feature extraction ability and low segmentation accuracy in metal coating defect image segmentation. Firstly, an improved receptive field block light (RFB_l) module was embedded in the U-shaped residual block (RSU) to form a new feature extraction layer, which enhanced the ability to learn detailed features and solved the problem of low segmentation accuracy caused by limited receptive field in the network. Secondly, in the decoding stage of the U2-Net segmentation model, an effective contour enhanced attention (CEA) mechanism was introduced to suppress redundant features in the network, obtain feature attention maps with detailed position information, enhance the difference between boundary and background information, and achieve the more accurate segmentation results. The experimental results show that the average intersection and union ratio, accuracy, precision, recall, and F1-measure of the model on two metal coating peeling and corrosion datasets are 80.36%, 96.29%, 87.43%, 84.61%, and 86.00%, respectively. Compared with commonly used SegNet, U-Net, and U2-Net segmentation networks, the performance of the model is significantly improved.

     

/

返回文章
返回