Resize tensor without converting to PIL image?

2

Hi,

You can do it using interpolate function and it supports different methods.
Here is an example:


import PIL.Image as Image
from torchvision.transforms import ToTensor, ToPILImage
import torch.nn.functional as F

img = Image.open('data/Places365_val_00000001.jpg')
img = ToTensor()(img)
out = F.interpolate(img, size=128)  #The resize operation on tensor.
ToPILImage()(out).save('test.png', mode='png')

Bests
Nik

15 Likes