diff options
| author | jules@lens <julescarbon@gmail.com> | 2018-05-08 21:30:57 +0200 |
|---|---|---|
| committer | jules@lens <julescarbon@gmail.com> | 2018-05-08 21:30:57 +0200 |
| commit | 2f4375e03ba13625497a5c4bae381a6ad0c6f937 (patch) | |
| tree | 5126460a7eb13d06493038b7a6be9a8d4a797827 /options | |
| parent | 5979723a35c577ffe70c1107415199444fb5c7fe (diff) | |
new dataset stuff
Diffstat (limited to 'options')
| -rw-r--r-- | options/base_options.py | 1 | ||||
| -rw-r--r-- | options/test_options.py | 1 |
2 files changed, 2 insertions, 0 deletions
diff --git a/options/base_options.py b/options/base_options.py index 5bdd85e..b04216d 100644 --- a/options/base_options.py +++ b/options/base_options.py @@ -41,6 +41,7 @@ class BaseOptions(): self.parser.add_argument('--resize_or_crop', type=str, default='resize_and_crop', help='scaling and cropping of images at load time [resize_and_crop|crop|scale_width|scale_width_and_crop]') self.parser.add_argument('--no_flip', action='store_true', help='if specified, do not flip the images for data augmentation') 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.initialized = True diff --git a/options/test_options.py b/options/test_options.py index f8c3a31..418d5e6 100644 --- a/options/test_options.py +++ b/options/test_options.py @@ -12,4 +12,5 @@ class TestOptions(BaseOptions): self.parser.add_argument('--phase', type=str, default='test', help='train, val, test, etc') self.parser.add_argument('--which_epoch', type=str, default='latest', help='which epoch to load? set to latest to use latest cached model') self.parser.add_argument('--how_many', type=int, default=50, help='how many test images to run') + self.parser.add_argument('--mov', type=str, help='filename of movie to export') self.isTrain = False |