No module named 'models.dpt_block'

[tiandaoyuxi]

(venv) D:\dust3r>python demo.py --weights checkpoints/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth Traceback (most recent call last): File "D:\dust3r\demo.py", line 19, in from dust3r.inference import inference, load_model File "D:\dust3r\dust3r\inference.py", line 10, in from dust3r.model import AsymmetricCroCo3DStereo, inf # noqa: F401, needed when loading the model File "D:\dust3r\dust3r\model.py", line 11, in from .heads import head_factory File "D:\dust3r\dust3r\heads__init__.py", line 8, in from .dpt_head import create_dpt_head File "D:\dust3r\dust3r\heads\dpt_head.py", line 17, in from models.dpt_block import DPTOutputAdapter # noqa ModuleNotFoundError: No module named 'models.dpt_block'

(venv) D:\dust3r\checkpoints>dir 驱动器 D 中的卷是 Data 卷的序列号是 3C50-8BA1

D:\dust3r\checkpoints 的目录

2024/03/04 21:16

. 2024/03/04 21:16 .. 2024/03/04 20:35 2,129,660,080 DUSt3R_ViTLarge_BaseDecoder_224_linear.pth 2024/03/04 20:40 2,285,019,929 DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth 2024/03/04 20:41 2,129,656,556 DUSt3R_ViTLarge_BaseDecoder_512_linear.pth

[ffrivera0]

Hi,

The dpt_block.py should be at the croco/models/dpt_block.py. Could you check if croco/models is under your current working directory? It should be installed with --recursive.

git clone --recursive https://github.com/naver/dust3r

[tiandaoyuxi]

cool ,it's work.