一种改进U-Net的遥感影像建筑物提取方法 |
| |
引用本文: | 胡荣明任乐宽苏瑞鹏米晓梅.一种改进U-Net的遥感影像建筑物提取方法[J].测绘科学,2023(1):39-48. |
| |
作者姓名: | 胡荣明任乐宽苏瑞鹏米晓梅 |
| |
作者单位: | 1.西安科技大学测绘科学与技术与技术学院710054; |
| |
基金项目: | 国家自然科学基金项目(42171394) |
| |
摘 要: | 针对高分辨率遥感影像信息复杂浅层网络难以对其目标物特征信息充分学习,图像因裁剪导致边缘信息损失使得模型对图像边缘预测效果较差的问题,该文将U-Net收缩路径加深以增强网络对特征信息的学习能力,并加入随机失活函数(Dropout)层抑制过拟合现象的发生,扩张路径中加入批量归一化层以提高网络训练速度,并将忽略边缘交叉熵函数与骰子函数结合构建联合损失函数作为本文模型的损失函数以提高模型对图像边缘的预测效果。实验结果表明:该文方法对建筑物边缘能够进行有效预测;对建筑物轮廓以及较小建筑物的提取较之SVM、主干网络为VGG的U-Net提取效果有所提高;并在应用扩展研究数据集中有着较好的表现。
|
关 键 词: | 高分辨率遥感影像 U-Net 联合损失函数 建筑物提取 |
本文献已被 维普 等数据库收录! |