您好,欢迎访问北京市农林科学院 机构知识库!

基于U-net语义分割模型的田间玉米冠层图像分割

文献类型: 中文期刊

作者: 王晓东 1 ; 樊江川 2 ; 杜建军 2 ; 云挺 3 ; 温维亮 2 ; 郭新宇 2 ;

作者机构: 1.中国农业大学信息与电气工程学院

2.国家农业信息化工程技术研究中心

3.南京林业大学信息科学技术学院

关键词: 神经网络;残差学习;图像分割;玉米冠层

期刊名称: 江苏科技大学学报(自然科学版)

ISSN: 1673-4807

年卷期: 2021 年 35 卷 002 期

页码: 59-67

摘要: 针对田间冠层图像质量易受光照和天气变化等外界因素影响造成的分割精度低且无法自动化实现的问题,提出一种利用残差学习改进U-net模型结构的玉米冠层图像分割方法,利用残差结构增强U-net模型收缩路径的深度,提高模型的特征提取能力,并对模型结构进行重组设计,引入批标准化和Dropout机制增强模型的泛化性,构成U型残差学习的ResU-net分割模型,并结合分割领域典型的损失函数和评价指标,设计多种损失函数,通过对比不同损失函数在测试集上的表现,最终选定Focal_Tversky损失函数构成ResU-net++语义分割模型.实验发现:ResU-net模型在测试集上的准确率为0.9398,优于传统的U-net模型的0.8936;且优化后的ResU-net++方法在测试集中的准确率为0.9539、精确率为0.931、召回率为0.9525、F1值为0.9419、IOU值为0.9123,优于U-net和ResU-net模型.以不同的玉米品种构建了验证样本集,结果表明,ResU-net++模型在验证样本中的分割精确率为0.9325,说明ResU-net++模型具有一定的泛化性,可以为不同光照条件下玉米原位冠层图像的分割提供参考.

  • 相关文献
作者其他论文 更多>>