From 57268c369fe96eff9b6321141227badec659fb91 Mon Sep 17 00:00:00 2001 From: Dominic-ZZ <57563083+Dominic-ZZ@users.noreply.github.com> Date: Fri, 4 Jun 2021 16:41:12 +0800 Subject: [PATCH 1/4] Update test.py --- src/test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/test.py b/src/test.py index 872493b..bcca0e3 100644 --- a/src/test.py +++ b/src/test.py @@ -35,7 +35,7 @@ def main_worker(args, use_gpu=True): for ext in ['.jpg', '.png']: image_paths.extend(glob(os.path.join(args.dir_image, '*'+ext))) image_paths.sort() - mask_paths = sorted(glob(os.path.join(args.dir_mask, '*.png'))) + mask_paths = sorted(glob(os.path.join(args.dir_mask,args.mask_type,'*.png'))) os.makedirs(args.outputs, exist_ok=True) # iteration through datasets From f5dd5958e076150f317df68334aa4d9787091bde Mon Sep 17 00:00:00 2001 From: Dominic-ZZ <57563083+Dominic-ZZ@users.noreply.github.com> Date: Fri, 4 Jun 2021 16:51:29 +0800 Subject: [PATCH 2/4] Update option.py --- src/utils/option.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/utils/option.py b/src/utils/option.py index f71d55c..276f5b0 100644 --- a/src/utils/option.py +++ b/src/utils/option.py @@ -9,8 +9,8 @@ parser.add_argument('--dir_mask', type=str, default='../../dataset', help='mask dataset directory') parser.add_argument('--data_train', type=str, default='places2', help='dataname used for training') -parser.add_argument('--data_test', type=str, default='places2', - help='dataname used for testing') +parser.add_argument('--dir_test', type=str, default='../datasets/test_imgs/', + help='test image dataset directory') parser.add_argument('--image_size', type=int, default=512, help='image size used during training') parser.add_argument('--mask_type', type=str, default='pconv', @@ -93,4 +93,4 @@ losses = list(args.rec_loss.split('+')) args.rec_loss = {} for l in losses: weight, name = l.split('*') - args.rec_loss[name] = float(weight) \ No newline at end of file + args.rec_loss[name] = float(weight) From 69a0b15bc05bbca50f6379e134350ca9985ee9af Mon Sep 17 00:00:00 2001 From: Dominic-ZZ <57563083+Dominic-ZZ@users.noreply.github.com> Date: Fri, 4 Jun 2021 16:52:22 +0800 Subject: [PATCH 3/4] Update test.py use train images to test? --- src/test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/test.py b/src/test.py index bcca0e3..486d000 100644 --- a/src/test.py +++ b/src/test.py @@ -33,7 +33,7 @@ def main_worker(args, use_gpu=True): # prepare dataset image_paths = [] for ext in ['.jpg', '.png']: - image_paths.extend(glob(os.path.join(args.dir_image, '*'+ext))) + image_paths.extend(glob(os.path.join(args.dir_test, '*'+ext))) image_paths.sort() mask_paths = sorted(glob(os.path.join(args.dir_mask,args.mask_type,'*.png'))) os.makedirs(args.outputs, exist_ok=True) 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 4/4] 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)