get_data_module

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

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