mmengine.dataset

Dataset

BaseDataset	BaseDataset for open source projects in OpenMMLab.
Compose	Compose multiple transforms sequentially.

Dataset Wrapper

ClassBalancedDataset	A wrapper of class balanced dataset.
ConcatDataset	A wrapper of concatenated dataset.
RepeatDataset	A wrapper of repeated dataset.

Sampler

DefaultSampler	The default data sampler for both distributed and non-distributed environment.
InfiniteSampler	It's designed for iteration-based runner and yields a mini-batch indices each time.

Utils

default_collate	Convert list of data sampled from dataset into a batch of data, of which type consistent with the type of each data_itement in data_batch.
pseudo_collate	Convert list of data sampled from dataset into a batch of data, of which type consistent with the type of each data_itement in data_batch.
worker_init_fn	This function will be called on each worker subprocess after seeding and before data loading.