Spaces:
Sleeping
Sleeping
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 |
)
|