基于U-Net网络的林木图像分割研究

• 作者: 张博洋 倪海明 胡馨月 戚大伟
• 来源期刊: 森林工程
• 关键词: 无人机图像 图像分割 U-Net 跳跃连接
• 年，卷（期）: 2021,（2）
• 所属期刊栏目: 森工技术与装备|Forest Engineering Technology and Equipment
• 页码范围: 67-73
• 页数: 7页
• 分类号: S758.5
• 语种: 中文
• DOI: 10.3969/j.issn.1006-8023.2021.02.010

摘要

针对传统方法进行图像分割易受噪声影响的问题,提出一种基于U-Net网络的无人机图像语义分割网络模型.该模型不需要对图像进行预处理,利用反卷积恢复图像分辨率,采用U型结构连接低层网络和高层网络的特征图,利用跳跃连接降低网络复杂度,同时使用Dropout正则化随机激活网络隐藏单元以防止过拟合.实验结果表明:该网络模型可以自动定位林木信息,准确分割林木区域,进一步优化边缘分割结果,实现端对端的图像分割.该模型具有良好的泛化能力,在其他图像分割领域也具有应用价值.