fix seg training.

main.py

@@ -63,8 +63,8 @@ def str2bool(v):
                         dest='network:checkpoints_name', help='The name of checkpoint model.')
     parser.add_argument('--backbone', default=None, type=str,
                         dest='network:backbone', help='The base network of model.')
-    parser.add_argument('--bn_type', default=None, type=str,
-                        dest='network:bn_type', help='The BN type of the network.')
+    parser.add_argument('--norm_type', default=None, type=str,
+                        dest='network:norm_type', help='The BN type of the network.')
     parser.add_argument('--multi_grid', default=None, nargs='+', type=int,
                         dest='network:multi_grid', help='The multi_grid for resnet backbone.')
     parser.add_argument('--pretrained', type=str, default=None,
@@ -148,8 +148,8 @@ def str2bool(v):
     if configer.get('gpu') is not None:
         os.environ["CUDA_VISIBLE_DEVICES"] = ','.join(str(gpu_id) for gpu_id in configer.get('gpu'))
 
-    if configer.get('network', 'bn_type') is None:
-        configer.update(['network', 'bn_type'], 'torchbn')
+    if configer.get('network', 'norm_type') is None:
+        configer.update(['network', 'norm_type'], 'batchnorm')
 
     project_dir = os.path.dirname(os.path.realpath(__file__))
     configer.add(['project_dir'], project_dir)

methods/tools/controller.py

@@ -23,9 +23,9 @@ def init(runner):
         runner.runner_state['max_performance'] = 0
         runner.runner_state['min_val_loss'] = 0
         if runner.configer.get('phase') == 'train':
-            assert len(runner.configer.get('gpu')) > 1 or runner.configer.get('network', 'bn_type') == 'torchbn'
+            assert len(runner.configer.get('gpu')) > 1 or runner.configer.get('network', 'norm_type') == 'batchnorm'
 
-        Log.info('BN Type is {}.'.format(runner.configer.get('network', 'bn_type')))
+        Log.info('BN Type is {}.'.format(runner.configer.get('network', 'norm_type')))
 
     @staticmethod
     def train(runner):

utils/helpers/mask_helper.py

@@ -10,7 +10,7 @@
 try:
     import pycocotools.mask as mask_util
 except ImportError:
-    Log.error('pycocotools ImportError.')
+    print('pycocotools ImportError.')
 
 
 class MaskHelper(object):