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 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

(limited to 'options')

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'
-- 
cgit v1.2.3-70-g09d2