no cuda??
Browse files
main.py
CHANGED
@@ -85,6 +85,7 @@ def poses(inferencer, video, vis_out_dir):
|
|
85 |
result = [result for result in result_generator] #next(result_generator)
|
86 |
|
87 |
out_file = glob.glob(os.path.join(vis_out_dir, "*.mp4"))
|
|
|
88 |
return out_file
|
89 |
|
90 |
def infer(video, check):
|
@@ -103,15 +104,15 @@ def infer(video, check):
|
|
103 |
out_file = poses(inferencer, video, vis_out_dir)
|
104 |
|
105 |
out_files.append(out_file)
|
|
|
106 |
|
107 |
-
return out_files
|
108 |
|
109 |
def run():
|
110 |
#https://github.com/open-mmlab/mmpose/blob/main/docs/en/user_guides/inference.md
|
111 |
check_web = gr.CheckboxGroup(choices = ["Estimate human 2d poses", "Estimate human 2d hand poses", "Estimate human 3d poses", "Detect and track"], label="Methods", type="value", info="Select the model(s) you want")
|
112 |
check_file = gr.CheckboxGroup(choices = ["Estimate human 2d poses", "Estimate human 2d hand poses", "Estimate human 3d poses", "Detect and track"], label="Methods", type="value", info="Select the model(s) you want")
|
113 |
|
114 |
-
|
115 |
webcam = gr.Interface(
|
116 |
fn=infer,
|
117 |
inputs= [gr.Video(source="webcam", height=412), check_web],
|
|
|
85 |
result = [result for result in result_generator] #next(result_generator)
|
86 |
|
87 |
out_file = glob.glob(os.path.join(vis_out_dir, "*.mp4"))
|
88 |
+
|
89 |
return out_file
|
90 |
|
91 |
def infer(video, check):
|
|
|
104 |
out_file = poses(inferencer, video, vis_out_dir)
|
105 |
|
106 |
out_files.append(out_file)
|
107 |
+
print(out_files)
|
108 |
|
109 |
+
return out_files[0], out_files[1], out_files[2], out_files[3]
|
110 |
|
111 |
def run():
|
112 |
#https://github.com/open-mmlab/mmpose/blob/main/docs/en/user_guides/inference.md
|
113 |
check_web = gr.CheckboxGroup(choices = ["Estimate human 2d poses", "Estimate human 2d hand poses", "Estimate human 3d poses", "Detect and track"], label="Methods", type="value", info="Select the model(s) you want")
|
114 |
check_file = gr.CheckboxGroup(choices = ["Estimate human 2d poses", "Estimate human 2d hand poses", "Estimate human 3d poses", "Detect and track"], label="Methods", type="value", info="Select the model(s) you want")
|
115 |
|
|
|
116 |
webcam = gr.Interface(
|
117 |
fn=infer,
|
118 |
inputs= [gr.Video(source="webcam", height=412), check_web],
|