Solved by Nikronic in post #4 
              
              
                Ow, I see, you need to unnormalize images after normalizing them for visualization purposes. 
Here is small snippet that does this: 
img = img.numpy().transpose((1, 2, 0))
mean = np.array([0.4451, 0.4262, 0.3959])  
std = np.array([0.2411, 0.2403, 0.2466])])
img = std * img + mean
img = np.clip(img*…