diff options
Diffstat (limited to 'options')
| -rw-r--r-- | options/base_options.py | 2 | ||||
| -rw-r--r-- | options/dataset_options.py | 17 |
2 files changed, 17 insertions, 2 deletions
diff --git a/options/base_options.py b/options/base_options.py index d569928..105292f 100644 --- a/options/base_options.py +++ b/options/base_options.py @@ -48,7 +48,7 @@ class BaseOptions(): def parse(self): if not self.initialized: self.initialize() - self.opt = self.parser.parse_args() + self.opt = self.parser.parse_known_args() self.opt.isTrain = self.isTrain # train or test return self.opt diff --git a/options/dataset_options.py b/options/dataset_options.py index 057b940..5eca374 100644 --- a/options/dataset_options.py +++ b/options/dataset_options.py @@ -59,6 +59,21 @@ class DatasetOptions(BaseOptions): ## IMAGE FILTERS + ### RECURSION + + self.parser.add_argument( + '--recursive', + action='store_true', + help='recurse on previous output' + ) + + self.parser.add_argument( + '--recursive-frac', + default=0.5, + type=float, + help='amount of previous step to use in recursion' + ) + ### GRAYSCALE self.parser.add_argument( @@ -166,5 +181,5 @@ class DatasetOptions(BaseOptions): def parse(self): if not self.initialized: self.initialize() - self.opt = self.parser.parse_args() + self.opt = self.parser.parse_known_args() return self.opt
\ No newline at end of file |