Beispiel #1
0
    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
Beispiel #2
0
    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