标签:

• pytorch_model_hub_mixin
• model_hub_mixin
• 图像转3D 库名称: dust3r 仓库地址: https://github.com/naver/dust3r

DUSt3R：轻松实现几何3D视觉

@inproceedings{dust3r_cvpr24,
      标题={DUSt3R：轻松实现几何3D视觉}, 
      作者={Shuzhe Wang、Vincent Leroy、Yohann Cabon、Boris Chidlovskii、Jerome Revaud},
      会议名称 = {CVPR},
      年份 = {2024}
}

@misc{dust3r_arxiv23,
      标题={DUSt3R：轻松实现几何3D视觉}, 
      作者={Shuzhe Wang、Vincent Leroy、Yohann Cabon、Boris Chidlovskii、Jerome Revaud},
      年份={2023},
      电子预印本={2312.14132},
      存档前缀={arXiv},
      主要类别={cs.CV},
      网址={https://arxiv.org/abs/2312.14132}, 
}

许可证

代码遵循CC BY-NC-SA 4.0许可证。更多信息请参阅LICENSE。 对于检查点，除了CC-BY-NC-SA 4.0外，请确保同意我们使用的所有公共训练数据集和基础检查点的许可证。详情请见我们的超参数部分。

模型信息

GitHub页面: https://github.com/naver/dust3r/ 项目页面: https://dust3r.europe.naverlabs.com/

使用方法

首先，安装dust3r。 加载模型：

from dust3r.model import AsymmetricCroCo3DStereo
import torch

model = AsymmetricCroCo3DStereo.from_pretrained("naver/DUSt3R_ViTLarge_BaseDecoder_224_linear")

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model.to(device)