def showRandomImages(data, targets, predictions=None, classes=None, count=20, muSigmaPair=None):
randImages = Utility.pickRandomElements(data, count)
images = data[randImages]
if (muSigmaPair is not None):
images = Utility.unnormalize(images, muSigmaPair[0], muSigmaPair[1])
images = images.permute(0, 2, 3, 1)
targets = __getLabels(targets, randImages, classes)
if predictions is not None:
predictions = __getLabels(predictions, randImages, classes)
showImages(images.numpy(), targets, predictions, cols=5)
def showLoaderImages(loader, classes=None, count=20, muSigmaPair=None):
"""
Takes random images from the loader and shows the images.
Optionally Mean and Sigma pair can be passed to unnormalize data before showing the image.
:param muSigmaPair: Default is (0, 1)
"""
d, l = iter(loader).next()
randImages = Utility.pickRandomElements(d, count)
images = d[randImages]
if (muSigmaPair is not None):
images = Utility.unnormalize(images, muSigmaPair[0], muSigmaPair[1])
# Loader has the channel at 1 index. But the show images need channel at the end.
images = images.permute(0, 2, 3, 1)
labels = __getLabels(l, randImages, classes)
showImages(images.numpy(), labels, cols=5)