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 returnsNormalizeInverse
.- Parameters:
- transform:
Torchvision transform.
- Returns:
- NormalizeInverse
NormalizeInverse object to undo the normalization in the input transform or None, if
Normalize
instance is not found.