more dataset stuff..

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'