Update app.py
Browse files
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] = (
|
| 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)
|