VGGT-Metric: 一个基于VGGT的度量深度估计模型
首发时间:2026-03-04
摘要:纯视觉深度估计在自动驾驶和三维感知等领域具有重要应用。近期提出的视觉几何基础Transformer(Visual Geometry Grounded Transformer, VGGT)在场景重建与相对深度估计任务上表现优异,但由于其缺乏对绝对物理尺度的感知能力,难以直接输出用于自动驾驶规划与控制的度量深度。针对这一“尺度模糊”难题,本文提出了一种端到端的度量深度估计模型 VGGT-Metric。该模型在原有 VGGT 架构的基础上,创新性地引入了可学习的尺度令牌。具体而言,本文充分利用自动驾驶场景中已知的相机标定先验,将相机内外参经多层感知机映射为特征嵌入,并将其注入至尺度令牌中。随后,该令牌与由 DINOv2 提取的图像特征共同送入交替的帧内与全局 Transformer 编码器中,通过自注意力机制实现全局尺度信息与局部视觉特征的充分交互。在解码阶段,本文设计了独立的尺度预测分支专门对尺度令牌进行解码以输出全局尺度因子,并将其与相对深度图相乘,从而实现完全免后处理的度量深度恢复。在 nuScenes 大规模自动驾驶数据集上的实验结果表明,相较于基线模型,本文方法在绝对相对误差等核心指标上取得了显著的性能提升,有效缓解了纯视觉深度估计中的尺度漂移问题。
关键词: 度量深度估计 视觉几何基础Transformer 尺度恢复 尺度建模 自动驾驶
For information in English, please click here
VGGT-Metric: A Metric Depth Estimation Model Based on VGGT
Abstract:Pure vision depth estimation plays a vital role in fields such as autonomous driving and 3D perception. The recently proposed Visual Geometry Grounded Transformer (VGGT) performs excellently in scene reconstruction and relative depth estimation tasks. However, due to its lack of absolute physical scale awareness, it struggles to directly output the metric depth required for autonomous driving planning and control. To address this scale ambiguity challenge, this paper proposes an end-to-end metric depth estimation model named VGGT-Metric. Building upon the original VGGT architecture, this model innovatively introduces a learnable scale token. Specifically, by fully leveraging the known camera calibration priors in autonomous driving scenarios, the camera intrinsic and extrinsic parameters are mapped into feature embeddings via a Multi-Layer Perceptron (MLP) and injected into the scale token. Subsequently, this token, along with image features extracted by DINOv2, is fed into alternating frame and global Transformer encoders, achieving thorough interaction between global scale information and local visual features through self-attention mechanisms. In the decoding phase, an independent scale prediction branch is designed to specifically decode the scale token and output a global scale factor, which is then multiplied by the relative depth map to achieve entirely post-processing-free metric depth recovery. Experimental results on the large-scale nuScenes autonomous driving dataset demonstrate that, compared to baseline models, our method achieves significant performance improvements on key metrics such as absolute relative error (Abs Rel), effectively mitigating the scale drift problem in pure vision depth estimation.
Keywords: metric depth estimation Visual Geometry Grounded Transformer scale recovery scale modeling autonomous driving
基金:
引用

No.****
同行评议
勘误表
VGGT-Metric: 一个基于VGGT的度量深度估计模型
评论
全部评论