ostapagon commited on
Commit
d0be95c
·
1 Parent(s): 3a1a6b5

Fix import error

Browse files
Files changed (1) hide show
  1. demo/mast3r_demo.py +1 -0
demo/mast3r_demo.py CHANGED
@@ -25,6 +25,7 @@ from mast3r.cloud_opt.tsdf_optimizer import TSDFPostProcess
25
  from mast3r.model import AsymmetricMASt3R
26
  from dust3r.image_pairs import make_pairs
27
  from dust3r.utils.image import load_images
 
28
  from dust3r.viz import add_scene_cam, CAM_COLORS, OPENGL, pts3d_to_trimesh, cat_meshes
29
  from dust3r.demo import get_args_parser as dust3r_get_args_parser
30
  from copy import deepcopy
 
25
  from mast3r.model import AsymmetricMASt3R
26
  from dust3r.image_pairs import make_pairs
27
  from dust3r.utils.image import load_images
28
+ from dust3r.utils.device import to_numpy
29
  from dust3r.viz import add_scene_cam, CAM_COLORS, OPENGL, pts3d_to_trimesh, cat_meshes
30
  from dust3r.demo import get_args_parser as dust3r_get_args_parser
31
  from copy import deepcopy