get_data_module

lightning_pose.utils.scripts.get_data_module(cfg: DictConfig, dataset: BaseTrackingDataset | HeatmapDataset | MultiviewHeatmapDataset, video_dir: str | None = None) → BaseDataModule | UnlabeledDataModule

Create a data module that splits a dataset into train/val/test iterators.