test effects script

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