xmrt commited on
Commit
1338b9b
1 Parent(s): 2c2a363

webcam test

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -117,8 +117,8 @@ def infer(video, check, kpt_thr, webcam=True):
117
  vidname = video.split("/")[-1]
118
  vis_out_dir = "/".join(["/".join(video.split("/")[:-1]), add_dir])
119
  out_file = poses(inferencer, video, vis_out_dir, kpt_thr)
120
- out_file = os.path.join(vis_out_dir, vidname)
121
- out_files.append(out_file)
122
 
123
  else:
124
  out_files.extend(out_file)
 
117
  vidname = video.split("/")[-1]
118
  vis_out_dir = "/".join(["/".join(video.split("/")[:-1]), add_dir])
119
  out_file = poses(inferencer, video, vis_out_dir, kpt_thr)
120
+ fullname = os.path.join(vis_out_dir, vidname)
121
+ out_files.append(fullname)
122
 
123
  else:
124
  out_files.extend(out_file)