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

一次观察足矣:面向高分辨率图像零样本单目深度估计的无缝分块细化

One Look is Enough: Seamless Patchwise Refinement for Zero-Shot Monocular Depth Estimation on High-Resolution Images

KAIST(韩国科学技术院)

Byeongjun Kwon, Munchurl Kim

AI总结:

针对高分辨率图像零样本深度估计存在的精度低、内存消耗大、分块拼接不连续及泛化性差等问题,提出含分组分块一致性训练和无偏掩码的PRO框架,可高效无缝集成到现有模型中并提升泛化性。

摘要:

零样本深度估计(DE)模型因在大规模数据集上训练而展现出强大的泛化性能。然而,由于训练(分辨率较低)和推理(针对高分辨率)的图像分辨率存在差异,现有模型在处理高分辨率图像时存在困难。以全分辨率处理这类图像会导致深度估计精度下降,且内存消耗极大;而将图像下采样至训练分辨率则会使估计得到的深度图像出现边缘模糊问题。主流的高分辨率深度估计方法采用基于分块的思路,这种方法在重新拼接估计出的深度分块时会引入深度不连续问题,还会导致测试阶段效率低下。此外,为获取细粒度的深度细节,这类方法因真实场景的深度真值稀疏而依赖合成数据集,进而导致泛化性较差。为解决上述局限,我们提出了Patch Refine Once(PRO,单次分块细化),这是一种高效且具备泛化性的基于图块的框架。我们的PRO包含两个核心组件:(i)分组分块一致性训练,该机制通过在单次反向传播步骤中联合处理四个重叠分块,并对其重叠区域施加一致性损失,在提升测试阶段效率的同时缓解了深度不连续问题;(ii)无偏掩码,该机制可防止DE模型过拟合于数据集特有的偏差,即便在合成数据上训练后,也能更好地泛化至真实世界数据集。在Booster、ETH3D、Middlebury 2014和NuScenes上开展的零样本评估表明,我们的PRO可以无缝集成到现有的深度估计模型中。

英文摘要:

Zero-shot depth estimation (DE) models exhibit strong generalization performance as they are trained on large-scale datasets. However, existing models struggle with high-resolution images due to the discrepancy in image resolutions of training (with smaller resolutions) and inference (for high resolutions). Processing them at full resolution leads to decreased estimation accuracy on depth with tremendous memory consumption, while downsampling to the training resolution results in blurred edges in the estimated depth images. Prevailing high-resolution depth estimation methods adopt a patch-based approach, which introduces depth discontinuity issues when reassembling the estimated depth patches, resulting in test-time inefficiency. Additionally, to obtain fine-grained depth details, these methods rely on synthetic datasets due to the real-world sparse ground truth depth, leading to poor generalizability. To tackle these limitations, we propose Patch Refine Once (PRO), an efficient and generalizable tile-based framework. Our PRO consists of two key components: (i) Grouped Patch Consistency Training that enhances test-time efficiency while mitigating the depth discontinuity problem by jointly processing four overlapping patches and enforcing a consistency loss on their overlapping regions within a single backpropagation step, and (ii) Bias Free Masking that prevents the DE models from overfitting to dataset-specific biases, enabling better generalization to real-world datasets even after training on synthetic data. Zero-shot evaluations on Booster, ETH3D, Middlebury 2014, and NuScenes demonstrate that our PRO can be seamlessly integrated into existing depth estimation models.

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

↑