arXiv:2412.12460cs.CV2024-12AAAI被引 9

用激光点做提示,让摄像头3D检测更准更快

PromptDet: A Lightweight 3D Object Detection Framework with LiDAR Prompts

  • 用激光点作轻量提示,仅加2%参数提升精度
  • 多模态下mAP提升22.8%,NDS提升21.1%
  • 可无缝切换为纯摄像头推理,几乎不增加耗时

多摄像头3D目标检测因成本优势受到关注,但受相机测距能力弱影响,深度估计不准。为此提出PromptDet,基于提示学习思想,结合通用摄像头检测模块(如BEVDet、BEVDepth)与激光雷达辅助提示器。该提示器利用激光点提供互补信号,仅引入少量可训练参数。框架灵活:既可作为轻量级多模态融合方法,也可在无激光点时转为纯摄像头检测,推理阶段几乎无额外开销。在nuScenes数据集上的实验表明,作为多模态检测器时,相比纯摄像头基线,mAP最高提升22.8%,NDS提升21.1%,参数增量低于2%;在无激光点情况下,仍能实现最大2.4% mAP和4.0% NDS的提升,且对摄像头推理时间影响可忽略。

原文摘要 · Abstract (English)

Multi-camera 3D object detection aims to detect and localize objects in 3D space using multiple cameras, which has attracted more attention due to its cost-effectiveness trade-off. However, these methods often struggle with the lack of accurate depth estimation caused by the natural weakness of the camera in ranging. Recently, multi-modal fusion and knowledge distillation methods for 3D object detection have been proposed to solve this problem, which are time-consuming during the training phase and not friendly to memory cost. In light of this, we propose PromptDet, a lightweight yet effective 3D object detection framework motivated by the success of prompt learning in 2D foundation model. Our proposed framework, PromptDet, comprises two integral components: a general camera-based detection module, exemplified by models like BEVDet and BEVDepth, and a LiDAR-assisted prompter. The LiDAR-assisted prompter leverages the LiDAR points as a complementary signal, enriched with a minimal set of additional trainable parameters. Notably, our framework is flexible due to our prompt-like design, which can not only be used as a lightweight multi-modal fusion method but also as a camera-only method for 3D object detection during the inference phase. Extensive experiments on nuScenes validate the effectiveness of the proposed PromptDet. As a multi-modal detector, PromptDet improves the mAP and NDS by at most 22.8\% and 21.1\% with fewer than 2\% extra parameters compared with the camera-only baseline. Without LiDAR points, PromptDet still achieves an improvement of at most 2.4\% mAP and 4.0\% NDS with almost no impact on camera detection inference time.

3D检测多模态轻量化提示学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。