flags for hsl transform

1 files changed, 29 insertions, 0 deletions

diff --git a/options/dataset_options.py b/options/dataset_options.py
index bffd60c..8b638e0 100644
--- a/options/dataset_options.py
+++ b/options/dataset_options.py
@@ -274,6 +274,35 @@ class DatasetOptions(BaseOptions):
           help='convert image to grayscale first'
         )
 
+        ### HSL
+
+        self.parser.add_argument(
+          '--hsl',
+          action='store_true',
+          help='adjust HSL'
+        )
+
+        self.parser.add_argument(
+          '--hue',
+          default=0.0,
+          type=float,
+          help='hue rotation'
+        )
+
+        self.parser.add_argument(
+          '--saturation',
+          default=0.0,
+          type=float,
+          help='saturation'
+        )
+
+        self.parser.add_argument(
+          '--luminosity',
+          default=0.0,
+          type=float,
+          help='luminosity'
+        )
+
         ### CLAHE
 
         self.parser.add_argument(