Muhammad Taqi Raza commited on
Commit
366a2ee
·
1 Parent(s): 7c35e92

adding depth visualization

Browse files
Files changed (1) hide show
  1. gradio_app.py +1 -1
gradio_app.py CHANGED
@@ -39,7 +39,7 @@ def visualize_depth_npy_as_video(npy_file):
39
  # Load .npy file
40
  depth_np = np.load(npy_file.name) # Shape: [T, 1, H, W]
41
  tensor = torch.from_numpy(depth_np)
42
-
43
 
44
  # Prepare video writer
45
  video_path = "depth_video_preview.mp4"
 
39
  # Load .npy file
40
  depth_np = np.load(npy_file.name) # Shape: [T, 1, H, W]
41
  tensor = torch.from_numpy(depth_np)
42
+ T, _, H, W = tensor.shape
43
 
44
  # Prepare video writer
45
  video_path = "depth_video_preview.mp4"