From 1da9734a690bea7e77df1d207b4673afff3076c9 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Sun, 13 May 2018 10:40:00 +0200
Subject: more dataset stuff..

---
 options/dataset_options.py      | 14 ++++++++++----
 scripts/builders/mogrify-dir.py |  4 ++--
 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,11 +123,17 @@ 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,
@@ -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)
 
-- 
cgit v1.2.3-70-g09d2