一种基于邻域关系的火星栅格地形地图生成方法技术

本发明专利技术公开了一种基于邻域关系的火星栅格地形地图生成方法，属于计算机图像处理技术领域。该火星栅格地形地图生成方法将收集的火星近地面图片输入地形分类网络中进行训练，当交叉熵损失函数收敛时，完成对地形分类网络的训练，以训练好的地形分类网络为基础，在火星近距离地面图片上划分栅格并逐窗口地预测地形类别，后续在上述栅格地形地图上进行结果的平滑，设计了基于邻域关系的多尺度融合与投票平滑算法，最终生成栅格地形地图。本发明专利技术的栅格地形地图生成方法仅依赖于能够提供火星近地面图片的传感器设备，方法简单，是基于导航相机、避障相机等设备实现地形理解的有效方法。法。法。

【技术实现步骤摘要】
一种基于邻域关系的火星栅格地形地图生成方法


[0001]本专利技术属于计算机图像处理
，尤其涉及一种基于邻域关系的火星栅格地形地图生成方法。

技术介绍

[0002]地形地图的生成是计算机视觉领域中的一个重要任务，被广泛运用在障碍避让、路径规划等诸多任务场景。尤其在当前的地外探测任务中，地面地形的识别是探险车实现环境语义理解、安全路径搜寻的基础。
[0003]区别于熟悉可见的地球地表环境，火星地外环境中地面数据存在着非结构化场景差异不明显、危险区域难界定、目标先验知识欠缺、地形纹理难区分等问题。现有的解决方案包括以下三种：（1）使用语义分割算法实现像素级别的地形识别；（2）由目标检测完成对典型目标的侦测后，组合判断地形种类；（3）直接通过分类模型给出粗略的结果。其中，语义分割方法需要专家级的地质知识、大量的高质量标注数据以及高性能的推理设备；而目标检测的方法则依赖于各个子目标识别效果；至于直接分类则往往只适用于单地形场景。

技术实现思路

