基于Unet网络多任务学习的遥感图像建筑地物语义分割

作者：刘尚旺,崔智勇,李道义

摘要：为准确分割出高分辨率遥感图像中的建筑地物,提出一种基于Unet网络多任务学习的建筑地物语义分割方法。首先,根据遥感图像建筑地物真值图生成边界距离图,并将该遥感图像及其真值图共同作为Unet网络的输入;然后,在基于ResNet网络构建的Unet网络末端加入建筑地物预测层与边界距离预测层,搭建多任务网络;最后,定义多任务网络的损失函数,并使用Adam优化算法训练该网络。在Inria航空遥感图像建筑地物标注数据集上进行实验,结果表明,与全卷积网络结合多层感知器方法相比,VGG16网络、VGG16+边界预测、ResNet50和本文方法的交并比值分别提升5.15,6.94,6.41和7.86百分点,准确度分别提升至94.71%,95.39%,95.30%和96.10%,可实现高精度的建筑地物提取。

发文机构：河南师范大学计算机与信息工程学院 “智慧商务与物联网技术”河南省工程实验室

关键词：Unet网络多任务学习遥感图像语义分割ResNet网络Unet networkmulti-task learningremote sensing imagesemantic segmentationResNet network

分类号： TP751.1[自动化与计算机技术—检测技术与自动化装置]