mmengine.dataset.pseudo_collate — mmengine 0.10.7 documentation (original) (raw)

mmengine.dataset.pseudo_collate(data_batch)[source]¶

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_collatewill 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 ofdata_batch.

Return type:

Any