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.