From b7b305a6dda37e0ac131d42007c4014e82decd2c Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Thu, 7 Jun 2018 02:07:23 +0200 Subject: augment paths --- options/base_options.py | 6 ++++++ options/dataset_options.py | 7 ------- 2 files changed, 6 insertions(+), 7 deletions(-) (limited to 'options') diff --git a/options/base_options.py b/options/base_options.py index f3e93c6..575d0d1 100644 --- a/options/base_options.py +++ b/options/base_options.py @@ -43,6 +43,12 @@ class BaseOptions(): self.parser.add_argument('--init_type', type=str, default='normal', help='network initialization [normal|xavier|kaiming|orthogonal]') self.parser.add_argument('--center_crop', action='store_true', help='center crop instead of random crop') self.parser.add_argument('--poll_delay', type=float, default=0.01, help='time to wait before checking for the next frame') + self.parser.add_argument( + '--module-name', + type=str, + default='pix2pix', + help='module name... basically says where to look for sequences' + ) self.initialized = True diff --git a/options/dataset_options.py b/options/dataset_options.py index 92b63ec..7ac5a8a 100644 --- a/options/dataset_options.py +++ b/options/dataset_options.py @@ -74,13 +74,6 @@ class DatasetOptions(BaseOptions): ## LIVE IMAGE PROCESSING - self.parser.add_argument( - '--module-name', - type=str, - default='pix2pix', - help='module name... basically says where to look for sequences' - ) - self.parser.add_argument( '--send-image', type=str, -- cgit v1.2.3-70-g09d2