许可证：Apache-2.0
语言：

• 英语
  基础模型：
• stabilityai/stable-diffusion-2
  管道标签：深度估计

DepthMaster：驯服扩散模型实现单目深度估计

宋子阳*, 王泽荣*, 李博, 张浩, 朱瑞杰, 刘力, 姜鹏涛†, 张天柱†,
*同等贡献，†通讯作者
中国科学技术大学，vivo移动通信有限公司
Arxiv 2025

我们提出DepthMaster，一个经过驯化的单步扩散模型，将扩散模型中的生成特征定制化以适应判别式深度估计任务。我们引入了特征对齐模块来减轻对纹理的过拟合，以及傅里叶增强模块来优化细粒度细节。DepthMaster展现出最先进的零样本性能和卓越的细节保留能力，在各种数据集上超越了其他基于扩散的方法。

🎓 引用

请引用我们的论文：

@article{song2025depthmaster,  
  title={DepthMaster：驯服扩散模型实现单目深度估计},  
  author={宋子阳 and 王泽荣 and 李博 and 张浩 and 朱瑞杰 and 刘力 and 姜鹏涛 and 张天柱},  
  journal={arXiv预印本 arXiv:2501.02576},  
  year={2025}  
}  

致谢

代码基于Marigold实现。

🎫 许可证

本作品采用Apache License 2.0许可（详见LICENSE）。

下载和使用代码及模型即表示您同意LICENSE中的条款。