[0004]针对现有技术的不足，本专利技术提供了一种基于邻域关系的火星栅格地形地图生成方法。
[0005]本专利技术的目的是通过以下技术方案来实现的：一种基于邻域关系的火星栅格地形地图生成方法，包括以下步骤：（1）收集火星近地面图片，进行地形类别标注，并对火星近地面图片进行随机翻转、剪裁、伽马变换进行图像增强，获得火星近地面图片数据集；（2）将步骤（1）获得的火星近地面图片数据集输入地形分类网络中进行训练，当交叉熵损失函数收敛时，完成对地形分类网络的训练；（3）将火星近距离地面图片划分栅格，选取栅格中的第一个栅格，所述第一个栅格采用若干组0~3的系数向外进行多尺度的扩张，形成若干组候选区域，所述候选区域在邻域范围内，所述邻域是以第一个栅格为中心的5
×
5栅格区域，并将所述若干组候选区域依次输入训练好的地形分类网络中，分别得到地形预测的概率值；（4）在第一个栅格的邻域区间中，计算各栅格候选区域与第一个栅格的第一个候选区域之间的重叠面积，生成相应的投票权重，再根据步骤（3）得到的地形预测的概率值，生成平滑的后概率值：（5）采用步骤（4）的方法遍历第一个栅格的所有候选区域，生成平滑的后概率值，融合所有后概率值，得到第一个栅格的平滑概率值；（6）将步骤（5）中生成的平滑概率值中最大概率值的地形类别作为标签，并进一步在第一个栅格中平滑掉特异性的结果，获得第一个栅格的地形地图；（7）遍历栅格中的每一个栅格，采用与第一个栅格相同的系数向外进行多尺度的扩张，
重复步骤（3）-（7），获得每一个栅格的地形地图。
[0006]与现有技术相比，本专利技术的有益效果是：本专利技术是基于邻域关系的栅格地形地图主要针对火星近地面图片进行地形识别，并在全局图像中构建稠密的栅格地图。由于，实际工程中因栅格大小不定、输入区域扰动等因素，独立的栅格预测会带来全局结果一致性差、平滑性欠缺等问题。本专利技术的栅格地图生成方法中引入多尺度预测加权融合、邻域关系投票平滑方法，从而提升火星栅格地形地图的整体表现，使得本专利技术的栅格地图生成方法仅依赖于能够提供火星近地面图片的传感器设备，方法简单，是基于导航相机、避障相机等设备实现地形理解的有效方法。
附图说明
[0007]图1为地形分类模型总体架构示意图；图2为栅格地形地图生成流程图；图3为本专利技术的效果示意图，图3中的（a）为火星近距离地面图片，图3中的（b）为通过本专利技术方法形成的栅格地形地图。
具体实施方式
[0008]本专利技术提供了一种基于邻域关系的火星栅格地形地图生成方法，包括以下步骤：（1）以NASA公开的火星数据集“Mars32k”作为任务样本，并从中挑选了部分火星近地面图片进行地形类别的标注。该图片集由火星车“好奇号”拍摄，包含了约32,000张涵盖了火星的各种地理和地质特征的彩色图片，如山脉和山谷，火山口，沙丘和岩石等地形。任务中总计选用了涵盖了10种地形，并对火星近地面图片进行随机翻转、剪裁、伽马变换进行图像增强，获得火星近地面图片数据集。
[0009]构建地形分类网络：如图1为地形分类模型总体架构示意图，所述地形分类网络基于开源的深度学习框架pytorch，在公开的网络模型的基础上进行改造，实现了主体网络、层级特征融合模块、损失函数模块的构建，具体构建过程如下：（a）构建主体网络：主体网络包括特征提取模块与分类器模块两个部分，所述主体网络可以直接选用公开的Resnet101结构，将Resnet101结构中的分类器的输出数目设为10类，与当前地形分类任务相匹配。
[0010]（b）构建层级特征融合模块：融合特征提取模块中的四个子模块分别输出特征图，四组特征图经由第一映射函数后，并下采样至尺寸与特征图保持一致：其中，下采样函数选用双线性差值函数；第一映射函数选用标准的3
×
3卷积操作。
[0011]函数选用函数，在通道上拼接上述四组特征图，并通过第二映射函数映射，得到融合的特征图，实现通道的调整，融合的特征图与特征图相加输入分类器模块，计算如下：
其中，第二映射函数为两组连续的包含3
×
3卷积、批量标准化、线性整流函数三个部分的组合模块。
[0012]（c）损失函数模块：以交叉熵损失作为损失函数，拟合地形分类概率与真实的标签，并以批量随机梯度下降法作为优化方法，反向传播并更新整个地形分类网络。
[0013]（2）将步骤（1）获得的火星近地面图片数据集输入地形分类网络中进行训练，当交叉熵损失函数收敛时，完成对地形分类网络的训练；（3）以地形分类网络为基础，在火星近距离地面图片上划分栅格并逐窗口地预测地形类别，后续在上述栅格地形地图上进行结果的平滑，设计了基于邻域关系的多尺度融合与投票平滑算法，最终生成地形地图，定义栅格与邻域并实现多尺度推理以适应不同尺度的地形目标；在邻域内以投票的方式平滑预测概率以提升结果的区域关联性；以特异性指标抑制局部预测偏差的问题。如图2所示为栅格地形地图生成流程图，包括以下过程：（3.1）栅格邻域内的多尺度推理将火星近距离地面图片划分栅格并从中选取第一个栅格。所述第一个栅格采用j组系数向外进行多尺度的扩张形成若干组候选区域。则第一个栅格的某一候选区域记为：其中，表示栅格外扩函数，为当前候选区域的外扩系数，m为j的索引。
[0014]所述候选区域在邻域范围内，邻域是以第一个栅格为中心外扩的5
×
5栅格区域。若干组候选区域依次输入训练好的地形分类网络中，分别得到地形预测的概率值，则第一个栅格的所有预测概率值集合记为。其中，为地形类别的数目。
[0015]（3.2）基于邻域关系的栅格平滑算法在第一个栅格的邻域区间中，计算各栅格候选区域与第一个栅格的候选区域之间的重叠面积，生成相应的投票权重，再根据步骤（3.1）得到的候选区域对应的地形概率，生成平滑的后概率值：其中，投票函数为栅格赋予一个以第一个栅格为中心的高斯分布权重，为第一个栅格的第m个预测概率值。
[0016]随后，遍历第一个栅本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于邻域关系的火星栅格地形地图生成方法，其特征在于，包括以下步骤：（1）收集火星近地面图片，进行地形类别标注，并对火星近地面图片进行随机翻转、剪裁、伽马变换进行图像增强，获得火星近地面图片数据集；（2）将步骤（1）获得的火星近地面图片数据集输入地形分类网络中进行训练，当交叉熵损失函数收敛时，完成对地形分类网络的训练；（3）将火星近距离地面图片划分栅格，选取栅格中的第一个栅格，所述第一个栅格采用若干组0~3的系数向外进行多尺度的扩张，形成若干组候选区域，所述候选区域在邻域范围内，所述邻域是以第一个栅格为中心的5
×
5栅格区域，并将所述若干组候选区域依次输入训练好的地形分...

【专利技术属性】
技术研发人员：李月华，张健，宋俊男，谢天，
申请(专利权)人：之江实验室，
类型：发明
国别省市：