Update app.py

app.py

@@ -119,9 +119,8 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
             if not ret:
                 break
 
-            rgb_frame = cv2.cvtColor(raw_frame, cv2.COLOR_BGR2RGB)
-            frame_pil =  Image.fromarray(rgb_frame.astype(np.uint8))
-            frame = rgb_frame / 255.0
+            frame = cv2.cvtColor(raw_frame, cv2.COLOR_BGR2RGB) / 255.0
+            frame_pil =  Image.fromarray((frame * 255).astype(np.uint8))
             frame = transform({'image': frame})['image']
             
             frame = torch.from_numpy(frame).unsqueeze(0).to(DEVICE)
@@ -147,7 +146,7 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
             depth_color[mask>0] = (0,0,0)
 
             rgba_frame = cv2.cvtColor(raw_frame, cv2.COLOR_RGB2RGBA)
-            cv2.mixChannels((depth_color,), (rgba_frame,), (0,3,))
+            cv2.mixChannels((depth_gray,), (rgba_frame,), (0,3,))
             blur_frame = rgba_frame.copy()
             i = 240
             l = 0