Function get_dataset

flowmap/dataset/__init__.py:28–37

(
        dataset_cfgs: list[DatasetCfg],
        stage: Stage,
        frame_sampler_cfg: FrameSamplerCfg,
    )

    )

    def get_dataset(
        dataset_cfgs: list[DatasetCfg],
        stage: Stage,
        frame_sampler_cfg: FrameSamplerCfg,
    ) -> DatasetMerged:
        frame_sampler = get_frame_sampler(frame_sampler_cfg)
        datasets = [
            DATASETS[cfg.name](cfg, stage, frame_sampler) for cfg in dataset_cfgs
        ]
        return DatasetMerged(datasets)

except ImportError:
    get_dataset = None