欢迎访问《生态学杂志》官方网站,今天是 分享到:

生态学杂志

• 方法与技术 • 上一篇    下一篇

基于U-Net卷积神经网络的年轮图像分割算法

宁霄,赵鹏*   

  1. (东北林业大学信息与计算机工程学院, 哈尔滨 150040)
  • 出版日期:2019-05-10 发布日期:2019-05-10

Segmentation algorithm of annual ring image based on U-Net convolution network.

NING Xiao, ZHAO Peng*   

  1. (Information and Computer Engineering College, Northeast Forestry University, Harbin 150040, China).
  • Online:2019-05-10 Published:2019-05-10

摘要: 树木年轮学的研究需要统计树龄和测量轮宽,由此推算环境变换和树木生长信息,因此准确提取年轮特征信息至关重要。精准识别出年轮图像中的早材、晚材和树皮是实现自动化测量年轮参数的首要工作。树木年轮的生长过程中存在年轮的早材和晚材间边界过渡模糊、节疤和伪年轮等现象,且年轮圆盘在砍伐和采集过程中表面会存在毛刺和噪声点,使用传统的图像分割算法难以得到理想的效果。本文结合深度神经网络的特点,针对年轮图像的分割问题,构建了基于U-Net卷积神经网络的年轮图像语义分割模型。首先,对采集的100张年轮圆盘图像进行标注,并通过旋转、透视和图像变形等方式做数据增强,生成20000张数据集,随机选择其中16000张作为训练数据集,4000张作为测试数据集。其次,根据图像数据集的特征,利用Tensorflow深度学习框架,设计构建基于U-Net卷积神经网络的年轮圆盘图像分割网络。然后,将训练样本输送进网络,设置优化训练参数,对年轮图像分割网络进行迭代训练,直至评价指标和损失函数不再变化。最后,用训练好的模型对测试集样本进行分割,并进行分割指标评估。结果表明:该算法可有效避免毛刺、锯痕和节疤等因素的影响,完整地分割出年轮的晚材和树皮区域,在4000张测试数据集上分割的平均准确率达到96.51%,平均区域重合度达到82.30%。与传统图像处理算法相比,本文所采用的基于U-Net卷积神经网络的年轮图像分割算法,能够达到更好的分割效果,同时具有更强的泛化能力和鲁棒性。

关键词: 空间异质性, 土壤化学因子, 滨海盐渍区, 耐盐植物

Abstract: Dendrochronological research uses tree-age and annual-ring width to estimate environmental changes and tree growth. Thus, it is important to accurately extract the characteristics such as the early wood, late wood, and bark parts in the annual-ring images for further analysis. It is difficult to obtain the desired effect using traditional image segmentation algorithm due to the existence of defects such as fuzzy interface between the early and late woods, knots and pseudo-annual rings during growth and there are burrs and noise spots on the image of the annual ring disc during the cutting and collecting process. Here, we proposed a novel approach to perform annualring image semantic segmentation based on convolutional neural network. Firstly, 100 annual-ring images were marked as late wood, bark and other parts. Data enhancement was implemented- through image rotation, perspective, and deformation to generate 20000 image data, from which 16000 images were randomly selected as training data sets and 4000 images were used as test dataset. Secondly, according to the characteristics of image dataset, an annual-ring disc image segmentation network was developed based on U-Net convolutional network using the Tensorflow framework. Then, the training dataset was sent into the network, the training parameters were optimized, and the annual-ring image segmentation network was iteratively trained until the evaluation index and the loss function no longer change. Finally, the test dataset was segmented using the trained model and the segmentation indicators were evaluated. Experimental results showed that the constructed model can effectively avoid the defects mentioned above, and completely separate the late wood and bark parts of the annual-ring images. The proposed approach was tested with dataset consisting of 4000 tree ring images, the corresponding accuracy of mean pixels and the mean intersection over union achieved 96.51% and 82.30%, respectively. Thisapproach based on U-Net convolutional network is a more efficient algorithm for annual-ring image segmentation, with stronger generalization ability and robustness.

Key words: spatial heterogeneity, coastal saline area, salt-tolerant plant, soil chemical factor