tanthinhdt commited on
Commit
4703605
·
verified ·
1 Parent(s): b7162ea

fix(utils): adjust some params

Browse files
Files changed (1) hide show
  1. utils.py +1 -3
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((data_height, data_width, 3), dtype=np.uint8),
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
  )