一种改进U-Net的高分辨率遥感影像道路提取方法

作者：王卓; 闫浩文^*; 禄小敏; 冯天文; 李亚珍

来源：遥感技术与应用, 2020, 35(04): 741-748.

• 高分辨率遥感影像
• 道路提取
• U-Net网络

从遥感影像中准确高效地提取道路信息,对基础地理数据库的建立与维护具有重大意义。高分辨率遥感影像背景信息复杂,导致现有算法无法较好地从中提取道路信息。U-Net网络在图像分割方面有较好的实验效果,但道路分割结果准确性不佳,因此,提出了一种改进U-Net网络的高分辨率影像道路提取方法。首先,设计基于U-Net的网络结构,将VGG16作为网络编码结构,可更好地提取特征语义信息;其次,利用Batch Normalization与Dropout解决网络训练过程中出现的过拟合;最后,对训练数据利用旋转与镜像变换进行扩充,采用ELU激活函数,提升了网络训练速度。实验结果表明:该方法可以较为准确高效地提取道路信息。

• 单位
  中国科学院西北生态环境资源研究院; 中国科学院大学; 兰州交通大学