webcam test
Browse files
main.py
CHANGED
|
@@ -113,9 +113,12 @@ def infer(video, check, kpt_thr, webcam=True):
|
|
| 113 |
|
| 114 |
else:
|
| 115 |
if webcam==True:
|
| 116 |
-
|
| 117 |
-
|
| 118 |
-
|
|
|
|
|
|
|
|
|
|
| 119 |
|
| 120 |
else:
|
| 121 |
out_files.extend(out_file)
|
|
|
|
| 113 |
|
| 114 |
else:
|
| 115 |
if webcam==True:
|
| 116 |
+
add_dir = str(uuid.uuid4())
|
| 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)
|