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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -85,7 +85,7 @@ def make_video(video_path, outdir='./vis_video_depth',encoder='vitl'):
85
  break
86
 
87
  frame = cv2.cvtColor(raw_frame, cv2.COLOR_BGR2RGB) / 255.0
88
- frame_pil = Image.fromarray(frame).convert('RGB')
89
  frame = transform({'image': frame})['image']
90
 
91
  frame = torch.from_numpy(frame).unsqueeze(0).to(DEVICE)
 
85
  break
86
 
87
  frame = cv2.cvtColor(raw_frame, cv2.COLOR_BGR2RGB) / 255.0
88
+ frame_pil = Image.fromarray((frame * 255).astype(np.uint8))
89
  frame = transform({'image': frame})['image']
90
 
91
  frame = torch.from_numpy(frame).unsqueeze(0).to(DEVICE)