[发明专利]一种生成三维边界框的方法及装置

说明书

本发明公开一种生成三维边界框的方法和装置，以解决现有技术中二维边界框提供的目标对象信息不全面的技术问题。方法包括：获取待标定图片中目标对象的顶点信息，该顶点信息中包含多个构成所述目标对象的三维边界框的长、宽、高的已知顶点以及该已知顶点在目标对象中的位置信息；根据已知顶点确定构成所述三维边界框的剩余顶点；根据已知顶点的位置信息，确定剩余顶点在目标对象中的位置信息；根据所述已知顶点和剩余顶点及其位置信息，生成所述目标对象的三维边界框。采用本发明技术方案生成的三维边界框能够体现目标对象的体面信息，为后续提高目标检测的准确性提供依据。

技术领域

本发明涉及计算机领域，特别涉及一种生成三维边界框的方法及装置。

背景技术

目前，在一些特定领域需要对拍摄得到的图片中的目标对象进行标注，例如：在自动驾驶领域，需要对车辆、行人、三轮车、自行车、电动车、摩托车等进行标注，在安防、安全监控等领域需要对行人进行标注。目前，对目标对象进行标注主要采用二维边界框(如矩形框)的方式进行标注，如图1所示。该种通过2D边界框的标定方式，只能识别目标对象二维边界，但是并不能体现该目标对象的详细信息(例如目标对象的不同体面等)，从而使得该标注结果不能为后续的算法(例如目标检测)提供准确的依据，从而导致计算结果准确性较低。

发明内容

鉴于上述问题，本发明提供一种生成三维边界框的方法及装置，以解决现有技术中二维边界框提供的目标对象信息不全面的技术问题。

本发明实施例，一方面，提供一种生成三维边界框的方法，方法包括：

获取待标定图片中目标对象的顶点信息，该顶点信息中包含多个构成所述目标对象的三维边界框的长、宽、高的已知顶点以及该已知顶点在目标对象中的位置信息；

根据所述多个已知顶点确定构成所述三维边界框的剩余顶点；

根据已知顶点的位置信息，确定剩余顶点在目标对象中的位置信息；

根据所述已知顶点和剩余顶点及其位置信息，生成所述目标对象的三维边界框。

本发明实施例第二方面，提供一种生成三维边界框的装置，该装置包括：

获取单元，用于获取待标定图片中目标对象的顶点信息，该顶点信息中包含多个构成所述目标对象的三维边界框的长、宽、高的已知顶点以及该已知顶点在目标对象中的位置信息；

顶点确定单元，用于根据所述多个已知顶点确定构成所述三维边界框的剩余顶点；

位置信息确定单元，用于根据已知顶点的位置信息，确定剩余顶点在目标对象中的位置信息；

生成单元，用于根据所述已知顶点和剩余顶点及其位置信息，生成所述目标对象的三维边界框。

本发明技术方案，一方面，能够根据目标对象在待标定图片中呈现的可视体面获取构成用于标定目标对象的三维边界框的已知顶点，并根据该已知顶点生成三维边界框的剩余顶点，从而能够根据已知顶点和剩余顶点生成适合目标对象尺寸的三维边界框；另一方面，构成三维边界框的顶点具有表示其在目标对象中的位置信息，从而使得生成的三维边界框能够体现目标对象的体面信息，例如目标对象的正面、背面、侧面等，为后续提高目标检测的准确性提供依据。

本发明的其它特征和优点将在随后的说明书中阐述，并且，部分地从说明书中变得显而易见，或者通过实施本发明而了解。本发明的目的和其他优点可通过在所写的说明书、权利要求书、以及附图中所特别指出的结构来实现和获得。

下面通过附图和实施例，对本发明的技术方案做进一步的详细描述。

附图说明

附图用来提供对本发明的进一步理解，并且构成说明书的一部分，与本发明的实施例一起用于解释本发明，并不构成对本发明的限制。