[发明专利]数据标注方法、装置、计算机设备及存储介质

说明书

本公开提供了一种数据标注方法、装置、计算机设备及存储介质，获取待处理的影像数据；所述影像数据携带标签信息；所述影像数据包括对目标空间进行图像采集得到的视频或者图像；基于所述影像数据，得到所述目标空间中对象的矢量模型、以及所述矢量模型在所述目标空间中的第一三维位置；以及基于所述标签信息在所述图像中的标注位置，确定所述标签信息在所述目标空间中的第二三维位置；基于所述第一三维位置、和所述第二三维位置，从所述对象中，为所述标签信息确定对应的目标对象；建立所述目标对象对应的矢量模型、和所述标签信息之间的关联关系。

技术领域

本公开涉及计算机技术领域，具体而言，涉及一种数据标注方法、装置、计算机设备及存储介质。

背景技术

为了便于工作人员对机房、设备室等目标空间内设备的标注，可以采用对目标空间进行拍摄，并在拍摄的图像上进行数据标注的方式进行。由于是在二维图像上对各个对象进行标注，因此需要在建立了目标空间的三维模型后，需要将二维图像中的标注数据匹配至三维模型中的目标对象。而当前将二维图像上的标注数据匹配至三维模型中的目标对象时，存在匹配准确性低的问题。

发明内容

本公开实施例至少提供一种数据标注方法、装置、计算机设备及存储介质。

第一方面，本公开实施例提供了一种数据标注方法，包括：获取待处理的影像数据；所述影像数据携带标签信息；所述影像数据包括对目标空间进行图像采集得到的视频或者图像；基于所述影像数据，得到所述目标空间中对象的矢量模型、以及所述矢量模型在所述目标空间中的第一三维位置；以及基于所述标签信息在所述图像中的标注位置，确定所述标签信息在所述目标空间中的第二三维位置；基于所述第一三维位置、和所述第二三维位置，从所述对象中，为所述标签信息确定对应的目标对象；建立所述目标对象对应的矢量模型、和所述标签信息之间的关联关系。

这样，由于可以将矢量模型以及标签信息分别对应的位置转换至同一三维坐标系下，因此利用确定的第一三维位置以及第二三维位置，可以较为准确地为标签信息确定匹配的目标对象。

一种可选的实施方式中，所述基于所述影像数据，得到所述目标空间中对象的矢量模型，包括：基于所述影像数据，对所述目标空间进行三维重建，得到所述目标空间的点云模型；对所述点云模型进行语义分割处理，得到所述点云模型中各个点云点的语义信息；基于所述点云模型中各个点云点的语义信息，生成所述目标空间中的对象对应的实体分割模型；基于所述对象对应的实体分割模型，确定所述对象对应的矢量模型。

这样，对于目标空间中相同类型或者相同型号的对象，利用点云点的位置信息以及语义信息，可以对点云点进行较为容易且较为准确的划分，避免将不同对象的点云点归至同一对象中。

一种可选的实施方式中，所述基于所述对象对应的实体分割模型，确定所述对象对应的矢量模型，包括：基于所述对象的实体分割模型中各个点云点分别对应的位置信息，对该对象进行几何形状拟合，得到所述对象对应的矢量模型。

这样，通过对象的实体分割模型中各个点云点分别对应的位置信息，对各个对象进行几何形状拟合，能够更加准确地反映目标对象的轮廓，提高矢量化的准确性。

一种可选的实施方式中，所述基于所述对象对应的实体分割模型，确定所述对象对应的矢量模型，包括：将所述对象对应的实体分割模型，与矢量模型库中的预设矢量模型进行匹配，从所述矢量模型库中，为所述对象确定对应的矢量模型。

这样，由于利用实体分割模型生成矢量模型的方式较为复杂，因此在数据标注时若直接生成矢量模型，可能会对数据标注的效率产生影响。采用匹配的方式，一方面可以减少生成矢量模型的步骤，以提高效率；另一方面，由于矢量模型库中的矢量模型在预先生成时即可以较为准确的根据对象的类型和型号生成，因此通过匹配的方式为对象确定的矢量模型也较为准确。