基于高空间分辨率遥感影像的储油罐提取方法技术

本发明专利技术提供了一种基于高空间分辨率遥感影像的储油罐提取方法，包括：采集储油罐样本，将所述样本随机分为训练集和测试集；基于Res2‑Unet模型结构构建储油罐提取模型；其中，基于UNet语义分割结构提出深度学习网络Res2‑Unet，采用Res2Net卷积模块，将特征层间学习改为粒度学习，并以残差方式布局；对所述测试集进行精度验证。本发明专利技术采用Res2net模块改进Unet网络结构，增强模型对储油罐多尺度特征的学习能力，可以增强对不同尺度储油罐的提取能力，进而提高模型的鲁棒性。

【技术实现步骤摘要】
基于高空间分辨率遥感影像的储油罐提取方法
本专利技术涉及图像处理、模式识别领域，特别涉及一种针对大空间范围高空间分辨率遥感影像的实用储油罐提取方法。
技术介绍
储油罐，作为储存油品的容器，在储备和中转石油及其相关产品时尤为重要。对大空间范围内的储油罐进行及时、准确的监测有助于估算各地区的石油储备量，为我国制定石油生产和储备相关政策提供数据支撑。同时，由于储油罐常建于居民区以及港口附近，对油罐的实时监测有助于评估其发生爆炸和漏油等事件时对周边居民以及海港构成的威胁。因此，开展储油罐的高效监测技术研究很有必要。近年来，遥感技术的不断发展与成熟和计算机视觉相关技术的日新月异，越来越多的国产高分卫星数据，包括高分一、高分二、高分六以及资源卫星等，被广泛的应用于各个领域的目标监测任务中，并取得了丰硕的成果。目前，基于高空间分辨率遥感影像的储油罐监测方法也得到了快速发展。由于油罐在遥感影像中多呈圆形，模板匹配和霍夫变换(Houghtransform)是两种比较常用的传统提取方法。这两种方法都存在计算量大、运算复杂和错分率高的缺点，难以对大空间范围内多种油罐进行提取。结合计算机视觉相关算法通过遥感影像分割对油罐提取也是一种常用的技术。自动阈值计算方法Otsu，均值漂移和最小生成树等传统的分割方法在油罐提取中都起到了相应的应用，但是其结果精度受储油罐的大小、颜色和与背景地物的对比度影响较大。随着机器学习和深度学习方法的不断改进，部分模型在基于遥感影像的油罐提取领域也取得了相应的成果。基于区域的卷积神经网络FasterR-CNN(region-basedconvolutionalneuralnetwork)目前已经用于高分二影像中的储油罐提取中，并取得了89％的精度。在实现本专利技术构思的过程中，专利技术人发现相关技术中至少存在如下问题：现有方法多针对亮色、尺寸较大，而且与背景地物呈现明显光谱与纹理区别的储油罐进行提取。方法的可迁移性不强，使用限制条件过多，实用性不强，很难做到大面积应用。导致这一问题的主要原因是不同地区油罐的构造材质不同，在遥感影像中呈现的光谱特性不同。而且不同应用场景下的油罐尺寸和颜色也不尽相同。受成像条件限制，光照对影像中的油罐影响较大。很多情况下的油罐呈亮白色，与背景地物极易混淆。此外，提取得到的储油罐多比较琐碎，不完整。导致这一问题的主要原因是每个储油罐本身的不同像素，其在影像中的反射率都不尽相同，因此其同物异谱现象比较严重。
技术实现思路
有鉴于此，本专利技术的主要目的在于提供一种基于高空间分辨率遥感影像的储油罐提取方法，以期部分地解决上述技术问题中的至少之一。为了实现上述目的，本专利技术提供了一种基于高空间分辨率遥感影像的储油罐提取方法一种基于高空间分辨率遥感影像的储油罐提取方法，包括：采集储油罐样本，将所述样本随机分为训练集和测试集；基于Res2-Unet模型结构构建储油罐提取模型；其中，基于UNet语义分割结构提出深度学习网络Res2-Unet，采用Res2Net卷积模块，将特征层间学习改为粒度学习，并以残差方式布局；对所述测试集进行精度验证。其中，还包括：采用边缘损失函数来辅助模型的学习，所述边缘损失函数的定义如式(1)和式(2)所示：boundary＝I-Ero(I)(1)；其中，boundary代表所提取的储油罐的边缘，I为提取的储油罐结果图，Ero(I)是对图像I采用腐蚀计算得到的图像，Bpred表示模型提取的储罐边缘，Bgt表示储罐实际边缘信息，Lboundary为边缘损失值。其中，所述储油罐提取模型整个模型结构的损失函数采用二值交叉熵BCE与所述边缘损失函数相结合的方式，如下式所示：L＝μLbce+Lboundary；其中，μ值设为2，L为整个储油罐提取模型的损失值，Lbce为采用二值交叉熵BCE部分的损失值，Lboundary为边缘损失值。其中，所述深度学习网络Res2-Unet包括编码和解码，其中所述编码包括原始输入图像经过逐层Res2Net卷积运算和池化操作得到图像特征；所述解码包括所述图像特征经过逐层反卷积，特征大小逐步增大，维数逐步降低，最后生成与输入图像大小一致的单维二值结果图。其中，所述Res2Net卷积模块将1x1卷积以后的特征图从通道这一维度分为四块，得到四个子特征图x1、x2、x3和x4；其中，x1直接输入y1，x2经过3x3卷积得到特征图y2，x3与y2拼接，共同经过3x3卷积得到特征图y3，y3与x4拼接，共同经过3x3卷积得到特征图y4，y1、y2、y3和y4拼接，共同经过1x1卷积，再与原始特征图拼接，得到最终Res2Net卷积模块的输出特征图。其中，所述采集储油罐样本是基于GF1、GF2、GF6和ZY卫星影像，通过目视解译采集的。其中，对所述储油罐样本进行如下处理：将每景影像根据油罐真实所处位置随机裁剪成512x512像素大小的图块，影像通道选取绿色、蓝色和近红外三个通道。其中，在所述储油罐提取模型训练过程中，先将输入数据逐级编码成多通道的特征，再通过与编码过程中生成的特征串联的方式，逐级解码到与输入图像相同大小的分割结果二值图，在二值图中，油罐标记为1，背景地物为0。其中，所述储油罐提取模型的学习率设置为0.01，优化策略采用随机梯度下降法。其中，所述对所述测试集进行精度验证计算了交并比、F1值、召回率和精度以评估测试集的精度。基于上述技术方案可知，本专利技术的基于高空间分辨率遥感影像的储油罐提取方法相对于现有技术至少具有如下有益效果之一：A、采用Res2net模块改进Unet网络结构，增强模型对储油罐多尺度特征的学习能力，可以增强对不同尺度储油罐的提取能力，进而提高模型的鲁棒性。B、采用Res2-Unet结构构建储油罐提取模型，模型结构简单，易与训练，而且采用unet的网络基本架构，对输入影像特征逐层编码和解码，得到最终的分类结果，可以增强对不同种类储油罐的提取能力，提高模型的可迁移性。C、采用边缘损失boundaryloss作为辅助损失函数增强模型对提取的储油罐边缘信息，进而保证其每个提取得到的储油罐的完整性，降低其同物异谱带来的影响。附图说明图1为本专利技术实施例提供的方法实现流程图；图2为本专利技术实施例提供的Res2-Unet网络结构图；图3为本专利技术实施例提供的Res2net卷积模块示意图，其中，x1、x2、x3、x4为特征图的四个子通道集，y1、y2、y3、y4为相应的卷积之后形成的特征子集图；图4为本专利技术实施例提供的储油罐提取图，其中，(a)和(c)为原始影像，(b)和(d)为储油罐提取二值图。具体实施方式为使本专利技术的目的、技术方案和优点更加清楚明白，以下结合具体实施例，并参照附图，对本专利技术作进一步的详细说明。如图1所示，为本专利技术所提出的方法流程图，具体包括以下步骤：步骤1：采集储油罐样本，将所述样本随机本文档来自技高网...

