new list
Browse files
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.
|
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
|