tool

pose estimation tool.

Classes

PoseEstimationTool

Base class for pose estimation tools from video.
class PoseEstimationTool(device='cpu', overwrite=False, verbose=True)[source]

Bases: VideoTool

Base class for pose estimation tools from video.

Parameters:

  • device (str)

  • overwrite (bool)

  • verbose (bool | int)

inference(video_path, poses_path)[source]

Template of the inference method.

Parameters:

  • video_path (Union[str, Path]) – Path to the video file.

  • poses_path (Union[str, Path]) – Path to the pose archive.

Returns:

visualization(video_path, poses_path, visualization_path, confidence_threshold=0.0)[source]

Renders a visualization where the estimated poses are overlaid on top of the video.

Parameters:

  • video_path (Union[str, Path]) – Path to the video file.

  • poses_path (Union[str, Path, list[Union[str, Path]]]) – Path or list of path to the pose archive.

  • visualization_path (Union[str, Path]) – Path to the visualization file.

  • confidence_threshold (float) – Threshold for not displaying low confidence keypoints.

Returns: