xmrt commited on
Commit
b28330f
·
1 Parent(s): e973eb8
Files changed (1) hide show
  1. main.py +4 -1
main.py CHANGED
@@ -31,6 +31,7 @@ print("[INFO]: Downloaded models!")
31
 
32
  def poses(photo, check):
33
  # Selecting the specific inferencer
 
34
  inferencer = inferencers[check] # 'hand', 'human , device='cuda'
35
 
36
  print("[INFO]: Running inference!")
@@ -69,6 +70,8 @@ def poses(photo, check):
69
  # cv2.destroyAllWindows() # Closing window
70
  output_files = glob.glob(os.path.join(".", "*.mp4"))
71
  print(output_files)
 
 
72
 
73
  return output_files[0]
74
 
@@ -88,7 +91,7 @@ def poses(photo, check):
88
  def run():
89
  #https://github.com/open-mmlab/mmpose/blob/main/docs/en/user_guides/inference.md
90
  available_methods = ["Estimate human 2d poses", "Estimate human 2d hand poses", "Estimate human 3d poses"]
91
-
92
  webcam = gr.Interface(
93
  fn=poses,
94
  inputs= [gr.Video(source="webcam"),
 
31
 
32
  def poses(photo, check):
33
  # Selecting the specific inferencer
34
+ print(check)
35
  inferencer = inferencers[check] # 'hand', 'human , device='cuda'
36
 
37
  print("[INFO]: Running inference!")
 
70
  # cv2.destroyAllWindows() # Closing window
71
  output_files = glob.glob(os.path.join(".", "*.mp4"))
72
  print(output_files)
73
+ # 00000.mp4
74
+ # 000000
75
 
76
  return output_files[0]
77
 
 
91
  def run():
92
  #https://github.com/open-mmlab/mmpose/blob/main/docs/en/user_guides/inference.md
93
  available_methods = ["Estimate human 2d poses", "Estimate human 2d hand poses", "Estimate human 3d poses"]
94
+
95
  webcam = gr.Interface(
96
  fn=poses,
97
  inputs= [gr.Video(source="webcam"),