variable init position bug fix

app.py

@@ -162,19 +162,21 @@ def track_and_mask(vid, masked_frame, original_list, mask_list, in_fps, dt_strin
                 # init tracker
                 state = siamese_init(
                     frame, target_pos, target_sz, siammask, cfg['hp'], device=device)
+                original_list.append(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
+                frame[:, :, 2] = (mask > 0) * 255 + \
+                        (mask == 0) * frame[:, :, 2]
             else:
                 # track
                 state = siamese_track(
                     state, frame, mask_enable=True, refine_enable=True, device=device)
-                
-            original_list.append(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
-            mask = state['mask'] > state['p'].seg_thr
-            frame[:, :, 2] = (mask > 0) * 255 + \
-                    (mask == 0) * frame[:, :, 2]
-
-            mask = mask.astype(np.uint8)  # convert to an unsigned byte
-            mask = mask * 255
-            mask_list.append(mask)
+                original_list.append(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
+                mask = state['mask'] > state['p'].seg_thr
+                frame[:, :, 2] = (mask > 0) * 255 + \
+                        (mask == 0) * frame[:, :, 2]
+
+                mask = mask.astype(np.uint8)  # convert to an unsigned byte
+                mask = mask * 255
+                mask_list.append(mask)
                 
             video_writer.write(frame)
             f = f + 1