mighty.utils.data.normalize.get_normalize¶
- mighty.utils.data.normalize.get_normalize(transform, normalize_cls=<class 'torchvision.transforms.transforms.Normalize'>)[source]¶
Traverses the input transform and finds an instance of normalize_cls.
- Parameters:
- transform:
Torchvision transform
- normalize_clstype, optional
A class to look for in the input transform. Default: Normalize
- Returns:
- Normalize
Found normalize instance or None.