| import numpy as np | |
| import torchvision | |
| import PIL | |
| def web_input(image): | |
| image_transforms = torchvision.transforms.Compose([torchvision.transforms.ToTensor()]) | |
| input_img = PIL.Image.fromarray(image) | |
| input_img = input_img.resize((100, 100), PIL.Image.LANCZOS) | |
| wd_new, ht_new = input_img.size | |
| wd_new = int(16 * np.ceil(wd_new / 16.0)) | |
| ht_new = int(16 * np.ceil(ht_new / 16.0)) | |
| input_img = input_img.resize((wd_new, ht_new), PIL.Image.LANCZOS) | |
| return image_transforms(input_img).unsqueeze(0) | |