改进U-Net模型的遥感影像水体提取研究

作者:刘佳典; 宋伟东*; 朱洪波; 张兵; 戴激光
来源:测绘科学, 2022, 47(08): 230-239.
DOI:10.16251/j.cnki.1009-2307.2022.08.026

摘要

针对传统遥感水体提取方法在大范围、高分辨率遥感影像水体提取任务中效率低等问题,该文提出了一种基于改进U-Net模型的遥感影像水体快速提取方法。通过减少以VGG16为特征提取网络的下采样层和相对应的上采样层来精简网络,提升模型运行效率。基于高分影像进行了沈阳市的水体提取研究,结果表明,改进U-Net模型各项精度指标均高于U-Net模型,交并比(IoU)和F1分别达到了90.3%和94.9%。与此同时,模型的训练时长和预测时长分别缩短了25.8%和21.6%。该文为遥感影像中水体的高效提取和分类提供了参考,并可用于水资源调查、管理和规划。

全文