【技术实现步骤摘要】
一种基于通道注意力特征融合的遥感图像语义分割方法
[0001]本申请属于图像处理
,具体涉及一种基于通道注意力特征融合的遥感图像语义分割方法。
技术介绍
[0002]高空间分辨率(High Spatial Resolution,HSR)遥感图像在距离地球表面很远的地方获取,通常包含丰富的地理空间信息。遥感语义分割作为遥感图像理解中的一项基本任务,对于环境监测、灾后重建、农业、林业和城市规划有着重大意义。相比于自然图像,遥感图像由于其独特的成像机制和场景特征,在语义分割中会面临特殊的挑战。遥感图像的前景目标存在较大的尺度变化,使得物体难以定位和识别。同时,前景目标往往隐藏在复杂的背景之中,引起了前景
‑
背景不平衡的问题。因此,如何准确捕捉图像语义特征,缓解尺度变化,抑制误报,对遥感图像语义分割至关重要。
[0003]最近,许多经典的卷积神经网络(CNN)在图像分割领域取得了良好的表现。全卷积网络(FCN)及其拓展应用也被广泛用于遥感分割领域。但是这些方法是直接采用或者修改自卷积神经网络架构,没有考虑遥感图像中小物体内部的弱特征,取得的效果并不理想。U
‑
Net和SegNet通过编码器
‑
解码器的结构,利用具有高空间分辨率的浅层特征来增强具有强空间细节语义的深层特征,但是缺乏对全局信息的捕获能力,对不明显的目标识别能力较差。FPN及其相关应用通过特征金字塔结构提取多尺度的特征,缓解了遥感图像中尺度变化问题。不过这些方法反而会影响图像中的大物体,而且无法解 ...
【技术保护点】
【技术特征摘要】
1.一种基于通道注意力特征融合的遥感图像语义分割方法,其特征在于,所述基于通道注意力特征融合的遥感图像语义分割方法,包括:将待预测的遥感图像输入特征提取模块,提取四个尺度的初始特征{F1,F2,F3,F4};将提取到的特征F4输入全局信息模块,扩大感受野并充分利用全局上下文信息,输出相同的特征和将特征与初始特征{F3,F2,F1}依次通过注意力融合模块,逐步获得多尺度通道注意力融合后的特征将特征与注意力融合后的特征依次通过注意力解码模块,逐步获得解码后的特征将解码后的最后一个特征通过卷积和上采样,然后使用Softmax分类器获得最终的语义分割结果。2.根据权利要求1所述的基于通道注意力特征融合的遥感图像语义分割方法,其特征在于,所述特征提取模块采用ResNet
‑
50骨干网络,在输入待预测的遥感图像尺寸为896
×
896
×
3时,提取四个初始特征{F1,F2,F3,F4}的尺度分别为:224
×
224
×
256,112
×
112
×
256,56
×
56
×
1024,28
×
28
×
2048。3.根据权利要求2所述的基于通道注意力特征融合的遥感图像语义分割方法,其特征在于,所述将提取到的特征F4输入全局信息模块,扩大感受野并充分利用全局上下文信息,输出相同的特征和包括:S21.将特征F4分别通过目标大小分别为1,3和6的三个平均池化层,得到三个不同尺寸的特征图尺度分别为1
×1×
2048,3
×3×
2048,6
×6×
2048;S22.将特征图分别通过1
×
1的卷积核,进行通道的压缩以去除冗余的通道信息,再进行上采样至F4相同大小,得到特征图尺度均为28
×
28
×
256;S23.将三个特征图进行通道维度上的拼接,得到尺度为28
×
28
×
768的特征S24.将特征通过1
×
1的卷积核,把通道数降低至256,获得糅合了多种尺度的复合特征...
【专利技术属性】
技术研发人员:郑建炜,邵安昊,严亦东,方创杰,李燕,
申请(专利权)人:浙江工业大学,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。