[发明专利]三维检测模型的训练方法、装置、计算机设备和存储介质

说明书

本申请涉及一种三维检测模型的训练方法、装置、计算机设备和存储介质。包括：获取样本行车区域图像和样本深度点云图像；确定样本深度点云图像中的样本点云所对应的样本点云特征；对样本深度点云图像进行随机扰动处理，并确定随机扰动后的样本深度点云图像中的每个样本点云各自对应的位于样本行车区域图像中的样本图像像素；对于样本深度点云图像中的每个样本点云，均将当前样本点云所对应的样本点云特征与相对应的样本图像像素的样本像素特征进行融合，得到样本目标点云特征；根据样本目标点云特征与点云特征标签间的差异，得到训练好的三维检测模型。采用本方法能够提高三维检测模型的训练准确性。

技术领域

本申请涉及计算机技术领域，特别是涉及一种三维检测模型的训练方法、装置、计算机设备和存储介质。

背景技术

随着计算机技术的发展，传感器信息融合技术在自动驾驶等领域扮演着重要角色，其中激光雷达和相机就是常用的传感器，相机对颜色等内容信息具有高分辨率的获取能力，而激光雷达对环境空间具有精确的三维测量能力，二者相互融合便能更准确的对移动目标进行识别。

目前，现有的图像数据与激光点云数据融合训练算法，需要建立二维平面图像到三维点云之间的映射模型，并在计算映射模型时需要在三维点云中寻找与二维平面图像上各标定点所对应的点云。

然而，激光雷达和相机易受外界环境影响使得位姿发生抖动和偏移，或者在发生拍摄延时，激光雷达和相机时间戳很难匹配准，因此，激光雷达和相机的时间及空间的不同步造成了映射模型训练准确性差。由此可以得出，如何准确地对图像数据与激光点云数据进行融合训练是本公开需要解决的问题。

发明内容

基于此，有必要针对上述技术问题，提供一种能够提高融合训练准确性的三维检测模型的训练方法、装置、计算机设备、计算机可读存储介质和计算机程序产品。

第一方面，本申请提供了一种三维检测模型的训练方法。所述方法包括：

获取样本行车区域图像、样本深度点云图像和所述样本深度点云图像中的样本点云所对应的点云特征标签；

确定所述样本深度点云图像中的样本点云所对应的样本点云特征；

对所述样本深度点云图像进行随机扰动处理，并确定随机扰动后的所述样本深度点云图像中的每个样本点云各自对应的位于所述样本行车区域图像中的样本图像像素；

对于所述样本深度点云图像中的每个样本点云，均将当前样本点云所对应的样本点云特征与相对应的样本图像像素的样本像素特征进行融合，得到样本目标点云特征；

根据所述样本目标点云特征与所述点云特征标签间的差异，对三维检测模型进行训练，直到达到训练停止条件时停止，得到训练好的三维检测模型；所述三维检测模型用于对行车区域中的目标对象进行识别。

在其中一个实施例中，所述对所述样本深度点云图像进行随机扰动处理，包括：获取所述样本深度点云图像对应的标定矩阵；确定所述标定矩阵对应的扰动旋转角和扰动偏移量，并根据所述扰动旋转角和所述扰动偏移量，确定所述标定矩阵对应的目标标定矩阵；通过所述目标标定矩阵对所述样本深度点云图像进行随机扰动处理。

在其中一个实施例中，所述通过所述目标标定矩阵对所述样本深度点云图像进行随机扰动处理，包括：获取所述样本深度点云图像对应的随机扰动概率；基于所述随机扰动概率，确定所述样本深度点云图像中目标样本点云；通过所述目标标定矩阵对所述目标样本点云进行随机扰动处理。

在其中一个实施例中，所述通过所述目标标定矩阵对所述目标样本点云进行随机扰动处理，包括：确定所述目标样本点云的点云坐标；通过所述目标标定矩阵对所述点云坐标进行转换，得到目标坐标；所述确定随机扰动后的所述样本深度点云图像中的每个样本点云各自对应的位于所述样本行车区域图像中的样本图像像素，包括：将所述样本行车区域图像中的具有所述目标坐标的样本图像像素作为与所述目标样本点云相对应的像素。