Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
-
|
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 |
|