目标上下文卷积神经网络高分遥感影像语义分割

作者：刘艳飞; 丁乐乐; 孟凡效

来源：遥感信息, 2021, 36(06): 66-74.

• 卷积神经网络
• 高分影像
• 目标上下文
• 语义分割

针对传统卷积神经网络高分影像语义分割方法中忽略全局影像中目标像素之间关系的问题，文章显式地对全局空间目标上下文建模，将目标上下文卷积神经网络用于高分遥感影像语义分割，使用的方法包含粗分割分支和精细分割分支两个分支网络。首先，利用粗语义分割分支获得每一个像素的类别概率分布；然后，基于得到的类别概率分布和精分割的特征图获得每个类别的特征中心，根据类别特征中心对每个像素进行编码得到融合全局目标上下文的编码特征；最后，将像素的编码特征和精分割特征叠加，作为最终的表达特征用于精分割分支的语义分割任务。为验证算法在高分遥感影像分割的有效性，文章在GID和WHU building两个语义分割数据集上进行实验。相比于PointRend网络，文章使用的方法将两个数据集上的MIoU指标分别提高了1.87和1.32。实验结果证明，该算法可以有效利用全局空间目标上下文进行特征融合，提高高分影像语义分割精度。