FiftyOneImagePlotter

class lightning_pose.utils.fiftyone.FiftyOneImagePlotter(cfg: DictConfig, keypoints_to_plot: List[str] | None = None, csv_filename: str = 'predictions.csv')

Bases: object

Attributes Summary

model_names
num_keypoints

Methods Summary

build_single_frame_keypoints(data_dict, ...)
create_dataset()
dataset_info_print()
get_gt_keypoints_list()
get_keypoints_per_image(data_dict)	iterates over the rows of the dataframe and gathers keypoints in fiftyone format
get_model_abs_paths()
get_pred_keypoints_dict()
img_height_width(idx)
load_model_predictions()

Attributes Documentation

model_names

num_keypoints

Methods Documentation

build_single_frame_keypoints(data_dict: Dict[str, Dict[str, array]], frame_idx: int, height: int, width: int) → List[Keypoint]

create_dataset() → Dataset

dataset_info_print() → str

get_gt_keypoints_list() → List[Keypoints]

get_keypoints_per_image(data_dict: Dict[str, Dict[str, array]]) → List[Keypoints]

iterates over the rows of the dataframe and gathers keypoints in fiftyone format

get_model_abs_paths() → List[str]

get_pred_keypoints_dict() → Dict[str, List[Keypoints]]

img_height_width(idx) → int

load_model_predictions() → None