deepsphere.data.transforms package

Submodules

deepsphere.data.transforms.transforms module

Transformations for samples of atmospheric rivers and tropical cyclones dataset.

class deepsphere.data.transforms.transforms.Normalize(mean, std)[source]

Bases: object

Normalize using mean and std.

class deepsphere.data.transforms.transforms.Permute[source]

Bases: object

Permute first and second dimension.

class deepsphere.data.transforms.transforms.Stack(dimension=0)[source]

Bases: object

Stack images in torch tensor.

class deepsphere.data.transforms.transforms.ToTensor[source]

Bases: object

Convert raw data and labels to PyTorch tensor.

Module contents