本发明专利技术提供一种遥感影像处理方法、装置、设备及存储介质,所述方法基于预先训练的UNet 3+网络对所述遥感影像进行图像分割,生成第一输出图像,基于空间注意力网络,在所述第一输出图像中,捕获与预定目标相关的所述区域图像,作为所述预定目标的第二输出图像,对所述第二输出图像进行形态学图像处理,得到所述预定目标分割图。本发明专利技术通过图像分割,将遥感影像生成第一输出图像,将第一输出图像输入到空间注意力网络,生成预定目标的第二输出图像,相比现有遥感影像地块分割技术处理遥感影像,能够获得更加清晰的预定目标分割图,解决遥感影像内的细长道路图像出现断续连接的问题。影像内的细长道路图像出现断续连接的问题。影像内的细长道路图像出现断续连接的问题。
【技术实现步骤摘要】
遥感影像处理方法、装置、设备及存储介质
[0001]本专利技术涉及遥感图像处理
,尤其涉及一种遥感影像处理方法、装置、设备及存储介质。
技术介绍
[0002]遥感影像地块分割技术是对遥感影像进行像素级内容解析,然后将遥感影像中关注的特征进行提取和分类;该技术在城乡规划、防汛救灾等领域具有很高的实用价值。但是,通过遥感影像地块分割技术处理遥感影像,会导致遥感影像内的道路或河流所在部分图像出现断续的问题。
技术实现思路
[0003]本专利技术的主要目的在于提供一种遥感影像处理方法、装置、设备及计算机可读存储介质,旨在解决现有遥感影像内的道路或河流所在部分图像出现断续的技术问题。
[0004]为实现上述目的,本专利技术提供一种遥感影像处理方法,所述遥感影像处理方法包括:基于预先训练的UNet 3+网络对所述遥感影像进行图像分割,生成第一输出图像;基于空间注意力网络,在所述第一输出图像中,捕获与预定目标相关的区域图像,作为所述预定目标的第二输出图像;对所述第二输出图像进行形态学图像处理,得到无断续的目标分割图。
[0005]此外,为实现上述目的,本专利技术还提供一种遥感影像处理装置,所述遥感影像处理装置包括:图像分割模块,用于基于预先训练的UNet 3+网络对所述遥感影像进行图像分割,生成第一输出图像;空间注意力模块,用于基于空间注意力网络,在所述第一输出图像中,捕获与预定目标相关的所述区域图像,作为所述预定目标的第二输出图像;后处理模块,用于对所述第二输出图像进行形态学图像处理,得到无断续的目标分割图。
[0006]此外,为实现上述目的,本专利技术还提供一种遥感影像处理设备,所述遥感影像处理设备包括处理器、存储器、以及存储在所述存储器上并可被所述处理器执行的遥感影像处理程序,其中所述遥感影像处理程序被所述处理器执行时,实现如上述的遥感影像处理方法的步骤。
[0007]此外,为实现上述目的,本专利技术还提供一种计算机可读存储介质,所述计算机可读存储介质上存储有遥感影像处理程序,其中所述遥感影像处理程序被处理器执行时,实现如上述的遥感影像处理方法的步骤。
[0008]本专利技术提供一种遥感影像处理方法,所述方法基于预先训练的UNet 3+网络对所述遥感影像进行图像分割,生成第一输出图像,基于空间注意力网络,在所述第一输出图像
中,捕获与预定目标相关的所述区域图像,作为所述预定目标的第二输出图像,对所述第二输出图像进行形态学图像处理,得到无断续的目标分割图,通过上述方式,本专利技术将所述遥感影像输入到预先训练过的UNet 3+网络中,通过图像分割的方法,将输入到UNet 3+网络中的遥感影像生成第一输出图像,将第一输出图像输入到空间注意力网络,将与预定目标相关的区域图像捕获,以此生成预定目标的第二输出图像,然后对第二输出图像进行形态学图像处理,获得图像内容完整的预定目标分割图,由此,基于UNet 3+网络与空间注意力网络生成预定目标分割图,相比现有遥感影像地块分割技术处理遥感影像,能够获得更加清晰的预定目标分割图,解决遥感影像内的细长道路或细长河流所在部分图像出现断续的问题。
附图说明
[0009]图1为本专利技术实施例方案中涉及的遥感影像处理设备的硬件结构示意图;图2为本专利技术遥感影像处理方法第一实施例的流程示意图;图3为本专利技术遥感影像处理方法第二实施例的流程示意图;图4为本专利技术遥感影像处理方法第三实施例的流程示意图;图5为本专利技术遥感影像处理方法第四实施例的流程示意图;图6为本专利技术遥感影像处理装置一实施例的功能模块示意图。
[0010]本专利技术目的的实现、功能特点及优点将结合实施例,参照附图做进一步说明。
具体实施方式
[0011]应当理解,此处所描述的具体实施例仅仅用以解释本专利技术,并不用于限定本专利技术。
[0012]本专利技术实施例涉及的遥感影像处理方法主要应用于遥感影像处理设备,该遥感影像处理设备可以是PC、便携计算机、移动终端等具有显示和处理功能的设备。
[0013]参照图1,图1为本专利技术实施例方案中涉及的遥感影像处理设备的硬件结构示意图。本专利技术实施例中,基于遥感影像处理设备可以包括处理器1001(例如CPU),通信总线1002,用户接口1003,网络接口1004,存储器1005。其中,通信总线1002用于实现这些组件之间的连接通信;用户接口1003可以包括显示屏(Display)、输入单元比如键盘(Keyboard);网络接口1004可选的可以包括标准的有线接口、无线接口(如WI
‑
FI接口);存储器1005可以是高速RAM存储器,也可以是稳定的存储器(non
‑
volatile memory),例如磁盘存储器,存储器1005可选的还可以是独立于前述处理器1001的存储装置。
[0014]本领域技术人员可以理解,图1中示出的硬件结构并不构成对遥感影像处理设备的限定,可以包括比图示更多或更少的部件,或者组合某些部件,或者不同的部件布置。
[0015]继续参照图1,图1中作为一种计算机可读存储介质的存储器1005可以包括操作系统、网络通信模块以及遥感影像处理程序。
[0016]在图1中,网络通信模块主要用于连接服务器,与服务器进行数据通信;而处理器1001可以调用存储器1005中存储的遥感影像处理程序,并执行本专利技术实施例提供的遥感影像处理方法。
[0017]本专利技术实施例提供了一种遥感影像处理方法。
[0018]参照图2,图2为本专利技术遥感影像处理方法第一实施例的流程示意图。
[0019]本实施例中,所述遥感影像处理方法包括以下步骤:步骤S10,基于预先训练的UNet 3+网络对所述遥感影像进行图像分割,生成第一输出图像;本实施例中,预先将遥感影像以及对应的第一输出图像作为训练数据,对全卷积网络UNet 3+(属于UNet的进一步改进网络)网络进行训练,UNet 3+网络包括编码器和解码器,编码器作为UNet 3+网络的输入层,解码器作为UNet 3+网络的输出层,UNet 3+网络具有多层级,编码器和解码器均具有多个,多个编码器与多个解码器之间采用跳跃连接并且多个解码器之间相互连接。
[0020]为了从遥感图像中学习层次表示,UNet 3+网络进一步采用了深度监督方式,UNet 3+网络中每个解码器阶段都有一个侧输出,与GT(ground truth,地面实况)进行比较计算Loss,从而实现全尺度的监督。
[0021]具体的,将遥感影像输入到预先训练的UNet 3+网络中,从遥感影像的任意目标图像出发,得到遥感影像对应的各自像素类别,实现目标提取,其中,所述分割目标由一些相互连通的像素组成的,当UNet 3+网络对遥感影像的对应目标的各自像素类别完成判断以后,得到第一输出图像。
[0022]步骤S20,基于空间注意力网络,在所述第一输出图像中,捕获与预定目标相关的所述区域图像,作为所述预定目标的第二输出图像;具体的,基于空间注意力网络,将所述第一输出图像输入到空间注意力网本文档来自技高网...
【技术保护点】
【技术特征摘要】
1.一种遥感影像处理方法,其特征在于,所述遥感影像处理方法包括以下步骤:基于预先训练的UNet 3+网络对所述遥感影像进行图像分割,生成第一输出图像;基于空间注意力网络,在所述第一输出图像中,捕获与预定目标相关的区域图像,作为所述预定目标的第二输出图像;对所述第二输出图像进行形态学图像处理,得到无断续的目标分割图。2.如权利要求1所述的遥感影像处理方法,其特征在于,所述基于空间注意力网络,在所述第一输出图像中,捕获与预定目标相关的所述区域图像,作为所述预定目标的第二输出图像,包括:基于所述空间注意力网络对所述第一输出图像进行注意力调整,生成空间注意力图;基于所述空间注意力网络将所述空间注意力图的像素点与所述第一输出图像的像素点相乘,生成所述第二输出图像。3.如权利要求1所述的遥感影像处理方法,其特征在于,所述基于预先训练的UNet 3+网络对所述遥感影像进行图像分割,生成第一输出图像,包括:基于所述UNet 3+网络的编码器生成所述遥感影像的至少两幅遥感特征图;获取两幅所述遥感特征图,分别作为第一遥感特征图、第二遥感特征图,基于所述UNet 3+网络的解码器分别捕获所述第一遥感特征图中的粗粒度语义特征和/或细粒度语义特征、所述第二遥感特征图中的粗粒度语义特征和/或细粒度语义特征;基于所述解码器,将所述第一遥感特征图中的粗粒度语义特征或细粒度语义特征与所述第二遥感特征图中的粗粒度语义特征或细粒度语义特征进行融合,生成所述第一输出图像。4.如权利要求1所述的遥感影像处理方法,其特征在于,所述对所述第二输出图像进行形态学图像处理,得到无断续的目标分割图,包括:对所述第二输出图像进行形态学闭运算,得到第一遥感图像;根据所述预定目标的轮廓特征和面积特征,对所述第一遥感图像进行边界追踪分析,得到第二遥感图像;根据预设阈值剔除所述第二遥感图像的孤立像素颗粒,得到第三遥感图像;对所述第三遥感图像进行膨胀运算后,得到第四遥感图像;将所述第四遥感图像进...
【专利技术属性】
技术研发人员:黄军文,汤红,赵士红,
申请(专利权)人:深圳市华付信息技术有限公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。