改进的Unet型木材缺陷图像分割方法

编号 zgly0001738796

文献类型 期刊论文

文献题名 改进的Unet型木材缺陷图像分割方法

作者 严飞  章继鸿  姚宇晨  刘军 

作者单位 南京林业大学机械电子工程学院 

母体文献 林业机械与木工设备 

年卷期 2022,50(1)

页码 41-45

年份 2022 

分类号 TP391.41 

关键词 木材缺陷  图像分割  卷积神经网络  Unet  FCN 

文摘内容 卷积神经网络(Convolutional Neural Network,CNN)是深度学习的最重要的网络之一,基于CNN的语义分割是当前研究的热点之一,Unet是在FCN基础上发展而来的卷积神经网络模型。针对木材缺陷图像分割问题,提出一种基于改进的Unet模型与像素阈值的木材缺陷图像分割方法。首先,在Unet的基本网络结构上,对网络的层数、通道数进行修改;然后,利用Unet实现网络训练,获得结构参数,最后用训练好的网络对图像进行测试,获取特征通道灰度图,并利用OTSU阈值算法对灰度图进行分割。结果表明,选择好的网络结构和阈值等参数,算法能够实现木材缺陷的图像分割,激活层通道灰度图分割效果优于卷积层。