first frame added to segmented video

app.py

@@ -165,6 +165,8 @@ 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 > 0) * 255 + \
+                        (masked_frame == 0) * frame[:, :, 2]
             else:
                 # track
                 state = siamese_track(
@@ -177,7 +179,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