Spaces:
Build error
Build error
fix(utils): adjust some params
Browse files
utils.py
CHANGED
|
@@ -137,8 +137,6 @@ def preprocess(
|
|
| 137 |
model_num_frames: int,
|
| 138 |
keypoints_detector,
|
| 139 |
source: str,
|
| 140 |
-
data_height: int,
|
| 141 |
-
data_width: int,
|
| 142 |
model_input_height: int,
|
| 143 |
model_input_width: int,
|
| 144 |
device: str,
|
|
@@ -157,7 +155,7 @@ def preprocess(
|
|
| 157 |
# Detect keypoints.
|
| 158 |
detection_results = keypoints_detector.process(frame)
|
| 159 |
skeleton_frame = draw_skeleton_on_image(
|
| 160 |
-
image=np.zeros((
|
| 161 |
detection_results=detection_results,
|
| 162 |
resize_to=(model_input_height, model_input_width),
|
| 163 |
)
|
|
|
|
| 137 |
model_num_frames: int,
|
| 138 |
keypoints_detector,
|
| 139 |
source: str,
|
|
|
|
|
|
|
| 140 |
model_input_height: int,
|
| 141 |
model_input_width: int,
|
| 142 |
device: str,
|
|
|
|
| 155 |
# Detect keypoints.
|
| 156 |
detection_results = keypoints_detector.process(frame)
|
| 157 |
skeleton_frame = draw_skeleton_on_image(
|
| 158 |
+
image=np.zeros((1080, 1080, 3), dtype=np.uint8),
|
| 159 |
detection_results=detection_results,
|
| 160 |
resize_to=(model_input_height, model_input_width),
|
| 161 |
)
|