FiftyOneImagePlotter

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

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][source]

create_dataset() → Dataset[source]

dataset_info_print() → str[source]

get_gt_keypoints_list() → list[Keypoints][source]

get_keypoints_per_image(data_dict: dict[str, dict[str, array]]) → list[Keypoints][source]: iterates over the rows of the dataframe and gathers keypoints in fiftyone format

get_model_abs_paths() → list[str][source]

get_pred_keypoints_dict() → dict[str, list[Keypoints]][source]

img_height_width(idx) → int[source]

load_model_predictions() → None[source]

__init__(cfg: DictConfig, keypoints_to_plot: list[str] | None = None, csv_filename: str = 'predictions.csv') → None[source]