def movenet_inference(model: tf.keras.Model, path: str, flip=False) -> tf.Tensor:
"""Function transforms an image to a tensor of shape [1, 17, 3]
containing y and x coordinates and confidence level for 17 keypoints.
:param model: movenet model in tensorflow format
:param path: Path to image file
:param flip: Optional argument indicating whether to flip the image left to right
:return: Tensor with data for 17 keypoints
"""
keypoints = model.inference_fn(get_image(path, flip))[0]
return keypoints
def movenet_inference_flat(model: tf.keras.Model, path: str, flip=False) -> np.array:
"""Function transforms an image to a numpy array shape [51,]
containing y and x coordinates and confidence level for 17 keypoints.
:param model: movenet model in tensorflow format
:param path: Path to image file
:param flip: Optional argument indicating whether to flip the image left to right
:return: 1D array with 51 values
"""
keypoints = model.inference_fn(get_image(path, flip))[0][0].numpy().flatten()
return keypoints
