augment script

1 files changed, 16 insertions, 0 deletions

diff --git a/options/dataset_options.py b/options/dataset_options.py
index d9c3bf7..29e07c6 100644
--- a/options/dataset_options.py
+++ b/options/dataset_options.py
@@ -293,6 +293,22 @@ class DatasetOptions(BaseOptions):
           help='amount of processed image (clahe, poster, etc) to use in feeder step'
         )
 
+        ### DATA AUGMENTATION
+
+        self.parser.add_argument(
+          '--augment-take',
+          default=100,
+          type=int,
+          help='number of random images to take'
+        )
+
+        self.parser.add_argument(
+          '--augment-make',
+          default=15,
+          type=int,
+          help='number of recursive images to generate'
+        )
+
         ### GRAYSCALE
 
         self.parser.add_argument(