Update app.py

app.py

@@ -225,7 +225,7 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', blur_data
             cv2.imwrite(f"f{count}_dmap.png", depth_color)
             depth_frames.append(f"f{count}_dmap.png")
 
-            depth_gray = seg_frame(depth_gray)
+            depth_gray = seg_frame(depth_gray) + 128
 
             cv2.imwrite(f"f{count}_mask.png", depth_gray)
             masks.append(f"f{count}_mask.png")
@@ -602,7 +602,7 @@ def draw_mask(l, t, b, v, d, evt: gr.EventData):
     x = points[len(points)-1][0]
     y = points[len(points)-1][1]
 
-    mask = cv2.imread(masks[frame_selected], cv2.IMREAD_GRAYSCALE).astype(np.uint8) + 128
+    mask = cv2.imread(masks[frame_selected], cv2.IMREAD_GRAYSCALE).astype(np.uint8)
     mask[mask==128] = 0
     d["layers"][0] = cv2.cvtColor(mask, cv2.COLOR_GRAY2RGBA)