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