skipping tracking

main.py

@@ -50,6 +50,7 @@ def show_tracking(video_content, vis_out_dir, model):
 
         # Prepare to save video
         out_file = os.path.join(vis_out_dir, "track.mp4")
+        out_file = "track.mp4"
         print("[INFO]: TRACK", out_file)
 
         fourcc = cv2.VideoWriter_fourcc(*"mp4v")  # Codec for MP4 video
@@ -62,7 +63,7 @@ def show_tracking(video_content, vis_out_dir, model):
         # Go through frames and write them 
         for frame_track in video_track:
             result_track = frame_track[0].plot()  # plot a BGR numpy array of predictions
-
+        print("[INFO] Done with frames")
         #print(type(result_pose)) numpy ndarray
         out_track.write(result_track)
 
@@ -99,6 +100,7 @@ def infer(video, check):
         inferencer = inferencers[i] # 'hand', 'human , device='cuda'
 
         if i == "Detect and track":
+            continue
             out_file = show_tracking(video, vis_out_dir, inferencer)
 
         else:
@@ -108,7 +110,7 @@ def infer(video, check):
         out_files.extend(out_file)
         print(out_files)
 
-    return out_files[0], out_files[1], out_files[2], out_files[3]
+    return out_files[0], out_files[1], out_files[2]#, out_files[3]
 
 def run():
     #https://github.com/open-mmlab/mmpose/blob/main/docs/en/user_guides/inference.md
@@ -118,7 +120,7 @@ def run():
     webcam = gr.Interface(
         fn=infer,
         inputs= [gr.Video(source="webcam", height=412), check_web],
-        outputs = [gr.PlayableVideo(), gr.PlayableVideo(), gr.PlayableVideo(), gr.PlayableVideo()],
+        outputs = [gr.PlayableVideo(), gr.PlayableVideo(), gr.PlayableVideo()],# gr.PlayableVideo()],
         title = 'Pose estimation', 
         description = 'Pose estimation on video',
         allow_flagging=False
@@ -127,7 +129,7 @@ def run():
     file = gr.Interface(
         infer,
         inputs = [gr.Video(source="upload", height=412), check_file],
-        outputs = [gr.PlayableVideo(), gr.PlayableVideo(), gr.PlayableVideo(), gr.PlayableVideo()],
+        outputs = [gr.PlayableVideo(), gr.PlayableVideo(), gr.PlayableVideo()],#, gr.PlayableVideo()],
         allow_flagging=False
     )