__C.DATASET.CITYSCAPES_AUG_DIR = ''
# Mapillary Dataset Dir Location
__C.DATASET.MAPILLARY_DIR = './data/mapillary'
# Kitti Dataset Dir Location
__C.DATASET.KITTI_DIR = './data/kitti'
# SDC Augmented Kitti Dataset Dir Location, We do not use it.
__C.DATASET.KITTI_AUG_DIR = ''
# Camvid Dataset Dir Location
__C.DATASET.CAMVID_DIR = './data/camVid'
# BDD Dataset Dir Location
__C.DATASET.BDD_DIR = './data/BDD/seg'
#Number of splits to support
__C.DATASET.CV_SPLITS = 3
__C.MODEL = AttrDict()
__C.MODEL.BN = 'regularnorm'
__C.MODEL.BNFUNC = None
def assert_and_infer_cfg(args, make_immutable=True, train_mode=True):
"""Call this function in your script after you have finished setting all cfg
values that are necessary (e.g., merging a config from a file, merging
command line config options, etc.). By default, this function will also
mark the global cfg as immutable to prevent changing the global cfg settings
during script execution (which can lead to hard to debug errors or code
that's harder to understand than is necessary).
"""
if hasattr(args, 'syncbn') and args.syncbn:
if args.apex:
cfg.DATASET.CITYSCAPES_DIR = '/private/home/yangwangx/datasets/cityscapes'
# SDC Augmented Cityscapes Dir Location
cfg.DATASET.CITYSCAPES_AUG_DIR = '/private/home/yangwangx/datasets/cityscapes/cs_aug'
# Mapillary Dataset Dir Location
cfg.DATASET.MAPILLARY_DIR = ''
# Kitti Dataset Dir Location
cfg.DATASET.KITTI_DIR = ''
# SDC Augmented Kitti Dataset Dir Location
cfg.DATASET.KITTI_AUG_DIR = ''
# Camvid Dataset Dir Location
cfg.DATASET.CAMVID_DIR = ''
# Number of splits to support
cfg.DATASET.CV_SPLITS = 3
# Attribute Dictionary for Model
cfg.MODEL = AttrDict()
cfg.MODEL.BN = 'regularnorm'
cfg.MODEL.BNFUNC = None
def assert_and_infer_cfg(args, make_immutable=True, train_mode=True):
"""Call this function in your script after you have finished setting all cfg
values that are necessary (e.g., merging a config from a file, merging
command line config options, etc.). By default, this function will also
mark the global cfg as immutable to prevent changing the global cfg settings
during script execution (which can lead to hard to debug errors or code
that's harder to understand than is necessary).
"""
if hasattr(args, 'syncbn') and args.syncbn:
if args.apex:
