From e8e48a2070bdcead2b6b684d02c61b6619cfc5ca Mon Sep 17 00:00:00 2001 From: Dominic-ZZ <57563083+Dominic-ZZ@users.noreply.github.com> Date: Fri, 4 Jun 2021 16:53:57 +0800 Subject: [PATCH] Update dataset.py AttributeError: module 'torchvision.transforms' has no attribute 'InterpolationMode' --- src/data/dataset.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/data/dataset.py b/src/data/dataset.py index 3874640..a9e77b7 100644 --- a/src/data/dataset.py +++ b/src/data/dataset.py @@ -30,10 +30,10 @@ class InpaintingData(Dataset): transforms.ColorJitter(0.05, 0.05, 0.05, 0.05), transforms.ToTensor()]) self.mask_trans = transforms.Compose([ - transforms.Resize(args.image_size, interpolation=transforms.InterpolationMode.NEAREST), + transforms.Resize(args.image_size), transforms.RandomHorizontalFlip(), transforms.RandomRotation( - (0, 45), interpolation=transforms.InterpolationMode.NEAREST), + (0, 45)), ]) @@ -77,4 +77,4 @@ if __name__ == '__main__': data = InpaintingData(args) print(len(data), len(data.mask_path)) img, mask, filename = data[0] - print(img.size(), mask.size(), filename) \ No newline at end of file + print(img.size(), mask.size(), filename)