Shortcuts

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.