一种数据自动标注方法、装置、设备及存储介质转让专利

申请号 : CN202210902948.0

文献号 : CN114972490B

文献日 :

基本信息:

PDF:

法律信息:

相似专利:

发明人 : 谢子锐胡兰王一夫张如高虞正华

申请人 : 苏州魔视智能科技有限公司

摘要 :

本申请是关于一种数据自动标注方法、装置、设备及存储介质,具体涉及计算机视觉技术领域。所述方法包括:获取激光雷达采集到的各帧雷达数据;对所述各帧雷达数据进行目标检测,分别获得各帧雷达数据的包围框;针对激光雷达位于雷达位姿,且在目标时刻采集到的目标雷达数据,根据所述目标雷达数据的包围框分别与所述各帧雷达数据的包围框的重合度,筛选出目标包围框;获取目标图像;基于所述相机位姿以及所述雷达位姿,将所述目标包围框投影至相机坐标系下获得图像包围框。上述方案不需要人工参数标注,在极大的降低真值数据的标注时间的前提下,保证了标注的准确性。

权利要求 :

1.一种数据自动标注方法,其特征在于,所述方法包括:

获取激光雷达采集到的按照时间排序的各帧雷达数据;所述各帧雷达数据为所述激光雷达按照时间,对周围环境进行探测得到的;

对所述各帧雷达数据进行目标检测,分别获得各帧雷达数据的包围框;

针对激光雷达位于雷达位姿,且在目标时刻采集到的目标雷达数据,根据所述目标雷达数据的包围框分别与所述各帧雷达数据的包围框的重合度,筛选出目标包围框;

获取目标图像;所述目标图像为处于相机位姿下的目标相机,在目标时刻采集到的;

基于所述相机位姿以及所述雷达位姿,将所述目标包围框投影至相机坐标系下获得图像包围框,以在所述目标图像上通过所述图像包围框标注出目标物体;

所述对所述各帧雷达数据进行目标检测,分别获得各帧雷达数据的包围框,包括:针对每帧雷达数据进行目标检测,获得候选包围框;

将所述候选包围框与各个活跃包围框计算交并比,且当交并比大于第一阈值时,将所述候选包围框确定为活跃包围框,且将所述候选包围框确定为所述雷达数据的包围框。

2.根据权利要求1所述的方法,其特征在于,所述基于所述相机位姿以及所述雷达位姿,将所述目标包围框投影至相机坐标系下获得图像包围框,包括;基于所述雷达位姿,确定各个所述目标包围框的目标距离,并基于所述相机位姿以及所述雷达位姿,将目标距离小于距离阈值的目标包围框投影至相机坐标系下获得所述图像包围框。

3.根据权利要求1所述的方法,其特征在于,所述相机位姿用于指示所述相机坐标系与世界坐标系之间的映射关系;所述雷达位姿用于指示雷达坐标系与所述世界坐标系下的映射关系;

所述基于所述相机位姿以及所述雷达位姿,将所述目标包围框投影至相机坐标系下获得图像包围框,包括:基于所述雷达位姿,将所述目标包围框投影至世界坐标系下,获得实际包围框;

基于所述相机位姿,将所述实际包围框投影至所述相机坐标系下,获得所述图像包围框。

4.根据权利要求3所述的方法,其特征在于,在基于所述相机位姿,将所述实际包围框坐标投影至所述相机坐标系下,获得所述图像包围框之前,还包括:基于所述雷达位姿,将所述雷达数据中的稀疏点云投影至世界坐标系下,判断实际包围框中的点云数量;

所述基于所述相机位姿,将所述实际包围框投影至所述相机坐标系下,获得所述图像包围框,包括:当所述实际包围框中的点云数量大于点云阈值时,将所述实际包围框投影至所述相机坐标系下,获得所述图像包围框。

5.根据权利要求1至4任一所述的方法,其特征在于,所述目标雷达数据的包围框包括第一包围框;

根据所述目标雷达数据的包围框分别与所述各帧雷达数据的包围框的重合度,筛选出目标包围框,包括:当所述各帧雷达数据中,存在第一包围框对应的第一数据的帧数大于帧数阈值,则将所述第一包围框确定为目标包围框。

6.根据权利要求5所述的方法,其特征在于,所述当所述各帧雷达数据中,存在第一包围框对应的第一数据的雷达数据的帧数大于帧数阈值,则将所述第一包围框确定为目标包围框,包括:当所述各帧雷达数据中,存在第一包围框对应的第一数据的候选雷达数据的帧数大于帧数阈值时,针对每个候选雷达数据,获取所述第一数据所在的包围框,与所述各帧雷达数据的包围框的目标交并比;

当各个候选雷达数据的目标交并比之和大于第二阈值,则将所述第一包围框确定为目标包围框。

