freealise commited on
Commit
29f2dbd
1 Parent(s): 89f1281

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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([255,255,255]), 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)
 
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)