文献类型：期刊文章

作　　者：熊彬[1] 张双德[1]

XIONG Bin;ZHANG Shuangde(School of Electrical and Electronic Engineering,Wuhan Polytechnic University,Wuhan 430000,China)

机构地区：[1]武汉轻工大学电气与电子工程学院,武汉430000

出　　处：《遥感信息》

年　　份：2023

卷　　号：38

期　　号：4

起止页码：73-79

语　　种：中文

收录情况：BDHX、BDHX2020、CSCD、CSCD2023_2024、JST、ZGKJHX、核心刊

摘　　要：针对卫星遥感图像中的建筑分割算法对特征相似的小目标建筑识别效果差以及对大目标建筑轮廓边缘分割精度低的问题,基于PSPNet算法,结合Swin Transformer主干网络和特征金字塔融合局部特征以及通道和空间注意力机制,提出一种改进的PSPNet算法,实现对卫星遥感图像中的建筑高精度分割。首先,将原始PSPNet算法的主干网络Resnet101替换为Swin Transformer以增加主干网络的全局特征提取能力;其次,以主干网络输出多层级特征图并结合特征金字塔融合模块获取图像的局部语义信息;最后,将全局信息和局部信息融合并送入通道注意力以及空间注意力模块CBAM,使其在空间和通道两个维度都能获得丰富的语义信息。实验结果表明,改进后的PSPNet算法平均准确度和平均交并比分别为91.76%和86.02%,较原始的PSPNet算法提高了2.48%和3.47%,且优于深度学习领域经典的U-Net、FCN、DeeplabV3+、DANet等语义分割算法。

关 键 词：PSPNet  注意力机制  特征融合  深度学习  语义分割  

分 类 号：TP79]