JohanDL commited on
Commit
7962bc6
1 Parent(s): 936d537
Files changed (1) hide show
  1. app.py +1 -1
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