一种基于遥感图像的目标检测与识别方法与流程

本发明属于图像识别技术领域，尤其涉及一种基于遥感图像的目标检测与识别方法。

背景技术：

在利用遥感图像进行信息提取和分析时，很多任务特别是涉及到地理范围较大、时间周期跨度较长时都需要结合不同遥感卫星，不同载荷的图像数据，而当采用这些图像数据时必然会涉及到不同空间分辨率、不同光谱谱段、不同时间采集的情况，因此需要根据不同遥感卫星、不同载荷的具体空间分辨率、时间分辨率、光谱谱段等特点，按照任务需求进行合理评估后，筛选出能满足任务需求的遥感图像数据，并按照各自遥感图像数据特点进行专业化针对性的前期处理，将来自不同源的基础产品按照具体任务要求进行空间分辨率、投影等的统一组织，以满足综合利用遥感卫星多源图像数据特点完成具体任务目的。

在遥感图像上进行目标检测和识别是一项在高分辨遥感图像上进行目标提取和分析时常规、基础且应用广泛的任务，在很多环境下都有大量的运用，是很多领域的基本且重要的输入信息。

现有的方法主要：

(1)根据目标样本经过目视解译进行目标识别。利用行业专家收集、整理、总结的目标样本表述，专业解译人员根据目标所特有的特征在遥感图像上逐个寻找和被识别目标一致的目标，并进行详细的勾绘；由于在不同遥感图像上被识别目标的特征和图像条件可能会发生变化，因此其识别目标不准确。

该方法需要经过专业的培训和积累丰富的行业经验才能胜任具体的目标检测与识别工作，而且在进行地理范围较大、时间跨度较长的任务时需要大量的解译人员参与，完成工作的效率低下、耗时长，容易造成项目的延误和滞后；在完成工作时需要进行反复的检查和核对，以便能将多人协作的成果进行统一协调和汇总。

(2)根据专属算法和目标规则进行半自动的提取和识别。

该方法通过在遥感图像上选取一定数量的包含目标的样本，同时根据识别目标的特点定义详细规则并设定对应的判定条件，经过迭代训练优化专属算法或模型的参数，达到满足指定精度指标或收敛条件后，从整个遥感图像或同类遥感图像上识别出所有目标。

专属算法虽然能半自动的进行目标的检测和识别，但是该方法需要根据具体的任务进行针对性的设计，需要对识别目标样本有很高的认识和专业知识作为指导；选取样本的质量、定义的规则、所定义规则的充分性以及采用的算法或模型，会严重影响识别的结果和精度；容易受到图像数据的变化而变化，需要大量的人工干预和根据要求对规则的进行优化调整，应用范围受限制和普适性不高，存在误判，检测和识别效率不高。

技术实现要素：

针对现有技术中的上述不足，本发明提供的一种遥感图像的目标检测与识别方法，可以大范围地、快速地、精确地检测和识别出目标。

为了达到以上目的，本发明采用的技术方案为：

本方案提供一种遥感图像的目标检测与识别方法，包括以下步骤：

s1、根据识别目标筛选出多源高分辨率的遥感图像；

s2、对所述遥感图像进行预处理；

s3、提取遥感图像的目标样本；

s4、将提取的目标样本划分成训练样本以及测试样本；

s5、利用所述训练样本训练目标识别模型，并利用所述测试样本对所述目标识别模型进行测试；

s6、利用经测试后的目标识别模型检测和识别出遥感图像中的目标，完成对目标的检测与识别。

进一步地，所述步骤s2包括以下步骤：

s201、对遥感图像进行几何变换；

s202、利用空域法对经几何变换后的遥感图像进行增强处理，完成对遥感图像的预处理。

再进一步地，所述步骤s203中增强处理的表达式如下：

g(x,y)＝f(x,y)*h(x,y)

其中，g(x,y)表示经增强处理后的遥感图像，f(x,y)表示原始遥感图像，h(x,y)表示空间转换函数，x,y均表示坐标。

再进一步地，所述步骤s3包括以下步骤：

s301、对经预处理后的遥感图像进行归一化处理；

s302、根据所述归一化处理后的遥感图像，计算得到遥感图像中每个像素的梯度方向值；

s303、将遥感图像划分若干个单元格，并对每个单元格进行权重投影处理；

s304、根据所述梯度方向值，统计每个单元格的梯度直方图；

s305、将若干个单元格组成一个区域块，并将区域块内重叠的单元格的梯度直方图进行归一化处理；

s306、将经归一化处理后的重叠的单元格的梯度直方图进行串联，完成对遥感图像的目标样本提取。

再进一步地，所述步骤s302中每个像素的梯度方向值的表达式如下：

gx(x,y)＝h(x+1,y)-h(x-1,y)

gy(x,y)＝h(x,y+1)-h(x,y-1)

