# L2 regularization hyperparameter
_C.weight_decay = 0.0001
# momentum with SGD
_C.momentum = 0.9
# maximum number of iterations, 1x: 180000, 2x: 360000
_C.max_iter = 180000
"""
ENV options
"""
# support both CPU and GPU
_C.use_gpu = True
# whether use parallel
_C.parallel = True
# support pyreader
_C.use_pyreader = True
# pixel mean values
_C.pixel_means = [102.9801, 115.9465, 122.7717]
def merge_cfg_from_args(args, mode):
"""Merge config keys,values in args into the global config."""
if mode == 'train':
sub_d = _C.TRAIN
else:
sub_d = _C.TEST
for k, v in sorted(six.iteritems(vars(args))):
# L2 regularization hyperparameter _C.weight_decay = 0.0001 # momentum with SGD _C.momentum = 0.9 # # ENV options # # support both CPU and GPU _C.use_gpu = True # Whether use parallel _C.parallel = False # Class number _C.class_num = 81 # support pyreader _C.use_pyreader = True # pixel mean values _C.pixel_means = [102.9801, 115.9465, 122.7717] # clip box to prevent overflowing _C.bbox_clip = np.log(1000. / 16.) def merge_cfg_from_args(args, mode):