xmrt commited on
Commit
8ce2567
1 Parent(s): 41707d0
Files changed (1) hide show
  1. main.py +3 -2
main.py CHANGED
@@ -50,6 +50,7 @@ def show_tracking(video_content, vis_out_dir, model):
50
 
51
  # Prepare to save video
52
  out_file = os.path.join(vis_out_dir, "track.mp4")
 
53
 
54
  fourcc = cv2.VideoWriter_fourcc(*"mp4v") # Codec for MP4 video
55
  fps = video.get(cv2.CAP_PROP_FPS)
@@ -103,10 +104,10 @@ def infer(video, check):
103
  else:
104
  out_file = poses(inferencer, video, vis_out_dir)
105
 
106
- out_files.append(out_file)
107
  print(out_files)
108
 
109
- return out_files[0], out_files[1], out_files[2], out_files[3]
110
 
111
  def run():
112
  #https://github.com/open-mmlab/mmpose/blob/main/docs/en/user_guides/inference.md
 
50
 
51
  # Prepare to save video
52
  out_file = os.path.join(vis_out_dir, "track.mp4")
53
+ print("[INFO]: TRACK", out_file)
54
 
55
  fourcc = cv2.VideoWriter_fourcc(*"mp4v") # Codec for MP4 video
56
  fps = video.get(cv2.CAP_PROP_FPS)
 
104
  else:
105
  out_file = poses(inferencer, video, vis_out_dir)
106
 
107
+ out_files.extend(out_file)
108
  print(out_files)
109
 
110
+ return out_files[0], out_files[1], out_files[2], out_files[3]
111
 
112
  def run():
113
  #https://github.com/open-mmlab/mmpose/blob/main/docs/en/user_guides/inference.md