diffengine.datasets.transforms.formatting

Module Contents

Classes

PackInputs

Pack the inputs data.

Functions

to_tensor(data)

Convert objects of various python types to torch.Tensor.

diffengine.datasets.transforms.formatting.to_tensor(data)

Convert objects of various python types to torch.Tensor.

Supported types are: numpy.ndarray, torch.Tensor, Sequence, int and float.

Return type:

torch.Tensor

class diffengine.datasets.transforms.formatting.PackInputs(input_keys=None, skip_to_tensor_key=None)

Bases: diffengine.datasets.transforms.BaseTransform

Pack the inputs data.

Required Keys:

• input_key

Deleted Keys:

All other keys in the dict.

Args:

input_keys (List[str]): The key of element to feed into the model

forwarding. Defaults to [‘img’, ‘text’].

skip_to_tensor_key (List[str]): The key of element to skip to_tensor.

Defaults to [‘text’].

transform(results)

Transform the data.

Parameters:

results (dict) –

Return type:

dict

Parameters:

• input_keys (list[str] | None) –

• skip_to_tensor_key (list[str] | None) –