Spaces:
Sleeping
Sleeping
Commit
·
eb20ed3
1
Parent(s):
5a28175
trying threading
Browse files
app.py
CHANGED
|
@@ -1,6 +1,8 @@
|
|
| 1 |
import cv2
|
| 2 |
import mediapipe as mp
|
| 3 |
import numpy as np
|
|
|
|
|
|
|
| 4 |
mp_drawing = mp.solutions.drawing_utils
|
| 5 |
mp_pose = mp.solutions.pose
|
| 6 |
import gradio as gr
|
|
@@ -35,13 +37,37 @@ def detect_pose(im):
|
|
| 35 |
|
| 36 |
return image
|
| 37 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 38 |
with gr.Blocks() as app:
|
| 39 |
gr.Markdown("# Webcam Testing")
|
| 40 |
with gr.Row():
|
| 41 |
-
inp =
|
| 42 |
out = gr.Image()
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 46 |
|
| 47 |
app.launch(debug=True)
|
|
|
|
| 1 |
import cv2
|
| 2 |
import mediapipe as mp
|
| 3 |
import numpy as np
|
| 4 |
+
import threading
|
| 5 |
+
|
| 6 |
mp_drawing = mp.solutions.drawing_utils
|
| 7 |
mp_pose = mp.solutions.pose
|
| 8 |
import gradio as gr
|
|
|
|
| 37 |
|
| 38 |
return image
|
| 39 |
|
| 40 |
+
pose_results = None
|
| 41 |
+
|
| 42 |
+
def update_pose_results(im):
|
| 43 |
+
global pose_results
|
| 44 |
+
pose_results = detect_pose(im)
|
| 45 |
+
|
| 46 |
+
|
| 47 |
with gr.Blocks() as app:
|
| 48 |
gr.Markdown("# Webcam Testing")
|
| 49 |
with gr.Row():
|
| 50 |
+
inp = gr.Image(source="webcam", streaming=True)
|
| 51 |
out = gr.Image()
|
| 52 |
+
|
| 53 |
+
# Start a separate thread for pose detection
|
| 54 |
+
def pose_detection_thread():
|
| 55 |
+
while True:
|
| 56 |
+
update_pose_results(inp.value)
|
| 57 |
+
|
| 58 |
+
pose_thread = threading.Thread(target=pose_detection_thread)
|
| 59 |
+
pose_thread.daemon = True # Allow the thread to exit when the main program exits
|
| 60 |
+
pose_thread.start()
|
| 61 |
+
|
| 62 |
+
# Continuously update the output with pose results
|
| 63 |
+
def update_output():
|
| 64 |
+
global pose_results
|
| 65 |
+
while True:
|
| 66 |
+
if pose_results is not None:
|
| 67 |
+
out.value = pose_results
|
| 68 |
+
|
| 69 |
+
output_thread = threading.Thread(target=update_output)
|
| 70 |
+
output_thread.daemon = True
|
| 71 |
+
output_thread.start()
|
| 72 |
|
| 73 |
app.launch(debug=True)
|