xmrt commited on
Commit
41707d0
1 Parent(s): 86010a2
Files changed (1) hide show
  1. main.py +3 -2
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],