Skip to content

Models

YoloV7

YoloV7PoseKeypoints

A class that represents pose keypoints and their confidences extracted by the YOLOv7 model.

__init__(raw_keypoints, height_prediction, width_prediction, height_original, width_original)

Initialize a YoloV7PoseKeypoints instance.

:param raw_keypoints: A list of raw keypoints and their confidences. :param height: The height of the image. :param width: The width of the image.

total_confidence_over(expected_confidence)

Returns keypoints whose confidence level exceeds the given threshold.

Attributes:

Name Type Description
expected_confidence float

confidence threshold to filter keypoints

Returns:

Type Description
list[YoloV7PoseKeypoint]

list[YoloV7PoseKeypoint]: keypoints above the confidence threshold.