original frame insertion place changed

app.py

@@ -165,7 +165,7 @@ def track_and_mask(vid, original_frame, masked_frame):
 
             if not ret:
                 break
-
+            
             # frame = cv2.resize(frame, (w - w % 8, h - h % 8))
             if f == 0:
                 target_pos = np.array([x + w / 2, y + h / 2])
@@ -177,6 +177,8 @@ def track_and_mask(vid, original_frame, masked_frame):
                 # track
                 state = siamese_track(
                     state, frame, mask_enable=True, refine_enable=True, device=device)
+                frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+                original_frame_list.append(frame)
                 location = state['ploygon'].flatten()
                 mask = state['mask'] > state['p'].seg_thr
                 frame[:, :, 2] = (mask > 0) * 255 + \
@@ -184,12 +186,10 @@ 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)
 
-                original_frame_list.append(frame)
-                mask_list.append(mask)
-
                 video_writer.write(frame)
                 video_writer2.write(mask)
             f = f + 1