arXiv:2502.00315cs.CV2025-02被引 5

用视觉大模型提升单目3D目标检测精度,实现端到端优化

MonoDINO-DETR: Depth-Enhanced Monocular 3D Object Detection Using a Vision Foundation Model

  • 基于ViT大模型提取全局特征,融合层次化特征块增强表达能力
  • 在KITTI和自建赛道数据集上,3D检测精度超越现有方法
  • 适合需要高精度单目3D感知的自动驾驶与机器人应用

本文提出一种新方法,通过利用视觉基础模型的泛化特征提取能力,提升单目3D目标检测性能。不同于依赖多阶段流程且深度估计不准的传统CNN方法,该研究采用基于视觉变压器(ViT)的基础模型作为主干网络,擅长捕捉全局特征以支持更优的深度估计。结合检测变压器(DETR)架构,实现深度估计与目标检测的一阶段联合优化。引入层次化特征融合模块,从基础模型中提取更丰富的视觉特征,进一步增强特征表达能力。通过在大规模数据上训练相对深度估计模型,并采用迁移学习进行微调,显著提升深度估计精度。此外,解码器中使用考虑参考点及2D边界框尺寸的查询机制,有效提升识别性能。在KITTI 3D基准测试及高海拔赛车环境自建数据集上,通过定量与定性评估验证了该模型优于近期最先进方法。代码已开源:https://github.com/JihyeokKim/MonoDINO-DETR。

原文摘要 · Abstract (English)

This paper proposes novel methods to enhance the performance of monocular 3D object detection models by leveraging the generalized feature extraction capabilities of a vision foundation model. Unlike traditional CNN-based approaches, which often suffer from inaccurate depth estimation and rely on multi-stage object detection pipelines, this study employs a Vision Transformer (ViT)-based foundation model as the backbone, which excels at capturing global features for depth estimation. It integrates a detection transformer (DETR) architecture to improve both depth estimation and object detection performance in a one-stage manner. Specifically, a hierarchical feature fusion block is introduced to extract richer visual features from the foundation model, further enhancing feature extraction capabilities. Depth estimation accuracy is further improved by incorporating a relative depth estimation model trained on large-scale data and fine-tuning it through transfer learning. Additionally, the use of queries in the transformer's decoder, which consider reference points and the dimensions of 2D bounding boxes, enhances recognition performance. The proposed model outperforms recent state-of-the-art methods, as demonstrated through quantitative and qualitative evaluations on the KITTI 3D benchmark and a custom dataset collected from high-elevation racing environments. Code is available at https://github.com/JihyeokKim/MonoDINO-DETR.

3D检测视觉大模型单目感知自动驾驶

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