Merge pull request #102 from sony/feature/20190924-deeplab-args

DeepLabV3+: Fix default image size arguments

semantic-segmentation/deeplabv3plus/args.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 
-def get_args(monitor_path='tmp.monitor', max_iter=10000, model_save_path=None, learning_rate=1e-3, batch_size=128, weight_decay=1e-4, description=None):
+def get_args(monitor_path='tmp.monitor', max_iter=10000, model_save_path=None, learning_rate=1e-3, batch_size=2, weight_decay=1e-4, description=None):
     """
     Get command line arguments.
 
@@ -68,7 +68,7 @@ def get_args(monitor_path='tmp.monitor', max_iter=10000, model_save_path=None, l
     parser.add_argument("--val-iter", "-j", type=int, default=10,
                         help='Each validation runs `val_iter mini-batch iteration.')
     parser.add_argument("--accum-grad",
-                        type=int, default=32,
+                        type=int, default=1,
                         help='Weight decay factor of SGD update.')
     parser.add_argument("--weight-decay", "-w",
                         type=float, default=weight_decay,
@@ -86,18 +86,11 @@ def get_args(monitor_path='tmp.monitor', max_iter=10000, model_save_path=None, l
     parser.add_argument("--pretrained-model-path",
                         type=str, default=model_save_path,
                         help='Path the pretrained model parameters saved.')
-    parser.add_argument("--net", "-n", type=str,
-                        default='lenet',
-                        help="Neural network architecure type (used only in classification*.py).\n  classification.py: ('lenet'|'resnet'),  classification_bnn.py: ('bincon'|'binnet'|'bwn'|'bwn'|'bincon_resnet'|'binnet_resnet'|'bwn_resnet')")
     parser.add_argument('--context', '-c', type=str,
                         default='cpu', help="Extension modules. ex) 'cpu', 'cudnn'.")
-    parser.add_argument('--augment-train', action='store_true',
-                        default=False, help="Enable data augmentation of training data.")
-    parser.add_argument('--augment-test', action='store_true',
-                        default=False, help="Enable data augmentation of testing data.")
     parser.add_argument('--channel', default=1, type=int)
-    parser.add_argument('--image-width', default=28, type=int)
-    parser.add_argument('--image-height', default=28, type=int)
+    parser.add_argument('--image-width', default=513, type=int)
+    parser.add_argument('--image-height', default=513, type=int)
     parser.add_argument('--dataset-path', type=str)
     parser.add_argument("--model-load-path", "-T",
                         type=str, default=model_save_path,