diff options
Diffstat (limited to 'options/dataset_options.py')
| -rw-r--r-- | options/dataset_options.py | 17 |
1 files changed, 16 insertions, 1 deletions
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 |