diff options
Diffstat (limited to 'options/base_options.py')
| -rw-r--r-- | options/base_options.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/options/base_options.py b/options/base_options.py index 150174b..d569928 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 |