From fd3198c0c799e7943f7f27758e97670535c94979 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Sun, 2 Sep 2018 22:09:49 +0200 Subject: augment script --- options/dataset_options.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) (limited to 'options/dataset_options.py') 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( -- cgit v1.2.3-70-g09d2