diff options
Diffstat (limited to 'options/dataset_options.py')
| -rw-r--r-- | options/dataset_options.py | 14 |
1 files changed, 10 insertions, 4 deletions
diff --git a/options/dataset_options.py b/options/dataset_options.py index 2547d11..057b940 100644 --- a/options/dataset_options.py +++ b/options/dataset_options.py @@ -123,12 +123,18 @@ class DatasetOptions(BaseOptions): self.parser.add_argument( '--blur', - default=0, - type=int, + action='store_true', help='blur image by N' ) self.parser.add_argument( + '--blur-radius', + default=3, + type=int, + help='blur sigma' + ) + + self.parser.add_argument( '--blur-sigma', default=0.0, type=float, @@ -144,14 +150,14 @@ class DatasetOptions(BaseOptions): ) self.parser.add_argument( - '--canny_lo', + '--canny-lo', default=100, type=int, help='canny low threshold' ) self.parser.add_argument( - '--canny_high', + '--canny-hi', default=200, type=int, help='canny high threshold' |