Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -258,7 +258,7 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', remove_bg
|
|
258 |
#stereo = cv2.StereoBM.create(numDisparities=16, blockSize=15)
|
259 |
#disparity = stereo.compute(raw_frame_l, raw_frame_r)
|
260 |
|
261 |
-
m = cv2.inRange(raw_frame, np.array([
|
262 |
raw_frame[m>0] = (239,239,239)
|
263 |
raw_frame[cv2.bitwise_and(mask_shadow, mask_bg)>0] = (raw_frame[cv2.bitwise_and(mask_shadow, mask_bg)>0] / 17 + 240).astype(np.uint8)
|
264 |
raw_frame[cv2.bitwise_and(mask_no_shadow, mask_bg)>0] = (255,255,255)
|
|
|
258 |
#stereo = cv2.StereoBM.create(numDisparities=16, blockSize=15)
|
259 |
#disparity = stereo.compute(raw_frame_l, raw_frame_r)
|
260 |
|
261 |
+
m = cv2.inRange(raw_frame, np.array([240,240,240]), np.array([255,255,255]))
|
262 |
raw_frame[m>0] = (239,239,239)
|
263 |
raw_frame[cv2.bitwise_and(mask_shadow, mask_bg)>0] = (raw_frame[cv2.bitwise_and(mask_shadow, mask_bg)>0] / 17 + 240).astype(np.uint8)
|
264 |
raw_frame[cv2.bitwise_and(mask_no_shadow, mask_bg)>0] = (255,255,255)
|