7.一种数据自动标注装置,其特征在于,所述装置包括:

雷达数据获取模块,用于获取激光雷达采集到的按照时间排序的各帧雷达数据;所述各帧雷达数据为所述激光雷达按照时间,对周围环境进行探测得到的;

目标检测模块,用于对所述各帧雷达数据进行目标检测,分别获得各帧雷达数据的包围框;

包围框筛选模块,用于针对激光雷达位于雷达位姿,且在目标时刻采集到的目标雷达数据,根据所述目标雷达数据的包围框分别与所述各帧雷达数据的包围框的重合度,筛选出目标包围框;

目标图像获取模块,用于获取目标图像;所述目标图像为处于相机位姿下的目标相机,在目标时刻采集到的;

包围框投影模块,用于基于所述相机位姿以及所述雷达位姿,将所述目标包围框投影至相机坐标系下获得图像包围框,以在所述目标图像上通过所述图像包围框标注出目标物体;

所述目标检测模块,还用于:

针对每帧雷达数据进行目标检测,获得候选包围框;

将所述候选包围框与各个活跃包围框计算交并比,且当交并比大于第一阈值时,将所述候选包围框确定为活跃包围框,且将所述候选包围框确定为所述雷达数据的包围框。

8.一种计算机设备,其特征在于,所述计算机设备包括处理器和存储器,所述存储器中存储有至少一条指令,所述至少一条指令由所述处理器加载并执行以实现如权利要求1至6任一所述的数据自动标注方法。

9.一种计算机可读存储介质,其特征在于,所述存储介质中存储有至少一条指令,所述至少一条指令由处理器加载并执行以实现如权利要求1至6任一所述的数据自动标注方法。

说明书 :

一种数据自动标注方法、装置、设备及存储介质

技术领域

[0001] 本发明涉及计算机视觉技术领域,具体涉及一种数据自动标注方法、装置、设备及存储介质。

背景技术

[0002] 在智能驾驶场景中,前端感知是主要功能模块之一,用于收集环境中的信息并输出给后续定位、决策等功能。
[0003] 现如今大部分自动驾驶的感知模块都会使用基于深度学习的检测、分割等方法,这些基于深度学习的方法的通常需要大量且精确的真值数据与训练数据。自动驾驶技术的应用场景中,单目深度估计,即从二维图像恢复三维信息是一项重要的技术。因此,在作为真值数据或训练数据时,需要从单张二维图像中标注出三维信息,通常以三维包围框(bounding box)的形式表示。现有的二维图像数据采集过程严重依赖于人工数据标注,真实数据采集后需要在二维图像对应的三维数据中,人工标注出感兴趣的数据,再根据三维传感器与二维传感器的外参将三维信息投影至二维图像上。
[0004] 但通过上述方案,人工与时间成本高,且标注精度取决于标注人员的技术,因此准确率不确定。

发明内容

