diff options
| -rw-r--r-- | options/dataset_options.py | 14 | ||||
| -rwxr-xr-x | 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,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) |