Build error fix
Browse files
main.py
CHANGED
@@ -75,10 +75,11 @@ def show_tracking(video_content, vis_out_dir, model):
|
|
75 |
def poses(inferencer, video, vis_out_dir):
|
76 |
|
77 |
result_generator = inferencer(video,
|
78 |
-
|
79 |
-
|
80 |
-
|
81 |
-
|
|
|
82 |
|
83 |
result = [result for result in result_generator] #next(result_generator)
|
84 |
|
|
|
75 |
def poses(inferencer, video, vis_out_dir):
|
76 |
|
77 |
result_generator = inferencer(video,
|
78 |
+
vis_out_dir = vis_out_dir,
|
79 |
+
return_vis=True,
|
80 |
+
thickness=2,
|
81 |
+
rebase_keypoint_height=True,
|
82 |
+
device="cuda")
|
83 |
|
84 |
result = [result for result in result_generator] #next(result_generator)
|
85 |
|