首页 | 官方网站   微博 | 高级检索  
     

基于改进UNet的人造板表面缺陷的图像分割方法
引用本文:张平均,翁悦,王小红,李稳稳,林艺斌.基于改进UNet的人造板表面缺陷的图像分割方法[J].福建工程学院学报,2022,0(4):373-377.
作者姓名:张平均  翁悦  王小红  李稳稳  林艺斌
作者单位:福建工程学院电子电气与物理学院
摘    要:为满足人造板表面缺陷图像分割的精度要求,提出了一种改进的UNet 语义分割网络模型。 在传统的UNet 网络结构上将编码部分改进为残差网络ResNet50 并去掉连接层与平均池化层,网络通过残差块堆叠获取更多特征的底层信息;同时在跳跃连接中嵌入聚焦注意力机制的模块,抑制干扰信息,保留有效位置信息,聚焦缺陷区域并加强学习。 对4 种UNet 网络模型的人造板表面缺陷图像分割进行仿真比较,结果表明,融合聚焦注意力机制的残差UNet 网络模型在像素准确率和平均交并比等指标上有较大提升,分割精度较高。

关 键 词:聚焦注意力机制模块  UNet  残差网络  人造板表面缺陷  图像分割

Image segmentation method of surface defects of wood-based panels based on improved UNet
ZHANG Pingjun,WENG Yue,WANG Xiaohong,LI Wenwen,LIN Yibin.Image segmentation method of surface defects of wood-based panels based on improved UNet[J].Journal of Fujian University of Technology,2022,0(4):373-377.
Authors:ZHANG Pingjun  WENG Yue  WANG Xiaohong  LI Wenwen  LIN Yibin
Affiliation:School of Electronic, Electrical Engineering and Physics, Fujian University of Technology
Abstract: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.
Keywords:attention focusing mechanism module  UNet  residual network  surface defects of wood-based panels  image segmentation
点击此处可从《福建工程学院学报》浏览原始摘要信息
点击此处可从《福建工程学院学报》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司    京ICP备09084417号-23

京公网安备 11010802026262号