video writer fixed

app.py

@@ -163,8 +163,6 @@ def track_and_mask(vid, masked_frame, original_list, mask_list, in_fps, dt_strin
                 state = siamese_init(
                     frame, target_pos, target_sz, siammask, cfg['hp'], device=device)
                 original_list.append(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
-                frame[:, :, 2] = (masked_frame == 255) + \
-                        (masked_frame == 0) * frame[:, :, 2]
             else:
                 # track
                 state = siamese_track(
@@ -177,8 +175,8 @@ def track_and_mask(vid, masked_frame, original_list, mask_list, in_fps, dt_strin
                 mask = mask.astype(np.uint8)  # convert to an unsigned byte
                 mask = mask * 255
                 mask_list.append(mask)
-                
-            video_writer.write(frame)
+                video_writer.write(frame)
+
             f = f + 1
 
         video_capture.release()