Hi!
I’m using save_image after some conv layer to output the image. But since the output shape of the tensor is torch.Size([32, 1, 3, 3]). I end up having very tiny images. How can I resize before calling save_image.
By the way, using scipy works
img = x.detach().cpu().numpy()
img = img[0] # Taking one image to test with
img = np.transpose(img, (2, 1, 0))
print(img.shape)
from scipy.misc import imsave, imresize
img = imresize(img, (224, 224))
imsave("./images/att.png", img)

Thank you