diff options
| -rw-r--r-- | options/dataset_options.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/options/dataset_options.py b/options/dataset_options.py index 8b638e0..1692766 100644 --- a/options/dataset_options.py +++ b/options/dataset_options.py @@ -284,22 +284,22 @@ class DatasetOptions(BaseOptions): self.parser.add_argument( '--hue', - default=0.0, - type=float, + default=0, + type=int, help='hue rotation' ) self.parser.add_argument( '--saturation', - default=0.0, - type=float, + default=0, + type=int, help='saturation' ) self.parser.add_argument( '--luminosity', - default=0.0, - type=float, + default=0, + type=int, help='luminosity' ) |