Update demo/model.py

demo/model.py

@@ -333,7 +333,7 @@ class Model_all:
         # get depth 
         if type_in_depth == 'Depth':
             im_depth_out = im_depth.copy()
-            depth = img2tensor(im).unsqueeze(0) / 255.
+            depth = img2tensor(im_depth).unsqueeze(0) / 255.
         elif type_in_depth == 'Image':
             im_depth = img2tensor(im_depth).unsqueeze(0) / 127.5 - 1.0
             depth = self.depth_model(im_depth.to(self.device)).repeat(1, 3, 1, 1)