From 1d06b8ee449ed6d80f71892caacad7d4fc764f82 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 16 Jul 2018 17:58:25 +0200 Subject: transform options --- options/dataset_options.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) (limited to 'options/dataset_options.py') diff --git a/options/dataset_options.py b/options/dataset_options.py index 1692766..74e6bdc 100644 --- a/options/dataset_options.py +++ b/options/dataset_options.py @@ -274,6 +274,28 @@ class DatasetOptions(BaseOptions): help='convert image to grayscale first' ) + ### TRANSFORM + + self.parser.add_argument( + '--transform', + action='store_true', + help='transform the image' + ) + + self.parser.add_argument( + '--rotate', + default=0.0, + type=float, + help='rotation' + ) + + self.parser.add_argument( + '--scale', + default=1.0, + type=float, + help='scaling' + ) + ### HSL self.parser.add_argument( -- cgit v1.2.3-70-g09d2