(ref_image_path)
| 179 | |
| 180 | |
| 181 | def get_image_pose(ref_image_path): |
| 182 | ref_image = cv2.imread(ref_image_path) |
| 183 | ref_image = cv2.cvtColor(ref_image, cv2.COLOR_BGR2RGB) |
| 184 | height, width, _ = ref_image.shape |
| 185 | ref_pose = dwpose_detector_aligned(ref_image) |
| 186 | pose_img = draw_pose(ref_pose, height, width) |
| 187 | return np.array(pose_img) |
| 188 | |
| 189 | if __name__ == '__main__': |
| 190 | parser = argparse.ArgumentParser(description="Skeleton extraction from images.") |
nothing calls this directly
no test coverage detected