[发明专利]一种物体识别方法及装置

说明书

本申请公开了一种物体识别方法及装置。该方法包括：获取三维点云数据；基于三维点云数据生成第一张量；将第一张量输入预先完成训练的识别模型，得到识别模型对第一张量处理后输出的当前环境中物体的位置信息和类别信息。本申请公开的技术方案，基于三维点云数据生成第一张量的过程，是将三维点云数据映射成多个通道的二维点云数据，从而将非结构化的数据转化为结构化数据，便于识别模型进行特征提取，从而提高物体识别效率。

技术领域

本申请属于物体检测技术领域，尤其涉及一种物体识别方法及装置。

背景技术

在智能驾驶领域中，环境感知是非常重要的部分，准确地确定环境中物体的位置和类别，是保证车辆安全行驶的必要条件。目前，主要通过激光雷达获取三维点云数据，基于三维点云数据确定物体的位置和类别。

但是，目前基于三维点云数据进行的物体识别存在效率低的问题。

发明内容

有鉴于此，本申请的目的在于提供一种物体识别方法及装置，以提高物体识别效率。

为实现上述目的，本申请提供如下技术方案：

一方面，本申请提供一种物体识别方法，包括：

获取三维点云数据；

基于所述三维点云数据生成第一张量，所述第一张量为P阶，所述第一张量的每一阶包括H行W列元素，所述第一张量的第k阶中的元素由所述三维点云数据中位于第k个预设高度范围内点的反射强度确定，所述第一张量的第k阶中位于第i行第j列的元素为：所述三维点云数据中位于第k个预设高度范围的第i行第j列的栅格内点的反射强度的平均值，其中，在水平方向上将所述三维点云数据对应的三维空间划分为H行W列的第一网格，在高度方向上将每个第一网格划分为P个栅格，P、H和W为大于2的整数，k＝1,2,…P，i＝1,2,…H，j＝1,2,…W；

将所述第一张量输入预先完成训练的识别模型，得到所述识别模型对所述第一张量处理后输出的当前环境中物体的位置信息和类别信息；所述识别模型根据三维点云样本数据训练得到，所述三维点云样本数据中样本物体的位置信息和类别信息已知。

可选的，所述识别模型对所述第一张量的处理，包括：

对所述第一张量进行特征提取，得到原始特征；

基于所述原始特征生成第二张量，所述第二张量指示各个第二网格中物体元素的类别信息，所述第一张量对应的三维空间在水平方向上划分为M行N列的第二网格，所述第二张量为C1阶，所述第二张量的每一阶包括M行N列元素，其中，C1为预设的物体类别的数量，所述第二张量中位于相同行相同列的C1个元素为同一个第二网格中物体元素属于各个类别的概率，所述第二张量的第a阶中的元素为各第二网格中的物体元素属于第a个类别的概率，a＝1,2,…C1；

基于所述原始特征生成第三张量，所述第三张量指示所述各个第二网格的相对位置信息，所述第三张量为C2阶，所述第三张量的每一阶包括M行N列元素，其中，C2为所述各个第二网格的相对位置信息所包含相对位置参数的数量，所述第三张量中位于相同行相同列的C2个元素为同一个第二网格的相对位置参数的值，所述第三张量的第b阶中的元素为所述各个第二网格的相对位置信息中第b个相对位置参数的值，b＝1,2,…C2；

根据所述第二张量指示的类别信息和所述第三张量指示的相对位置信息确定当前环境中物体的位置信息和类别信息。

可选的，所述根据所述第二张量指示的类别信息和所述第三张量指示的相对位置信息确定当前环境中物体的位置信息和类别信息，包括：

根据所述第二张量在所述各个第二网格中确定类别为第一类别的目标网格，所述第一类别为预设的C1个物体类别中的任意一个；

从所述第三张量获取所述目标网格的相对位置信息；