mml.core.visualization.utils
- undo_image_normalization(images: Tensor, means: RGBInfo, stds: RGBInfo) Tensor[source]
Undoes the augmentation step of image normalization. This allows to visualize already augmented images.
- Parameters:
images (torch.Tensor) – batch of images
means (RGBInfo) – channel means used to normalize
stds (RGBInfo) – channel stds used to normalize
- Returns:
batch of images, interpretable for matplotlib, BUT: not clipped nor stretched to [0,255] range
- Return type: