get_dataset

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