SRMD-pytorch icon indicating copy to clipboard operation
SRMD-pytorch copied to clipboard

Degradation method didn't match

Open wyf0912 opened this issue 5 years ago • 1 comments

In torchvision the default degradation method is PIL.Image.BILINEAR

wyf0912 avatar Nov 24 '19 13:11 wyf0912

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)

wyf0912 avatar Nov 24 '19 13:11 wyf0912