其中，a(x,y)表示每个像素的梯度方向值，gx(x,y)、gy(x,y)和h(x,y)分别表示遥感图像中像素点(x,y)处的水平方向梯度，垂直方向梯度和像素值。

再进一步地，所述步骤s5中目标识别模型包括依次连接的输入层、第一卷积层、第一最大池化层、第二卷积层、第二最大池化层、第三卷积层、第四卷积层、第五卷积层、第六卷积层、第三最大池化层、第七卷积层、第八卷积层、第九卷积层、第十卷积层、第四最大池化层、第十一卷积层、第十二卷积层、第十三卷积层、第十四卷积层、第十五卷积层、第十六卷积层、第十七卷积层、第十八卷积层、第一全连接层、第二全连接层以及输出层。

再进一步地，所述第一卷积层的卷积核大小为7×7，其维度为64，步长为2；所述第一最大池化层的卷积核大小为2×2，步长为2；所述第二卷积层的卷积核大小为3×3，其维度为192；所述第二最大池化层的卷积核大小为2×2，步长为2；所述第三卷积层的卷积核大小为1×1，其维度为128；所述第四卷积层的卷积核大小为3×3，其维度为256；所述第五卷积层的卷积核大小为1×1，其维度为256；所述第六卷积层的卷积核大小为3×3，其维度为512；所述第三最大池化层的卷积核大小为2×2，步长为2；所述第七卷积层的卷积核大小为1×1，其维度为256；所述第八卷积层的卷积核大小为3×3，其维度为512；所述第九卷积层的卷积核大小为1×1，其维度为512；所述第十卷积层的卷积核大小为3×3，其维度为1024；所述第四最大池化层的卷积核大小为2×2，步长为2；所述第十一卷积层的卷积核大小为1×1，其维度为512；所述第十二卷积层的卷积核大小为3×3，其维度为1024；所述第十三卷积层的卷积核大小为3×3，其维度为1024；所述第十四卷积层的卷积核大小为3×3，其维度为1024，步长为2；所述第十五卷积层的卷积核大小为3×3，其维度为1024；所述第十六卷积层的卷积核大小为3×3，其维度为1024。

再进一步地，所述目标识别模型的损失函数的表达式如下：

其中，l表示目标识别模型的损失函数，λcoord表示预测数据与标定数据之间的坐标误差修正值，λnoobj表示误差修正值，classes表示分类误差，s和b均表示全连接层的输出维度，表示落入区域i中的目标物体，表示落入区域i的第j个边框中的目标物体，xi、yi、wi、ci和pi均表示神经网络预测值，和均表示标注值，hi表示神经网络预测的图像边框高度，表示遥感图像边框高度，c表示物体个数。

本发明的有益效果：

(1)本发明对涉及的高分辨遥感图像进行综合协调组织在选取少量样本后，经过对样本进行自动学习便可以快速、精确的检测和识别出目标，最终的成果能根据具体要求进行快速的导出(可以导出kml、geojson、shapefile、xlsx、csv等格式)。

(2)本发明将优化后目标识别模型应用到筛选出遥感图像上的目标识别任务，有效地减少计算难度。

(3)本发明不需要培训和协调大量的参与人员，避免时间和资源的浪费。无需过多的专业知识和先导经验，操作简单且容易掌握。特征提取和识别过程都是自动化的，无需人工设计和干预，其中，抽取出的特征特别是基础特征，具有良好的迁移性和普适性。能应用到空间分辨率相近的遥感图像上同类和类似目标识别任务，不用重新设计和节省大量的训练时间。

(4)本发明通过收集更多的有效和优质的样本数据，通过对目标识别模型自动反复的迭代升级，能进一步地提高识别的精度高，同时能应用到更多的目标识别任务。

(5)本发明利用各卷积层用来提取图像特征，以及利用各全连接层用来预测图像位置和类别概率值，防止过拟合，以及利用损失函数提高模型识别精度，降低误差。

附图说明

图1为本发明的方法流程图。

图2为本实施例中目标识别模型的结构示意图。

具体实施方式

下面对本发明的具体实施方式进行描述，以便于本技术领域的技术人员理解本发明，但应该清楚，本发明不限于具体实施方式的范围，对本技术领域的普通技术人员来讲，只要各种变化在所附的权利要求限定和确定的本发明的精神和范围内，这些变化是显而易见的，一切利用本发明构思的发明创造均在保护之列。

实施例

如图1所示，本发明提供了基于遥感图像的目标检测与识别方法，其实现方法如下：

s1、根据识别目标筛选出多源高分辨率的遥感图像；

s2、对遥感图像进行预处理，其实现方法如下：

s201、对遥感图像进行几何变换；

s202、利用空域法对经几何变换后的遥感图像进行增强处理，完成对遥感图像的预处理：

g(x,y)＝f(x,y)*h(x,y)

其中，g(x,y)表示经增强处理后的遥感图像，f(x,y)表示原始遥感图像，h(x,y)表示空间转换函数，x,y均表示坐标；

