def modify_commandline_options(parser, is_train): parser = AlignedDataset.modify_commandline_options(parser, is_train) parser.set_defaults(preprocess_mode='resize_and_crop') load_size = 286 if is_train else 256 parser.set_defaults(load_size=load_size) parser.set_defaults(crop_size=256) parser.set_defaults(display_winsize=256) parser.set_defaults(label_nc=13) parser.set_defaults(contain_dontcare_label=False) parser.add_argument( '--image_dir', type=str, required=True, help='path to the directory that contains photo images') return parser
def modify_commandline_options(parser, is_train=True): AlignedDataset.modify_commandline_options(parser=parser, is_train=is_train) parser.add_argument('--unaligned_dir', help='Unaligned dir') return parser