diff --git a/dust3r/viz.py b/dust3r/viz.py index 9150e8b..cb7f98a 100644 --- a/dust3r/viz.py +++ b/dust3r/viz.py @@ -171,6 +171,7 @@ def add_pointcloud(self, pts3d, color=(0,0,0), mask=None, denoise=False): return self def add_rgbd(self, image, depth, intrinsics=None, cam2world=None, zfar=np.inf, mask=None): + image = img_to_arr(image) # make up some intrinsics if intrinsics is None: H, W, THREE = image.shape