JohanDL commited on
Commit
ab4ab64
1 Parent(s): 65fb3e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -91,7 +91,7 @@ def make_video(video_path, outdir='./vis_video_depth',encoder='vitl'):
91
  frame = torch.from_numpy(frame).unsqueeze(0).to(DEVICE)
92
 
93
 
94
- depth = predict_depth(depth_anything, frame_pil)
95
 
96
  depth = F.interpolate(depth[None], (frame_height, frame_width), mode='bilinear', align_corners=False)[0, 0]
97
  depth = (depth - depth.min()) / (depth.max() - depth.min()) * 255.0
 
91
  frame = torch.from_numpy(frame).unsqueeze(0).to(DEVICE)
92
 
93
 
94
+ depth = torch.Tensor(predict_depth(depth_anything, frame_pil))
95
 
96
  depth = F.interpolate(depth[None], (frame_height, frame_width), mode='bilinear', align_corners=False)[0, 0]
97
  depth = (depth - depth.min()) / (depth.max() - depth.min()) * 255.0