[发明专利]一种确定对象信息的方法、装置、设备及存储介质

权利要求书

1.一种确定对象信息的方法，其特征在于，所述方法包括：

获取目标图像，所述目标图像上包括至少一个对象以及所述对象在所述目标图像上的二维标签；

将所述目标图像输入至预先完成训练的机器学习模型，以便由所述机器学习模型输出得到所述二维标签指示的对象所对应的对象信息，所述对象信息至少包括所述对象的位置、方向角以及尺寸中的任一种信息；

其中，所述机器学习模型预先基于样本图像以及所述样本图像上包括的对象所对应的对象信息完成训练。

2.根据权利要求1所述的方法，其特征在于，所述二维标签为二维分割标签，所述获取目标图像，包括：

获取初始目标图像，所述初始目标图像上包括所述对象；

在所述目标图像上对所述对象进行图像分割，得到包括所述对象在所述目标图像上的二维分割标签的所述目标图像。

3.根据权利要求1或2所述的方法，其特征在于，所述方法还包括：

获取样本图像，所述样本图像上包括至少一个对象以及所述对象在所述样本图像上的二维标签；

确定所述样本图像上所述对象对应的对象信息；

将所述样本图像作为所述机器学习模型的输入，将所述样本图像上包括的对象所对应的对象信息作为所述机器学习模型的输出，对所述机器学习模型进行训练。

4.根据权利要求3所述的方法，其特征在于，所述确定所述样本图像上所述对象对应的对象信息，包括：

获取利用雷达检测得到的点云数据；

从所述点云数据中确定出与所述样本图像上所述对象对应的目标点云数据；

根据所述样本图像上所述对象对应的目标点云数据，确定所述样本图像上所述对象对应的对象信息。

5.一种确定对象信息的装置，其特征在于，所述装置包括：

第一获取模块，用于获取目标图像，所述目标图像上包括至少一个对象以及所述对象在所述目标图像上的二维标签；

输入模块，用于将所述目标图像输入至预先完成训练的机器学习模型，以便由所述机器学习模型输出得到所述二维标签指示的对象所对应的对象信息，所述对象信息至少包括所述对象的位置、方向角以及尺寸中的任一种信息；

其中，所述机器学习模型预先基于样本图像以及所述样本图像上包括的对象所对应的对象信息完成训练。

6.根据权利要求5所述的装置，其特征在于，所述二维标签为二维分割标签，所述第一获取模块，包括：

第一获取单元，用于获取初始目标图像，所述初始目标图像上包括所述对象；

图像分割单元，用于在所述目标图像上对所述对象进行图像分割，得到包括所述对象在所述目标图像上的二维分割标签的所述目标图像。

7.根据权利要求5或6所述的装置，其特征在于，所述装置还包括：

第二获取模块，用于获取样本图像，所述样本图像上包括至少一个对象以及所述对象在所述样本图像上的二维标签；

确定模块，用于确定所述样本图像上所述对象对应的对象信息；

训练模块，用于将所述样本图像作为所述机器学习模型的输入，将所述样本图像上包括的对象所对应的对象信息作为所述机器学习模型的输出，对所述机器学习模型进行训练。

8.根据权利要求7所述的装置，其特征在于，所述确定模块，包括：

第二获取单元，用于获取利用雷达检测得到的点云数据；

第一确定单元，用于从所述点云数据中确定出与所述样本图像上所述对象对应的目标点云数据；

第二确定单元，用于根据所述样本图像上所述对象对应的目标点云数据，确定所述样本图像上所述对象对应的对象信息。

9.一种设备，其特征在于，所述设备包括处理器以及存储器：

所述存储器用于存储程序代码，并将所述程序代码传输给所述处理器；

所述处理器用于根据所述程序代码中的指令执行权利要求1至4任一项所述的确定对象信息的方法。

10.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质用于存储计算机程序，所述计算机程序用于执行权利要求1-4中任一项所述的确定对象信息的方法。