summaryrefslogtreecommitdiff
path: root/options
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-06-07 02:07:23 +0200
committerJules Laplace <julescarbon@gmail.com>2018-06-07 02:07:23 +0200
commitb7b305a6dda37e0ac131d42007c4014e82decd2c (patch)
tree52b20357630152cea4cedfd9a0d356112fd59196 /options
parenta438c6e88484a68abdd17384e720452ffa2b96bb (diff)
augment paths
Diffstat (limited to 'options')
-rw-r--r--options/base_options.py6
-rw-r--r--options/dataset_options.py7
2 files changed, 6 insertions, 7 deletions
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
@@ -75,13 +75,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,
default='b',