DefaultTransform

class colibri.data.datasets.DefaultTransform(name)

Bases: object

Default transformation class.

This class is used to apply the default transformations to the data.

The default transformations are:

• input: transforms.ToTensor()

• output: transforms.ToTensor()

The default transformation for the output is transforms.ToTensor(), but it can be changed by providing a dictionary with the key ‘output’ in the transform_dict parameter. The default transformation for the input is transforms.ToTensor(), but it can be changed by providing a dictionary with the key ‘input’ in the transform_dict parameter.

Parameters:

name (string) – Name of the dataset.

default_transform(data)

Parameters:

data (object) – Data to transform

Returns:

Transformed data.

Return type:

object