Spaces:
Paused
Paused
Muhammad Taqi Raza
commited on
Commit
·
366a2ee
1
Parent(s):
7c35e92
adding depth visualization
Browse files- 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"
|