more dataset stuff..

2 files changed, 12 insertions, 6 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'
diff --git a/scripts/builders/mogrify-dir.py b/scripts/builders/mogrify-dir.py
index b7370f5..5209dcb 100755
--- a/scripts/builders/mogrify-dir.py
+++ b/scripts/builders/mogrify-dir.py
@@ -73,8 +73,8 @@ for i, fn in enumerate(images):
     img = cv2.pyrMeanShiftFiltering(img, opt.spatial_window, opt.color_window)
   if opt.grayscale is True:
     img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
-  if opt.blur != 0:
-    img = cv2.GaussianBlur(img, (opt.blur, opt.blur), opt.blur_sigma)
+  if opt.blur is True:
+    img = cv2.GaussianBlur(img, (opt.blur_radius, opt.blur_radius), opt.blur_sigma)
   if opt.canny is True:
     img = cv2.Canny(img, opt.canny_lo, opt.canny_hi)