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

PatchRefiner V2:快速轻量的真实域高分辨率度量深度估计

PatchRefiner V2: Fast and Lightweight Real-Domain High-Resolution Metric Depth Estimation

King Abdullah University of Science and Technology (KAUST)(阿卜杜拉国王科技大学)

Zhenyu Li, Wenqing Cui, Shariq Farooq Bhat, Peter Wonka

AI总结:

PatchRefiner V2 通过轻量级编码器、由粗到细模块与引导去噪单元、噪声预训练策略及 SSIGM 损失,在保持高精度的同时大幅降低模型规模和推理时间,并在多个真实域数据集上验证了其高效性与通用性。

摘要:

当前高分辨率深度估计方法虽然取得了优异的结果,但由于依赖重量级模型和多次推理步骤,往往存在计算效率低下的问题,导致推理时间增加。为解决这一问题,我们提出了 PatchRefiner V2(PRV2),用轻量级编码器替代重量级细化器模型。这减小了模型规模并缩短了推理时间,但引入了噪声特征。为克服这一问题,我们提出了一个由粗到细(Coarse-to-Fine,C2F)模块,其中包含一个引导去噪单元(Guided Denoising Unit),用于细化和去噪细化器特征,并提出了一种噪声预训练(Noisy Pretraining)策略来预训练细化器分支,以充分发挥轻量级细化器分支的潜力。此外,我们引入了一种尺度与位移不变的梯度匹配(Scale-and-Shift Invariant Gradient Matching,SSIGM)损失,以增强合成域到真实域的迁移能力。PRV2 在 UnrealStereo4K 上以更少的参数和更快的推理速度,在精度和速度两方面均超越了最先进的深度估计方法。它还在 CityScape、ScanNet++ 和 KITTI 等真实世界数据集上展现出改进的深度边界刻画能力,证明了其跨领域的通用性。

英文摘要:

While current high-resolution depth estimation methods achieve strong results, they often suffer from computational inefficiencies due to reliance on heavyweight models and multiple inference steps, increasing inference time. To address this, we introduce PatchRefiner V2 (PRV2), which replaces heavy refiner models with lightweight encoders. This reduces model size and inference time but introduces noisy features. To overcome this, we propose a Coarse-to-Fine (C2F) module with a Guided Denoising Unit for refining and denoising the refiner features and a Noisy Pretraining strategy to pretrain the refiner branch to fully exploit the potential of the lightweight refiner branch. Additionally, we introduce a Scale-and-Shift Invariant Gradient Matching (SSIGM) loss to enhance synthetic-to-real domain transfer. PRV2 outperforms state-of-the-art depth estimation methods on UnrealStereo4K in both accuracy and speed, using fewer parameters and faster inference. It also shows improved depth boundary delineation on real-world datasets like CityScape, ScanNet++, and KITTI, demonstrating its versatility across domains.

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

↑