Spaces:
Running
Running
Fix bug
Browse files
app.py
CHANGED
@@ -83,7 +83,7 @@ def make_video(video_path, outdir='./vis_video_depth',encoder='vitl'):
|
|
83 |
frame = transform({'image': frame})['image']
|
84 |
frame = torch.from_numpy(frame).unsqueeze(0).to(DEVICE)
|
85 |
|
86 |
-
predict_depth(depth_anything, frame)
|
87 |
|
88 |
depth = F.interpolate(depth[None], (frame_height, frame_width), mode='bilinear', align_corners=False)[0, 0]
|
89 |
depth = (depth - depth.min()) / (depth.max() - depth.min()) * 255.0
|
|
|
83 |
frame = transform({'image': frame})['image']
|
84 |
frame = torch.from_numpy(frame).unsqueeze(0).to(DEVICE)
|
85 |
|
86 |
+
depth = predict_depth(depth_anything, frame)
|
87 |
|
88 |
depth = F.interpolate(depth[None], (frame_height, frame_width), mode='bilinear', align_corners=False)[0, 0]
|
89 |
depth = (depth - depth.min()) / (depth.max() - depth.min()) * 255.0
|