Spaces:

oguzakif
/

video-object-remover

Runtime error

oguzakif commited on Apr 20, 2023

Commit

5ad413a

1 Parent(s): 9ccc177

change mask adding to list

Files changed (1) hide show

app.py CHANGED Viewed

@@ -180,7 +180,6 @@ def track_and_mask(vid, original_frame, masked_frame):
                 mask = mask.astype(np.uint8)  # convert to an unsigned byte
                 mask = mask * 255
-                mask_list.append(mask)
                 cv2.polylines(frame, [np.int0(location).reshape(
                     (-1, 1, 2))], True, (0, 255, 0), 3)

                 mask = mask.astype(np.uint8)  # convert to an unsigned byte
                 mask = mask * 255
                 cv2.polylines(frame, [np.int0(location).reshape(
                     (-1, 1, 2))], True, (0, 255, 0), 3)