track name
Browse files
main.py
CHANGED
@@ -50,7 +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 |
-
out_file =
|
54 |
print("[INFO]: TRACK", out_file)
|
55 |
|
56 |
fourcc = cv2.VideoWriter_fourcc(*"mp4v") # Codec for MP4 video
|
@@ -103,7 +103,7 @@ def infer(video, check):
|
|
103 |
|
104 |
if i == "Detect and track":
|
105 |
#continue
|
106 |
-
out_file = show_tracking(video, vis_out_dir, inferencer)
|
107 |
|
108 |
else:
|
109 |
out_file = poses(inferencer, video, vis_out_dir)
|
|
|
50 |
|
51 |
# Prepare to save video
|
52 |
#out_file = os.path.join(vis_out_dir, "track.mp4")
|
53 |
+
out_file = "track.mp4"
|
54 |
print("[INFO]: TRACK", out_file)
|
55 |
|
56 |
fourcc = cv2.VideoWriter_fourcc(*"mp4v") # Codec for MP4 video
|
|
|
103 |
|
104 |
if i == "Detect and track":
|
105 |
#continue
|
106 |
+
out_file = list(show_tracking(video, vis_out_dir, inferencer))
|
107 |
|
108 |
else:
|
109 |
out_file = poses(inferencer, video, vis_out_dir)
|