mighty.utils.data.normalize.get_normalize_inverse

mighty.utils.data.normalize.get_normalize_inverse(transform)[source]

Traverses the input transform, finds a class of Normalize, and returns NormalizeInverse.

Parameters:
transform:

Torchvision transform.

Returns:
NormalizeInverse

NormalizeInverse object to undo the normalization in the input transform or None, if Normalize instance is not found.