基于改进UNet的人造板表面缺陷的图像分割方法

Image segmentation method of surface defects of wood-based panels based on improved UNet

为满足人造板表面缺陷图像分割的精度要求,提出了一种改进的UNet语义分割网络模型。在传统的UNet网络结构上将编码部分改进为残差网络ResNet50并去掉连接层与平均池化层,网络通过残差块堆叠获取更多特征的底层信息;同时在跳跃连接中嵌入聚焦注意力机制的模块,抑制干扰信息,保留有效位置信息,聚焦缺陷区域并加强学习。对4种UNet网络模型的人造板表面缺陷图像分割进行仿真比较,结果表明,融合聚焦注意力机制的残差UNet网络模型在像素准确率和平均交并比等指标上有较大提升,分割精度较高。

In order to meet the requirements of the precision of image segmentation of surface defects of wood-based panels,an improved UNet semantic segmentation network model was proposed.The coding part of the traditional UNet network was modified into residual network ResNet50,and the connection layer and average pooling layer were removed.The network was stacked with residual blocks to obtain more underlying information of features.At the same time,the module of attention focusing mechanism is embedded in the jump connection to suppress interference information,retain effective location information,focus defect location and enhance learning.The simulation comparison of image segmentation of surface defects of wood-based panels based on four UNet models shows that the residual UNet model integrating the attention focusing mechanism has been greatly improved in pixel accuracy and average intersection ratio,as well as the segmentation accuracy.