track name

main.py

@@ -84,7 +84,9 @@ def poses(inferencer, video, vis_out_dir):
                                  return_vis=True,
                                  thickness=2,
                                  rebase_keypoint_height=True,
-                                 device="cuda")    
+                                 #kpt_thr=0.15,
+                                 device="cuda"
+                                 )    
     
     result = [result for result in result_generator] #next(result_generator)    
 
@@ -103,15 +105,16 @@ def infer(video, check):
 
         if i == "Detect and track":
             #continue
-            out_file = list(show_tracking(video, vis_out_dir, inferencer))
+            trackfile = show_tracking(video, vis_out_dir, inferencer)
 
         else:
             out_file = poses(inferencer, video, vis_out_dir)                                                                                                                                                                                                    
         
-        out_files.extend(out_file)
+            out_files.extend(out_file)
+        
         print(out_files)
 
-    return "track.mp4", out_files[1], out_files[2], out_files[3] # out_files[3]
+    return "track.mp4", 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