s3、提取遥感图像的目标样本，其实现方法如下：

s301、对经预处理后的遥感图像进行归一化处理；

s302、根据归一化处理后的遥感图像，计算得到遥感图像中每个像素的梯度方向值；

其中，每个像素的梯度方向值的表达式如下：

gx(x,y)＝h(x+1,y)-h(x-1,y)

gy(x,y)＝h(x,y+1)-h(x,y-1)

其中，a(x,y)表示每个像素的梯度方向值，gx(x,y)、gy(x,y)和h(x,y)分别表示遥感图像中像素点(x,y)处的水平方向梯度、垂直方向梯度和像素值；

s303、将遥感图像划分若干个单元格，并对每个单元格进行权重投影处理；

s304、根据梯度方向值，统计每个单元格的梯度直方图；

s305、将若干个单元格组成一个区域块，并将区域块内重叠的单元格的梯度直方图进行归一化处理；

s306、将经归一化处理后的重叠的单元格的梯度直方图进行串联，完成对遥感图像的目标样本提取；

s4、将提取的目标样本划分成训练样本以及测试样本；

s5、利用所述训练样本训练目标识别模型，并利用所述测试样本对所述目标识别模型进行测试；

如图2所示，目标识别模型包括依次连接的输入层、第一卷积层、第一最大池化层、第二卷积层、第二最大池化层、第三卷积层、第四卷积层、第五卷积层、第六卷积层、第三最大池化层、第七卷积层、第八卷积层、第九卷积层、第十卷积层、第四最大池化层、第十一卷积层、第十二卷积层、第十三卷积层、第十四卷积层、第十五卷积层、第十六卷积层、第十七卷积层、第十八卷积层、第一全连接层、第二全连接层以及输出层，其中，第一卷积层的卷积核大小为7×7，其维度为64，步长为2；第一最大池化层的卷积核大小为2×2，步长为2；第二卷积层的卷积核大小为3×3，其维度为192；第二最大池化层的卷积核大小为2×2，步长为2；第三卷积层的卷积核大小为1×1，其维度为128；第四卷积层的卷积核大小为3×3，其维度为256；第五卷积层的卷积核大小为1×1，其维度为256；第六卷积层的卷积核大小为3×3，其维度为512；第三最大池化层的卷积核大小为2×2，步长为2；第七卷积层的卷积核大小为1×1，其维度为256；第八卷积层的卷积核大小为3×3，其维度为512；第九卷积层的卷积核大小为1×1，其维度为512；第十卷积层的卷积核大小为3×3，其维度为1024；第四最大池化层的卷积核大小为2×2，步长为2；第十一卷积层的卷积核大小为1×1，其维度为512；第十二卷积层的卷积核大小为3×3，其维度为1024；第十三卷积层的卷积核大小为3×3，其维度为1024；第十四卷积层的卷积核大小为3×3，其维度为1024，步长为2；第十五卷积层的卷积核大小为3×3，其维度为1024；第十六卷积层的卷积核大小为3×3，其维度为1024。

s6、利用经测试后的目标识别模型检测和识别出遥感图像中的目标，完成对目标的检测与识别，其中，目标识别模型的损失函数的表达式如下：

其中，l表示目标识别模型的损失函数，λcoord表示预测数据与标定数据之间的坐标误差修正值，λnoobj表示误差修正值，classes表示分类误差，s和b均表示全连接层的输出维度，表示落入区域i中的目标物体，表示落入区域i的第j个边框中的目标物体，xi、yi、wi、ci和pi均表示神经网络预测值，和均表示标注值，hi表示神经网络预测的图像边框高度，表示遥感图像边框高度，c表示物体个数。

本实施例中，根据具体识别的目标筛选出符合要求的多源高分辨率遥感图像，由于各自遥感卫星载荷的设计，选取的高分辨率图像具有针对每种高分辨遥感图像按照自身的特点进行预处理，主要包括几何变换以及图像增强处理。再提取目标样本，根据提取的目标样本从该图像上选取具有代表性的目标样本，选取满足一定数量样本并形成具有规范的样本。将选取的样本分成训练样本和测试样本，且每类样本都是随机选择和多组组合分配。选择深度学习模型，并设定模型结构及参数。参数主要包括卷积层数、卷积核大小、池化层数等。利用训练样本训练模型，深度学习模型中的不同的卷积层和池化层能自动的获取各尺度上不同的特征，利用测试样本对训练模型进行测试，通过准确率、精确率、召回率、roc曲线等对模型进行评估，得到满足各项精度要求的最优的目标识别模型。利用该优化的深度学习模型检测和识别出遥感图像上所有的目标。本实施例中通过对优化后深度学习模型经过参数的微调和简单训练，而不用从头开始选择样本和重新训练模型，就可以直接将其应用到筛选出遥感图像上的目标识别任务。

本发明通过以上设计，可以快速、精确的检测和识别出目标。