arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2310.00920cs.CV

每个数据集都很重要:通过联合数据集训练扩大单目3D目标检测

Every Dataset Counts: Scaling up Monocular 3D Object Detection with Joint Datasets Training

The Hong Kong University of Science and Technology (Guangzhou)(香港科技大学(广州));The Hong Kong University of Science and Technology(香港科技大学)

Fulong Ma, Xiaoyang Yan, Guoyang Zhao, Xiaojie Xu, Yuxuan Liu, Jun Ma, Ming Liu

AI总结:

本研究提出一种联合数据集训练框架,通过鲁棒单目3D模型、选择性训练策略和伪3D训练方法,利用多种3D/2D数据集提升单目3D目标检测的泛化能力与性能。

摘要:

单目3D目标检测在自动驾驶中起着至关重要的作用。然而,现有的单目3D检测算法依赖于从LiDAR测量中获取的3D标签,这对于新数据集而言获取成本高昂,且在全新环境中部署极具挑战。具体而言,本研究探讨了在多样化的3D和2D数据集集合上训练单目3D目标检测模型的流程。所提出的框架包含三个部分:(1)一个能够在各种相机设置下运行的鲁棒单目3D模型,(2)一种用于适应具有不同类别标注的数据集的选择性训练策略,以及(3)一种利用2D标签的伪3D训练方法,以提升在仅包含2D标签场景中的检测性能。借助该框架,我们能够在由各种开放3D/2D数据集组成的联合集合上训练模型,从而获得具有显著更强泛化能力且在仅有2D标签的新数据集上性能增强的模型。我们在KITTI/nuScenes/ONCE/Cityscapes/BDD100K数据集上进行了广泛实验,以证明所提方法的扩展能力。

英文摘要:

Monocular 3D object detection plays a crucial role in autonomous driving. However, existing monocular 3D detection algorithms depend on 3D labels derived from LiDAR measurements, which are costly to acquire for new datasets and challenging to deploy in novel environments. Specifically, this study investigates the pipeline for training a monocular 3D object detection model on a diverse collection of 3D and 2D datasets. The proposed framework comprises three components: (1) a robust monocular 3D model capable of functioning across various camera settings, (2) a selective-training strategy to accommodate datasets with differing class annotations, and (3) a pseudo 3D training approach using 2D labels to enhance detection performance in scenes containing only 2D labels. With this framework, we could train models on a joint set of various open 3D/2D datasets to obtain models with significantly stronger generalization capability and enhanced performance on new dataset with only 2D labels. We conduct extensive experiments on KITTI/nuScenes/ONCE/Cityscapes/BDD100K datasets to demonstrate the scaling ability of the proposed method.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