Spaces:
Running on Zero
Running on Zero
No CUDA GPUs fix attempt
#3
by cbensimon HF Staff - opened
app.py
CHANGED
|
@@ -120,15 +120,14 @@ import cv2
|
|
| 120 |
from PIL import Image
|
| 121 |
|
| 122 |
from dwpose import DwposeDetector
|
| 123 |
-
processor = DwposeDetector.from_pretrained_default() #load pose processor by default
|
| 124 |
|
|
|
|
| 125 |
_depth_processor = None
|
| 126 |
|
| 127 |
|
| 128 |
def _get_pose_processor():
|
| 129 |
global _pose_processor
|
| 130 |
if _pose_processor is None:
|
| 131 |
-
from dwpose import DwposeDetector
|
| 132 |
_pose_processor = DwposeDetector.from_pretrained_default()
|
| 133 |
print("[Preprocess] DWPose processor loaded")
|
| 134 |
return _pose_processor
|
|
@@ -173,7 +172,7 @@ def extract_first_frame(video_path: str) -> str:
|
|
| 173 |
|
| 174 |
def preprocess_video_pose(frames: list[np.ndarray], width: int, height: int) -> list[np.ndarray]:
|
| 175 |
"""Extract DWPose skeletons from each frame. Returns float [0,1] frames."""
|
| 176 |
-
|
| 177 |
result = []
|
| 178 |
for frame in frames:
|
| 179 |
pil = Image.fromarray(frame.astype(np.uint8)).convert("RGB")
|
|
|
|
| 120 |
from PIL import Image
|
| 121 |
|
| 122 |
from dwpose import DwposeDetector
|
|
|
|
| 123 |
|
| 124 |
+
_pose_processor = None
|
| 125 |
_depth_processor = None
|
| 126 |
|
| 127 |
|
| 128 |
def _get_pose_processor():
|
| 129 |
global _pose_processor
|
| 130 |
if _pose_processor is None:
|
|
|
|
| 131 |
_pose_processor = DwposeDetector.from_pretrained_default()
|
| 132 |
print("[Preprocess] DWPose processor loaded")
|
| 133 |
return _pose_processor
|
|
|
|
| 172 |
|
| 173 |
def preprocess_video_pose(frames: list[np.ndarray], width: int, height: int) -> list[np.ndarray]:
|
| 174 |
"""Extract DWPose skeletons from each frame. Returns float [0,1] frames."""
|
| 175 |
+
processor = _get_pose_processor()
|
| 176 |
result = []
|
| 177 |
for frame in frames:
|
| 178 |
pil = Image.fromarray(frame.astype(np.uint8)).convert("RGB")
|