transform options

1 files changed, 22 insertions, 0 deletions

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(