summaryrefslogtreecommitdiff
path: root/options/dataset_options.py
diff options
context:
space:
mode:
Diffstat (limited to 'options/dataset_options.py')
-rw-r--r--options/dataset_options.py29
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(