yes im hving this issue too.
Any method that can fix it?
hi, this is my solution to that problem
#normalize images
image_transforms = Compose([
Resize((256,256)),
ToTensor(),
Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
])
#unnormalize images
def imshow(image):
npimg = image.numpy()
npimg = np.transpose(npimg, (1,2,0))
npimg = ((npimg * [0.229, 0.224, 0.225]) + [0.485, 0.456, 0.406])
plt.imshow(npimg, interpolation=‘nearest’)