diff options
Diffstat (limited to 'options/dataset_options.py')
| -rw-r--r-- | options/dataset_options.py | 16 |
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( |