[0005] 本申请提供了一种数据自动标注方法、装置、计算机设备及存储介质,在极大的降低真值数据的标注时间的前提下,保证了标注的准确性,该技术方案如下。
[0006] 一方面,提供了一种数据自动标注方法,所述方法包括:
[0007] 获取激光雷达采集到的各帧雷达数据;
[0008] 对所述各帧雷达数据进行目标检测,分别获得各帧雷达数据的包围框;
[0009] 针对激光雷达位于雷达位姿,且在目标时刻采集到的目标雷达数据,根据所述目标雷达数据的包围框分别与所述各帧雷达数据的包围框的重合度,筛选出目标包围框;
[0010] 获取目标图像;所述目标图像为处于相机位姿下的目标相机,在目标时刻采集到的;
[0011] 基于所述相机位姿以及所述雷达位姿,将所述目标包围框投影至相机坐标系下获得图像包围框,以在所述目标图像上通过所述图像包围框标注出目标物体。
[0012] 又一方面,提供了一种数据自动标注装置,所述装置包括:
[0013] 雷达数据获取模块,用于获取激光雷达采集到的各帧雷达数据;
[0014] 目标检测模块,用于对所述各帧雷达数据进行目标检测,分别获得各帧雷达数据的包围框;
[0015] 包围框筛选模块,用于针对激光雷达位于雷达位姿,且在目标时刻采集到的目标雷达数据,根据所述目标雷达数据的包围框分别与所述各帧雷达数据的包围框的重合度,筛选出目标包围框;
[0016] 目标图像获取模块,用于获取目标图像;所述目标图像为处于相机位姿下的目标相机,在目标时刻采集到的;
[0017] 包围框投影模块,用于基于所述相机位姿以及所述雷达位姿,将所述目标包围框投影至相机坐标系下获得图像包围框,以在所述目标图像上通过所述图像包围框标注出目标物体。
[0018] 在一种可能的实现方式中,所述包围框筛选模块,还用于,
[0019] 基于所述雷达位姿,确定各个所述目标包围框的目标距离,并基于所述相机位姿以及所述雷达位姿,将目标距离小于距离阈值的目标包围框投影至相机坐标系下获得所述图像包围框。
[0020] 在一种可能的实现方式中,所述相机位姿用于指示所述相机坐标系与世界坐标系之间的映射关系;所述雷达位姿用于指示雷达坐标系与所述世界坐标系下的映射关系;
[0021] 所述包围框投影模块,还用于,
[0022] 基于所述雷达位姿,将所述目标包围框投影至世界坐标系下,获得实际包围框;
[0023] 基于所述相机位姿,将所述实际包围框投影至所述相机坐标系下,获得所述图像包围框。
[0024] 在一种可能的实现方式中,所述装置还包括:
[0025] 点云数量判断模块,用于基于所述雷达位姿,将所述雷达数据中的稀疏点云投影至世界坐标系下,判断实际包围框中的点云数量;
[0026] 在一种可能的实现方式中,所述包围框投影模块,还用于当所述实际包围框中的点云数量大于点云阈值时,将所述实际包围框投影至所述相机坐标系下,获得所述图像包围框。
[0027] 在一种可能的实现方式中,所述目标检测模块,还用于,
[0028] 针对每帧雷达数据进行目标检测,获得候选包围框;
[0029] 将所述候选包围框与各个活跃包围框计算交并比,且当交并比大于第一阈值时,将所述候选包围框确定为活跃包围框,且将所述候选包围框确定为所述雷达数据的包围框。
[0030] 在一种可能的实现方式中,所述目标雷达数据的包围框包括第一包围框;
[0031] 所述包围框筛选模块,还用于,
[0032] 当所述各帧雷达数据中,存在第一包围框对应的第一数据的帧数大于帧数阈值,则将所述第一包围框确定为目标包围框。
[0033] 在一种可能的实现方式中,所述包围框筛选模块,还用于,
[0034] 当所述各帧雷达数据中,存在第一包围框对应的第一数据的候选雷达数据的帧数大于帧数阈值时,针对每个候选雷达数据,获取所述第一数据所在的包围框,与所述各帧雷达数据的包围框的目标交并比;
[0035] 当各个候选雷达数据的目标交并比之和大于第二阈值,则将所述第一包围框确定为目标包围框。
[0036] 再一方面,提供了一种计算机设备,所述计算机设备包括处理器和存储器,所述存储器中存储有至少一条指令,所述至少一条指令由所述处理器加载并执行以实现上述的数据自动标注方法。
[0037] 又一方面,提供了一种计算机可读存储介质,所述存储介质中存储有至少一条指令,所述至少一条指令由处理器加载并执行以实现上述的数据自动标注方法。
[0038] 再一方面,提供了一种计算机程序产品或计算机程序,所述计算机程序产品或计算机程序包括计算机指令,所述计算机指令存储在计算机可读存储介质中。计算机设备的处理器从计算机可读存储介质中读取所述计算机指令,处理器执行所述计算机指令,使得所述计算机设备执行上述数据自动标注方法。
[0039] 本申请提供的技术方案可以包括以下有益效果:
[0040] 在对图像中进行标注时,可以先通过激光雷达对周围环境进行采集,得到各帧雷达数据,计算机设备根据激光雷达采集到的雷达数据进行目标检测,从而得到各帧雷达数据中的包围框,此时计算机设备再根据各帧雷达数据之间的包围框的重合度,筛选掉异常的包围框,保留下目标包围框;此时当需要对目标时刻采集到的目标图像进行标注时,可以直接根据目标时刻的雷达数据中的包围框,通过雷达位姿以及相机姿态进行坐标系转换,从而将雷达数据中的包围框,转换至相机坐标系,也就是在目标图像上生成对应的图像包围框,从而实现数据的自动标注。上述方案不需要人工参数标注,在极大的降低真值数据的标注时间的前提下,保证了标注的准确性。

附图说明

[0041] 为了更清楚地说明本申请具体实施方式或现有技术中的技术方案,下面将对具体实施方式或现有技术描述中所需要使用的附图作简单地介绍,显而易见地,下面描述中的附图是本申请的一些实施方式,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据这些附图获得其他的附图。
[0042] 图1是根据一示例性实施例示出的一种数据自动标注系统的结构示意图。
[0043] 图2是根据一示例性实施例示出的一种数据自动标注方法的流程图。
[0044] 图3是根据一示例性实施例示出的一种数据自动标注方法的流程图。
[0045] 图4是根据一示例性实施例示出的一种数据自动标注装置的结构方框图。
[0046] 图5示出了本申请一示例性实施例示出的计算机设备的结构框图。

具体实施方式