【技术保护点】
1.一种基于高空间分辨率遥感影像的储油罐提取方法，其特征在于，包括：/n采集储油罐样本，将所述样本随机分为训练集和测试集；/n基于Res2-Unet模型结构构建储油罐提取模型；其中，基于UNet语义分割结构提出深度学习网络Res2-Unet，采用Res2Net卷积模块，将特征层间学习改为粒度学习，并以残差方式布局；/n对所述测试集进行精度验证。/n

【技术特征摘要】
1.一种基于高空间分辨率遥感影像的储油罐提取方法，其特征在于，包括：
采集储油罐样本，将所述样本随机分为训练集和测试集；
基于Res2-Unet模型结构构建储油罐提取模型；其中，基于UNet语义分割结构提出深度学习网络Res2-Unet，采用Res2Net卷积模块，将特征层间学习改为粒度学习，并以残差方式布局；
对所述测试集进行精度验证。


2.根据权利要求1所述的储油罐提取方法，其特征在于，还包括：
采用边缘损失函数来辅助模型的学习，所述边缘损失函数的定义如式(1)和式(2)所示：
boundary＝I-Ero(I)(1)；



其中，boundary代表所提取的储油罐的边缘，I为提取的储油罐结果图，Ero(I)是对图像I采用腐蚀计算得到的图像，Bpred表示模型提取的储罐边缘，Bgt表示储罐实际边缘信息，Lboundary为边缘损失值。


3.根据权利要求2所述的储油罐提取方法，其特征在于，所述储油罐提取模型整个模型结构的损失函数采用二值交叉熵BCE与所述边缘损失函数相结合的方式，如下式所示：
L＝μLbce+Lboundary；
其中，μ值设为2，L为整个储油罐提取模型的损失值，Lbce为采用二值交叉熵BCE部分的损失值，Lboundary为边缘损失值。


4.根据权利要求1所述的储油罐提取方法，其特征在于，所述深度学习网络Res2-Unet包括编码和解码，其中所述编码包括原始输入图像经过逐层Res2Net卷积运算和池化操作得到图像特征；所述解码包括所述图像特征经过逐层反卷积，特征大小逐步增大，维数逐步...

【专利技术属性】
技术研发人员：于博，王玉，陈方，王雷，
申请(专利权)人：中国科学院空天信息创新研究院，
类型：发明
国别省市：北京;11