first frames added to seg and inpainted results

app.py

@@ -134,8 +134,11 @@ def getBoundaries(mask):
 def track_and_mask(vid, masked_frame, original_list, mask_list, in_fps, dt_string):
     x, y, w, h = getBoundaries(masked_frame)
     f = 0
-
+    
+    #add first mask frame of the video by default
+    mask_list.append(masked_frame)
     video_capture = cv2.VideoCapture()
+
     if video_capture.open(vid):
         width, height = int(video_capture.get(cv2.CAP_PROP_FRAME_WIDTH)), int(
             video_capture.get(cv2.CAP_PROP_FRAME_HEIGHT))
@@ -163,19 +166,17 @@ def track_and_mask(vid, masked_frame, original_list, mask_list, in_fps, dt_strin
                 # track
                 state = siamese_track(
                     state, frame, mask_enable=True, refine_enable=True, device=device)
-                original_list.append(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
-                location = state['ploygon'].flatten()
-                mask = state['mask'] > state['p'].seg_thr
-                frame[:, :, 2] = (mask > 0) * 255 + \
+                
+            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)
-                #cv2.polylines(frame, [np.int0(location).reshape(
-                #    (-1, 1, 2))], True, (0, 255, 0), 3)
-
-                video_writer.write(frame)
+            mask = mask.astype(np.uint8)  # convert to an unsigned byte
+            mask = mask * 255
+            mask_list.append(mask)
+                
+            video_writer.write(frame)
             f = f + 1
 
         video_capture.release()