Spaces:
Running
on
Zero
Running
on
Zero
enable zerogpu
Browse files
app.py
CHANGED
@@ -254,7 +254,7 @@ hands = mp_hands.Hands(
|
|
254 |
min_detection_confidence=0.1,
|
255 |
)
|
256 |
|
257 |
-
@spaces.GPU(duration=
|
258 |
def get_ref_anno(ref):
|
259 |
print("inside get_ref_anno")
|
260 |
if ref is None:
|
@@ -270,6 +270,7 @@ def get_ref_anno(ref):
|
|
270 |
keypts = np.zeros((42, 2))
|
271 |
print("ready to run mediapipe")
|
272 |
if REF_POSE_MASK:
|
|
|
273 |
mp_pose = hands.process(img)
|
274 |
print("processed mediapipe")
|
275 |
detected = np.array([0, 0])
|
|
|
254 |
min_detection_confidence=0.1,
|
255 |
)
|
256 |
|
257 |
+
@spaces.GPU(duration=240)
|
258 |
def get_ref_anno(ref):
|
259 |
print("inside get_ref_anno")
|
260 |
if ref is None:
|
|
|
270 |
keypts = np.zeros((42, 2))
|
271 |
print("ready to run mediapipe")
|
272 |
if REF_POSE_MASK:
|
273 |
+
print(f"type(img): {type(img)}, img.shape: {img.shape}, img.dtype: {img.dtype}")
|
274 |
mp_pose = hands.process(img)
|
275 |
print("processed mediapipe")
|
276 |
detected = np.array([0, 0])
|