Update app.py

app.py

@@ -196,7 +196,7 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', blur_data
             mask = cv2.inRange(depth_gray[int(depth_gray.shape[0]/7*6):depth_gray.shape[0], 0:depth_gray.shape[1]], 180, 255)
             depth_gray[int(depth_gray.shape[0]/7*6):depth_gray.shape[0], 0:depth_gray.shape[1]][mask>0] = 180
 
-            depth_color = cv2.cvtColor(depth_gray, cv2.COLOR_GRAY2BGR)
+            depth_color = cv2.cvtColor(depth_gray, cv2.COLOR_GRAY2BGRA)
             # split_region = np.ones((frame_height, margin_width, 3), dtype=np.uint8) * 255
             # combined_frame = cv2.hconcat([raw_frame, split_region, depth_color])
             
@@ -219,7 +219,7 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', blur_data
                       continue
               thumbnail_old = thumbnail
 
-            blur_frame = blur_image(raw_frame, depth_color, blur_data)
+            blur_frame = cv2.cvtColor(blur_image(raw_frame, depth_color, blur_data), cv2.COLOR_BGR2BGRA)
             comb_frame = np.concatenate((blur_frame, depth_color), axis=0)
 
             cv2.imwrite(f"f{count}_comb.png", comb_frame)