mmengine.dataset.pseudo_collate

mmengine.dataset.pseudo_collate(data_batch)

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.

The default behavior of dataloader is to merge a list of samples to form a mini-batch of Tensor(s). However, in MMEngine, pseudo_collate will not stack tensors to batch tensors, and convert int, float, ndarray to tensors.

This code is referenced from: Pytorch default_collate.

Parameters

data_batch (Sequence) – Batch of data from dataloader.

Returns

Transversed Data in the same format as the data_itement of data_batch.

Return type

Any