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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -85,9 +85,9 @@ 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
-
89
  frame = transform({'image': frame})['image']
90
- frame_pil = Image.fromarray(np.uint8(frame)).convert('RGB')
91
  frame = torch.from_numpy(frame).unsqueeze(0).to(DEVICE)
92
 
93
 
 
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)
92
 
93