diff options
| -rw-r--r-- | options/base_options.py | 3 | ||||
| -rw-r--r-- | options/dataset_options.py | 7 | ||||
| -rw-r--r-- | test-mogrify.py | 2 |
3 files changed, 8 insertions, 4 deletions
diff --git a/options/base_options.py b/options/base_options.py index 814ea3e..f741927 100644 --- a/options/base_options.py +++ b/options/base_options.py @@ -48,7 +48,8 @@ class BaseOptions(): def parse(self): if not self.initialized: self.initialize() - self.opt = self.parser.parse_known_args() + self.opt, unknown = self.parser.parse_known_args() + self.opt.unknown = unknown self.opt.isTrain = self.isTrain # train or test str_ids = self.opt.gpu_ids.split(',') diff --git a/options/dataset_options.py b/options/dataset_options.py index 5eca374..fcb9654 100644 --- a/options/dataset_options.py +++ b/options/dataset_options.py @@ -178,8 +178,11 @@ class DatasetOptions(BaseOptions): help='canny high threshold' ) - def parse(self): + def parse(self, args=None): if not self.initialized: self.initialize() - self.opt = self.parser.parse_known_args() + if args is not None: + self.opt, unknown = self.parser.parse_known_args(args) + else: + self.opt, unknown = self.parser.parse_known_args() return self.opt
\ No newline at end of file diff --git a/test-mogrify.py b/test-mogrify.py index 769a431..5d90ff2 100644 --- a/test-mogrify.py +++ b/test-mogrify.py @@ -27,7 +27,7 @@ frac_b = 1 - frac_a if __name__ == '__main__': opt = TestOptions().parse() - data_opt = DatasetOptions().parse() + data_opt = DatasetOptions().parse(opt.unknown) opt.nThreads = 1 # test code only supports nThreads = 1 opt.batchSize = 1 # test code only supports batchSize = 1 opt.serial_batches = True # no shuffle |