vijish
(vijish)
1
def preprocess(image_name, image_size,image_name2,image_size2):
image1 = Image.open(image_name).convert('RGB')
if type(image_size) is not tuple:
image_size = tuple([int((float(image_size) / max(image.size))*x) for x in (image.height,
image.width)])
Loader = transforms.Compose([transforms.Resize(image_size), transforms.ToTensor()])
tensor1 = (Loader(image1) * 256).unsqueeze(0)
image2 = Image.open(image_name2).convert('RGB')
if type(image_size2) is not tuple:
image_size2 = tuple([int((float(image_size2) / max(image.size2))*x) for x in (image.height,
image.width)])
Loader = transforms.Compose([transforms.Resize(image_size2), transforms.ToTensor()])
tensor2 = (Loader(image2) * 256).unsqueeze(0)
lin_mask_1 = torch.linspace(0,1,1080).repeat(1080,1).repeat(3,1,1).unsqueeze(0)
lin_mask_2 = torch.linspace(1,0,1080).repeat(1080,1).repeat(3,1,1).unsqueeze(0)
output_tensor= tensor1 * lin_mask_1 + tensor2 * lin_mask_2
output_tensor = output_tensor.squeeze(0).cpu() / 256
output_tensor.clamp_(0, 1)
Image2PIL = transforms.ToPILImage()
image = Image2PIL(output_tensor.cpu())
return image