HyperAIHyperAI

Command Palette

Search for a command to run...

DepthMaster:单目深度估计中的扩散模型控制

Ziyang Song Zerong Wang Bo Li Hao Zhang Ruijie Zhu Li Liu Peng-Tao Jiang Tianzhu Zhang

摘要

在扩散去噪范式下的单目深度估计展示了令人印象深刻的泛化能力,但其推理速度较低。近期的方法采用了单步确定性范式来提高推理效率,同时保持了相当的性能。然而,这些方法忽视了生成特征与判别特征之间的差距,导致结果次优。在这项工作中,我们提出了DepthMaster,这是一种单步扩散模型,旨在将生成特征适应于判别深度估计任务。首先,为了缓解生成特征引入的纹理细节过拟合问题,我们提出了一种特征对齐模块(Feature Alignment module),该模块融合高质量的语义特征以增强去噪网络的表示能力。其次,为了解决单步确定性框架中缺乏细粒度细节的问题,我们提出了一种傅里叶增强模块(Fourier Enhancement module),以自适应地平衡低频结构和高频细节。我们采用两阶段训练策略来充分发挥这两个模块的潜力。在第一阶段,我们专注于利用特征对齐模块学习全局场景结构;而在第二阶段,则利用傅里叶增强模块提升视觉质量。通过这些努力,我们的模型在泛化能力和细节保留方面达到了最先进的性能,在多个数据集上均优于其他基于扩散的方法。项目页面可访问 https://indu1ge.github.io/DepthMaster_page


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供