checkbox
Browse files
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"),
|