summaryrefslogtreecommitdiff
path: root/options/base_options.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-05-12 18:03:02 +0200
committerJules Laplace <julescarbon@gmail.com>2018-05-12 18:03:02 +0200
commit6e02d55fa01b0532ac9f152765b815a0415310b2 (patch)
treedc79a35368e94e8c3f5ac72c475a542bc5b8985b /options/base_options.py
parent0ad66a09aaf55023d8eaff70094b0798b01adb19 (diff)
parent0e03791f3f7db3f157bed785ba06d21270b98059 (diff)
merge
Diffstat (limited to 'options/base_options.py')
-rw-r--r--options/base_options.py1
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