freealise commited on
Commit
93d854e
·
verified ·
1 Parent(s): a8b085b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -446,7 +446,7 @@ def draw_mask(o, b, v, d, evt: gr.EventData):
446
  mask = np.where((mask==2)|(mask==0),0,1).astype('uint8')
447
 
448
  frame = cv2.imread(frames[frame_selected], cv2.IMREAD_UNCHANGED).astype(np.uint8)
449
- frame[mask>0] = (255,255,255)
450
  cv2.imwrite(frames[frame_selected], frame)
451
 
452
  switch_rows(False)
 
446
  mask = np.where((mask==2)|(mask==0),0,1).astype('uint8')
447
 
448
  frame = cv2.imread(frames[frame_selected], cv2.IMREAD_UNCHANGED).astype(np.uint8)
449
+ frame[mask>0] = (0,0,0,0)
450
  cv2.imwrite(frames[frame_selected], frame)
451
 
452
  switch_rows(False)