Update option.py

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)
+    args.rec_loss[name] = float(weight)