[0047] 下面将结合附图对本申请的技术方案进行清楚、完整地描述,显然,所描述的实施例是本申请一部分实施例,而不是全部的实施例。基于本申请中的实施例,本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例,都属于本申请保护的范围。
[0048] 应理解,在本申请的实施例中提到的“指示”可以是直接指示,也可以是间接指示,还可以是表示具有关联关系。举例说明,A指示B,可以表示A直接指示B,例如B可以通过A获取;也可以表示A间接指示B,例如A指示C,B可以通过C获取;还可以表示A和B之间具有关联关系。
[0049] 在本申请实施例的描述中,术语“对应”可表示两者之间具有直接对应或间接对应的关系,也可以表示两者之间具有关联关系,也可以是指示与被指示、配置与被配置等关系。
[0050] 本申请实施例中,“预定义”可以通过在设备(例如,包括终端设备和网络设备)中预先保存相应的代码、表格或其他可用于指示相关信息的方式来实现,本申请对于其具体的实现方式不做限定。
[0051] 图1是根据一示例性实施例示出的一种数据自动标注系统的结构示意图。该数据自动标注统中包含服务器110以及目标车辆120。其中,该目标车辆120中可以包括数据处理设备、图像采集设备以及数据存储模块等模块。
[0052] 可选的,该目标车辆120中还包括激光雷达,该激光雷达可以在目标车辆的运行过程中,对目标车辆的周围的环境进行数据采集,以获得雷达数据。
[0053] 可选的,该目标车辆120中包括图像采集设备以及数据存储模块,该图像采集设备可以在目标车辆的运行过程中,对目标车辆周围的环境进行图像采集,并将采集到的图像保存在目标车辆中的数据存储模块中。
[0054] 可选的,该目标车辆120通过传输网络(如无线通信网络)与服务器110实现通信连接,该目标车辆120可以通过无线通信网络,将数据存储模块中存储的各个数据(如采集到的图像以及雷达数据)上传至服务器110,以便服务器110通过采集到的雷达数据,对采集到的图像进行数据标注。
[0055] 可选的,该目标车辆120中还包含有数据处理设备,该数据处理设备可以在目标车辆120的图像采集设备采集到图像时,通过激光雷达采集到的雷达数据,对图像进行数据标注。
[0056] 可选的,上述服务器可以是由多个物理服务器构成的服务器集群或者是分布式系统,还可以是提供云服务、云数据库、云计算、云函数、云存储、网络服务、云通信、中间件服务、域名服务、安全服务、CDN、以及大数据和人工智能平台等技术运计算服务的云服务器。
[0057] 可选的,该系统还可以包括管理设备,该管理设备用于对该系统进行管理(如管理各个模块与服务器之间的连接状态等),该管理设备与服务器之间通过通信网络相连。可选的,该通信网络是有线网络或无线网络。
[0058] 可选的,上述的无线网络或有线网络使用标准通信技术和/或协议。网络通常为因特网,但也可以是其他任何网络,包括但不限于局域网、城域网、广域网、移动、有限或无线网络、专用网络或者虚拟专用网络的任何组合。在一些实施例中,使用包括超文本标记语言、可扩展标记语言等的技术和/或格式来代表通过网络交换的数据。此外还可以使用诸如安全套接字层、传输层安全、虚拟专用网络、网际协议安全等常规加密技术来加密所有或者一些链路。在另一些实施例中,还可以使用定制和/或专用数据通信技术取代或者补充上述数据通信技术。
[0059] 图2是根据一示例性实施例示出的一种数据自动标注方法的流程图。该方法由计算机设备执行,该计算机设备可以是如图1中所示的目标车辆中的数据处理设备。如图2所示,该数据自动标注方法可以包括如下步骤:
[0060] 步骤201,获取激光雷达采集到的各帧雷达数据。
[0061] 在本申请实施例的一种可能的实现方式中,该激光雷达可以是目标车辆上的激光雷达设备,该激光雷达可以按照时间,对周围环境进行探测,从而得到按照时间排序的各帧雷达数据。
[0062] 进一步的,在获取到激光雷达采集到的各帧雷达数据后,该目标车辆中的数据处理设备,可以基于激光雷达采集到的各帧雷达数据,通过SLAM算法,实现全局稀疏建图(即构建目标车辆周围的地图信息),并基于周围的地图信息实现目标车辆的定位。也就是说,在通过激光雷达获取各帧雷达数据的同时,目标车辆还可以基于雷达数据实现周围地图的构建以及目标车辆的定位。
[0063] 步骤202,对该各帧雷达数据进行目标检测,分别获得各帧雷达数据的包围框。
[0064] 计算机设备在获取到雷达数据后,则可以对各帧雷达数据分别进行目标检测,从而在各帧雷达数据中确定出需要检测的目标物体。
[0065] 以人体图像检测为例,当计算机设备需要进行人体图像检测时,可以选取任一帧雷达数据,对雷达数据通过人体检测算法(例如通过机器学习模型实现)进行处理,从而获得雷达数据中的特征点,以及该特征点的检测框(即包围框)。
[0066] 步骤203,针对激光雷达位于雷达位姿,且在目标时刻采集到的目标雷达数据,根据该目标雷达数据的包围框分别与该各帧雷达数据的包围框的重合度,筛选出目标包围框。
[0067] 在本申请实施例的一种可能的实现方式中,该激光雷达采集到各帧雷达数据可能是在不同时刻,位于不同雷达位姿时所采集到的,因此激光雷达在保存每一帧雷达数据时,还可以同时保存采集到该帧雷达数据时激光雷达的雷达位姿,以及采集到该帧雷达数据的时间。
[0068] 而对于其中任一帧雷达数据(即目标雷达数据)来说,为了尽可能避免误识别对后续图像标注的影响,此时需要尽量去除误检数据,因此计算机设备可以将目标雷达数据中的包围框,与各帧雷达数据的包围框进行对比,计算其中的重合度,当重合度小于阈值时,说明该包围框中的数据出现的次数较小,可能是无关数据,将其筛除,从而保留重合度较高的包围框。
[0069] 步骤204,获取目标图像。
[0070] 该目标图像为处于相机位姿下的目标相机,在目标时刻采集到的。
[0071] 步骤205,基于该相机位姿以及该雷达位姿,将该目标包围框投影至相机坐标系下获得图像包围框,以在该目标图像上通过该图像包围框标注出目标物体。
[0072] 当计算机设备需要对目标图像进行标注时,可以选取到目标时刻采集到的目标雷达数据,并将在该目标雷达数据中通过目标检测处理,并筛选得到的目标包围框,通过相机位姿(也就是相机坐标系到世界坐标系的旋转矩阵以及平移矩阵)以及雷达位姿(也就是雷达坐标系到世界坐标系的旋转矩阵以及平移矩阵),将目标包围框从雷达坐标系转换至相机坐标系,从而在目标图像中生成图像包围框,从而在目标图像中标注出目标物体。
[0073] 综上所述,在对图像中进行标注时,可以先通过激光雷达对周围环境进行采集,得到各帧雷达数据,计算机设备根据激光雷达采集到的雷达数据进行目标检测,从而得到各帧雷达数据中的包围框,此时计算机设备再根据各帧雷达数据之间的包围框的重合度,筛选掉异常的包围框,保留下目标包围框;此时当需要对目标时刻采集到的目标图像进行标注时,可以直接根据目标时刻的雷达数据中的包围框,通过雷达位姿以及相机姿态进行坐标系转换,从而将雷达数据中的包围框,转换至相机坐标系,也就是在目标图像上生成对应的图像包围框,从而实现数据的自动标注。上述方案不需要人工参数标注,在极大的降低真值数据的标注时间的前提下,保证了标注的准确性。
[0074] 图3是根据一示例性实施例示出的一种数据自动标注方法的流程图。该方法由计算机设备执行,该计算机设备可以是如图1中所示的目标车辆中的数据处理设备。如图3所示,该数据自动标注方法可以包括如下步骤:
[0075] 步骤301,获取激光雷达采集到的各帧雷达数据。
[0076] 在本申请实施例中,利用激光雷达数据,并通过现有的激光雷达SLAM对计算出激光雷达在数据采集过程中的一系列位姿。
[0077] 由于激光雷达SLAM获取的位姿与图像在时间上可能存在不一致,计算机设备需对位姿估计结果按图像时间戳进行插值获取到图像对应时刻的位姿。
[0078] 步骤302,针对每帧雷达数据进行目标检测,获得候选包围框。
[0079] 计算机设备可以对每一帧激光雷达检测进行目标检测,获取到其位姿及包围框尺寸,并将目标位姿通过对应时刻的雷达位姿转换至世界坐标系。
[0080] 步骤303,将该候选包围框与各个活跃包围框计算交并比,且当交并比大于第一阈值时,将该候选包围框确定为活跃包围框,且将该候选包围框确定为该雷达数据的包围框。
[0081] 可选的,计算机设备可以通过活跃跟踪器,筛选出各帧雷达数据中的活跃包围框。
[0082] 首先,计算机设备将跟踪器初始化,将第一帧检测结果D1全部加入活跃跟踪器;计算机设备再遍历当前帧雷达数据的检测结果,并逐一与活跃跟踪器中的所有检测目标计算交并比,若交并比大于第一阈值 ,则将跟踪目标 的检测信息加入相对应的活跃检测器 ,记录其位姿、包围框信息、交并比及其对应时间,此时活跃检测器中的包围框即为活跃包围框。
[0083] 步骤304,针对激光雷达位于雷达位姿,且在目标时刻采集到的目标雷达数据,根据该目标雷达数据的包围框分别与该各帧雷达数据的包围框的重合度,筛选出目标包围框。
[0084] 在一种可能的实现方式中,该目标雷达数据的包围框包括第一包围框,根据该目标雷达数据的包围框分别与该各帧雷达数据的包围框的重合度,筛选出目标包围框,包括:
[0085] 当该各帧雷达数据中,存在第一包围框对应的第一数据的帧数大于帧数阈值,则将该第一包围框确定为目标包围框。
[0086] 也就是说,在目标雷达数据中检测出第一包围框(也就是存在第一包围框中的第一物体)后,可以检测第一包围框在各帧雷达数据中是否存在,当有大于帧数阈值的雷达数据中都存在第一包围框所对应的数据(也就是存在第一包围框中的第一物体的数据)时,则可以判定第一包围框中的第一物体为较重要的物体,直接将该第一包围框确定为目标包围框。
[0087] 在一种可能的实现方式中,当该各帧雷达数据中,存在第一包围框对应的第一数据的候选雷达数据的帧数大于帧数阈值时,针对每个候选雷达数据,获取该第一数据所在的包围框,与该各帧雷达数据的包围框的目标交并比;
[0088] 当各个候选雷达数据的目标交并比之和大于第二阈值,则将该第一包围框确定为目标包围框。
[0089] 而当各帧雷达数据中的任一帧雷达数据,存在第一包围框对应的第一数据时,则将该雷达数据确定为候选雷达数据,此时对于每个候选雷达数据,计算机设备都将候选雷达数据中的第一数据所在的包围框,与所有帧雷达数据的各个包围框计算交并比,当各个候选雷达数据中,第一数据所在的包围框,与所有帧雷达数据的各个包围框的目标交并比之和大于第二阈值时,则第一包围框被确定为较为重要的目标包围框。
[0090] 可选的,计算机设备可以检查活跃跟踪器 ,若某一活跃检测器 对应的跟踪目标丢失帧数大于阈值,检查其跟踪帧数是否大于阈值以及历史交并比是否大于第二阈值 ,若满足则将活跃检测器中记录的信息加入不活跃跟踪器 (也就是作为目标包围框),若不满足则直接删除。
[0091] 计算机设备对于每一帧雷达数据都可以重复上述步骤303与步骤304所示出的操作,最后得到的不活跃跟踪器中的目标包围框即为跟踪结果,从而在该跟踪结果中筛除异常物体的包围框,提高了目标检测的准确性。
[0092] 进一步的,计算机设备还可以对上述得到的各个目标包围框进行进一步的数据检测。
[0093] 可选的,计算机设备对于每一跟踪目标(例如第一数据所指示的物体),可以遍历该第一数据在各帧雷达数据中的包围框,当该第一数据在各帧雷达数据中的包围框的平均交并比大于阈值,且第一数据的位姿的平方差小于阈值,将该第一数据对应的目标确认为静态目标,否则将其确认为动态目标,从而进一步增加的数据标注的标注类别。
[0094] 可选的,计算机设备在进行目标检测时,还可以对第一数据进行滑动窗口的中值滤波优化,以降低误检概率。
[0095] 可选的,计算机设备对于每一跟踪目标(例如第一数据所指示的物体),可以获取该第一数据在各帧雷达数据中的包围框尺寸,并取各帧雷达数据中的第一数据的包围框尺寸的中值或平均值更新为各帧雷达数据中的第一数据的包围框尺寸。
[0096] 步骤305,获取目标图像。
[0097] 该目标图像为处于相机位姿下的目标相机,在目标时刻采集到的。
[0098] 步骤306,基于该相机位姿以及该雷达位姿,将该目标包围框投影至相机坐标系下获得图像包围框,以在该目标图像上通过该图像包围框标注出目标物体。
[0099] 在一种可能的实现方式中,基于该雷达位姿,确定各个该目标包围框的目标距离,并基于该相机位姿以及该雷达位姿,将目标距离小于距离阈值的目标包围框投影至相机坐标系下获得该图像包围框。
[0100] 即计算机设备可以依照图像时间戳获得当前时刻世界坐标系下的雷达位姿,并在跟踪结果 中查找距离阈值范围内的目标并记录。
[0101] 在一种可能的实现方式中,该基于该相机位姿以及该雷达位姿,将该目标包围框投影至相机坐标系下获得图像包围框,包括:
[0102] 基于该雷达位姿,将该目标包围框投影至世界坐标系下,获得实际包围框;
[0103] 基于该相机位姿,将该实际包围框投影至该相机坐标系下,获得该图像包围框。
[0104] 可选的,计算机设备还可以根据雷达位姿以及相机位姿直接计算出雷达到相机的外参 ,以将目标包围框投影相机坐标系下,至此标定完成。
[0105] 在一种可能的实现方式中,当该实际包围框中的点云数量大于点云阈值时,将该实际包围框投影至该相机坐标系下,获得该图像包围框。
[0106] 即计算机设备还可以对目标图像中的被遮挡图像进行筛选,计算机设备可以将当前帧雷达检测的稀疏点云(也就是目标雷达数据)投影至世界坐标系,统计落在目标包围框中的点云数量,若点云数量低于阈值,则认为此目标在图像中被遮挡,删除此目标。
[0107] 综上所述,在对图像中进行标注时,可以先通过激光雷达对周围环境进行采集,得到各帧雷达数据,计算机设备根据激光雷达采集到的雷达数据进行目标检测,从而得到各帧雷达数据中的包围框,此时计算机设备再根据各帧雷达数据之间的包围框的重合度,筛选掉异常的包围框,保留下目标包围框;此时当需要对目标时刻采集到的目标图像进行标注时,可以直接根据目标时刻的雷达数据中的包围框,通过雷达位姿以及相机姿态进行坐标系转换,从而将雷达数据中的包围框,转换至相机坐标系,也就是在目标图像上生成对应的图像包围框,从而实现数据的自动标注。上述方案不需要人工参数标注,在极大的降低真值数据的标注时间的前提下,保证了标注的准确性。
[0108] 图4是根据一示例性实施例示出的一种数据自动标注装置的结构方框图。该数据自动标注装置包括:
[0109] 雷达数据获取模块401,用于获取激光雷达采集到的各帧雷达数据;
[0110] 目标检测模块402,用于对所述各帧雷达数据进行目标检测,分别获得各帧雷达数据的包围框;
[0111] 包围框筛选模块403,用于针对激光雷达位于雷达位姿,且在目标时刻采集到的目标雷达数据,根据所述目标雷达数据的包围框分别与所述各帧雷达数据的包围框的重合度,筛选出目标包围框;
[0112] 目标图像获取模块404,用于获取目标图像;所述目标图像为处于相机位姿下的目标相机,在目标时刻采集到的;
[0113] 包围框投影模块405,用于基于所述相机位姿以及所述雷达位姿,将所述目标包围框投影至相机坐标系下获得图像包围框,以在所述目标图像上通过所述图像包围框标注出目标物体。
[0114] 在一种可能的实现方式中,所述包围框筛选模块,还用于,
[0115] 基于所述雷达位姿,确定各个所述目标包围框的目标距离,并基于所述相机位姿以及所述雷达位姿,将目标距离小于距离阈值的目标包围框投影至相机坐标系下获得所述图像包围框。
[0116] 在一种可能的实现方式中,所述相机位姿用于指示所述相机坐标系与世界坐标系之间的映射关系;所述雷达位姿用于指示雷达坐标系与所述世界坐标系下的映射关系;
[0117] 所述包围框投影模块,还用于,
[0118] 基于所述雷达位姿,将所述目标包围框投影至世界坐标系下,获得实际包围框;
[0119] 基于所述相机位姿,将所述实际包围框投影至所述相机坐标系下,获得所述图像包围框。
[0120] 在一种可能的实现方式中,所述装置还包括:
[0121] 点云数量判断模块,用于基于所述雷达位姿,将所述雷达数据中的稀疏点云投影至世界坐标系下,判断实际包围框中的点云数量;
[0122] 在一种可能的实现方式中,所述包围框投影模块,还用于当所述实际包围框中的点云数量大于点云阈值时,将所述实际包围框投影至所述相机坐标系下,获得所述图像包围框。
[0123] 在一种可能的实现方式中,所述目标检测模块,还用于,
[0124] 针对每帧雷达数据进行目标检测,获得候选包围框;
[0125] 将所述候选包围框与各个活跃包围框计算交并比,且当交并比大于第一阈值时,将所述候选包围框确定为活跃包围框,且将所述候选包围框确定为所述雷达数据的包围框。
[0126] 在一种可能的实现方式中,所述目标雷达数据的包围框包括第一包围框;
[0127] 所述包围框筛选模块,还用于,
[0128] 当所述各帧雷达数据中,存在第一包围框对应的第一数据的帧数大于帧数阈值,则将所述第一包围框确定为目标包围框。
[0129] 在一种可能的实现方式中,所述包围框筛选模块,还用于,
[0130] 当所述各帧雷达数据中,存在第一包围框对应的第一数据的候选雷达数据的帧数大于帧数阈值时,针对每个候选雷达数据,获取所述第一数据所在的包围框,与所述各帧雷达数据的包围框的目标交并比;
[0131] 当各个候选雷达数据的目标交并比之和大于第二阈值,则将所述第一包围框确定为目标包围框。
[0132] 综上所述,在对图像中进行标注时,可以先通过激光雷达对周围环境进行采集,得到各帧雷达数据,计算机设备根据激光雷达采集到的雷达数据进行目标检测,从而得到各帧雷达数据中的包围框,此时计算机设备再根据各帧雷达数据之间的包围框的重合度,筛选掉异常的包围框,保留下目标包围框;此时当需要对目标时刻采集到的目标图像进行标注时,可以直接根据目标时刻的雷达数据中的包围框,通过雷达位姿以及相机姿态进行坐标系转换,从而将雷达数据中的包围框,转换至相机坐标系,也就是在目标图像上生成对应的图像包围框,从而实现数据的自动标注。上述方案不需要人工参数标注,在极大的降低真值数据的标注时间的前提下,保证了标注的准确性。
[0133] 图5示出了本申请一示例性实施例示出的计算机设备500的结构框图。该计算机设备可以实现为本申请上述方案中的服务器。所述计算机设备500包括中央处理单元(Central Processing Unit,CPU)501、包括随机存取存储器(Random Access Memory,RAM)502和只读存储器(Read‑Only Memory,ROM)503的系统存储器504,以及连接系统存储器504和中央处理单元501的系统总线505。所述计算机设备500还包括用于存储操作系统509、应用程序510和其他程序模块511的大容量存储设备506。
[0134] 所述大容量存储设备506通过连接到系统总线505的大容量存储控制器(未示出)连接到中央处理单元501。所述大容量存储设备506及其相关联的计算机可读介质为计算机设备500提供非易失性存储。也就是说,所述大容量存储设备506可以包括诸如硬盘或者只读光盘(Compact Disc Read‑Only Memory,CD‑ROM)驱动器之类的计算机可读介质(未示出)。
[0135] 不失一般性,所述计算机可读介质可以包括计算机存储介质和通信介质。计算机存储介质包括以用于存储诸如计算机可读指令、数据结构、程序模块或其他数据等信息的任何方法或技术实现的易失性和非易失性、可移动和不可移动介质。计算机存储介质包括RAM、ROM、可擦除可编程只读寄存器(Erasable Programmable Read Only Memory,EPROM)、电子抹除式可复写只读存储器(Electrically‑Erasable Programmable Read‑Only Memory,EEPROM)闪存或其他固态存储其技术,CD‑ROM、数字多功能光盘(Digital Versatile Disc,DVD)或其他光学存储、磁带盒、磁带、磁盘存储或其他磁性存储设备。当然,本领域技术人员可知所述计算机存储介质不局限于上述几种。上述的系统存储器504和大容量存储设备506可以统称为存储器。
[0136] 根据本公开的各种实施例,所述计算机设备500还可以通过诸如因特网等网络连接到网络上的远程计算机运行。也即计算机设备500可以通过连接在所述系统总线505上的网络接口单元507连接到网络508,或者说,也可以使用网络接口单元507来连接到其他类型的网络或远程计算机系统(未示出)。
[0137] 所述存储器还包括至少一条计算机程序,所述至少一条计算机程序存储于存储器中,中央处理单元501通过执行该至少一条计算机程序来实现上述各个实施例所示的方法中的全部或部分步骤。
[0138] 在一示例性实施例中,还提供了一种计算机可读存储介质,用于存储有至少一条计算机程序,所述至少一条计算机程序由处理器加载并执行以实现上述方法中的全部或部分步骤。例如,该计算机可读存储介质可以是只读存储器(Read‑Only Memory,ROM)、随机存取存储器(Random Access Memory,RAM)、只读光盘(Compact Disc Read‑Only Memory,CD‑ROM)、磁带、软盘和光数据存储设备等。
[0139] 在一示例性实施例中,还提供了一种计算机程序产品或计算机程序,该计算机程序产品或计算机程序包括计算机指令,该计算机指令存储在计算机可读存储介质中。计算机设备的处理器从计算机可读存储介质读取该计算机指令,处理器执行该计算机指令,使得该计算机设备执行上述图2或图3任一实施例所示方法的全部或部分步骤。
[0140] 本领域技术人员在考虑说明书及实践这里公开的发明后,将容易想到本申请的其它实施方案。本申请旨在涵盖本申请的任何变型、用途或者适应性变化,这些变型、用途或者适应性变化遵循本申请的一般性原理并包括本申请未公开的本技术领域中的公知常识或惯用技术手段。说明书和实施例仅被视为示例性的,本申请的真正范围和精神由下面的权利要求指出。
[0141] 应当理解的是,本申请并不局限于上面已经描述并在附图中示出的精确结构,并且可以在不脱离其范围进行各种修改和改变。本申请的范围仅由所附的权利要求来限制。