diff options
| -rw-r--r-- | options/dataset_options.py | 22 |
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( |