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] 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)