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. # noqa: E501 :param data_batch: Batch of data from dataloader. :type data_batch: Sequence

返回

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

返回类型

Any

参数

data_batch (Sequence) –