SRMD-pytorch
SRMD-pytorch copied to clipboard
Degradation method didn't match
In torchvision the default degradation method is PIL.Image.BILINEAR
def __getitem__(self, index):
"""Read an image from a file and preprocesses it and returns."""
image_path = self.image_paths[index]
image = Image.open(image_path).convert('RGB')
# target (high-resolution image)
transform = transforms.RandomCrop(self.image_size * self.scale_factor) # transforms.Resize((self.image_size, self.image_size),interpolation=Image.BICUBIC)
hr_image = transform(image)
# input (low-resolution image)
transform = transforms.Compose([
transforms.Lambda(lambda x: self.randkern.RandomBlur(x)),
**transforms.Resize((self.image_size, self.image_size)),**
transforms.Lambda(lambda x: Scaling(x)),
transforms.Lambda(lambda x: self.randkern.ConcatDegraInfo(x))
])
lr_image = transform(hr_image)
transform = transforms.ToTensor()
lr_image, hr_image = transform(lr_image), transform(hr_image)
return lr_image.to(torch.float64), hr_image.to(torch.float64)