xmrt commited on
Commit
577ab65
·
1 Parent(s): 43c10eb

Build error fix

Browse files
Files changed (1) hide show
  1. main.py +5 -4
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
- vis_out_dir = vis_out_dir,
79
- return_vis=True,
80
- thickness=2,
81
- rebase_keypoint_height=True)
 
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