Shortcuts

mmengine.dataset.pseudo_collate

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_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

Read the Docs v: v0.8.3
Versions
latest
stable
v0.8.3
v0.8.2
v0.8.1
v0.8.0
v0.7.4
v0.7.3
v0.7.2
v0.7.1
v0.7.0
v0.6.0
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.