Build error fix
Browse files
main.py
CHANGED
@@ -45,7 +45,7 @@ def show_tracking(video_content, vis_out_dir, model):
|
|
45 |
video = cv2.VideoCapture(video_content)
|
46 |
|
47 |
# Track
|
48 |
-
video_track = tracking(video_content, model
|
49 |
|
50 |
# Prepare to save video
|
51 |
out_file = os.path.join(vis_out_dir, "track.mp4")
|
@@ -95,7 +95,7 @@ def infer(video, check):
|
|
95 |
inferencer = inferencers[i] # 'hand', 'human , device='cuda'
|
96 |
|
97 |
print("[INFO]: Running inference!")
|
98 |
-
out_file = poses(inferencer, video)
|
99 |
|
100 |
if i == "Detect and track":
|
101 |
out_file = show_tracking(video, vis_out_dir)
|
|
|
45 |
video = cv2.VideoCapture(video_content)
|
46 |
|
47 |
# Track
|
48 |
+
video_track = tracking(video_content, model.track)
|
49 |
|
50 |
# Prepare to save video
|
51 |
out_file = os.path.join(vis_out_dir, "track.mp4")
|
|
|
95 |
inferencer = inferencers[i] # 'hand', 'human , device='cuda'
|
96 |
|
97 |
print("[INFO]: Running inference!")
|
98 |
+
out_file = poses(inferencer, video)
|
99 |
|
100 |
if i == "Detect and track":
|
101 |
out_file = show_tracking(video, vis_out_dir)
|