get_dataset

lightning_pose.utils.scripts.get_dataset(cfg: DictConfig, data_dir: str, imgaug_transform: Sequential) → BaseTrackingDataset | HeatmapDataset | MultiviewHeatmapDataset[source]: Create a dataset that contains labeled data.