def setup(args):
"""
Create configs and perform basic setups.
"""
cfg = get_cfg()
add_sparsercnn_config(cfg)
cfg.merge_from_file(args.config_file)
cfg.merge_from_list(args.opts)
cfg.freeze()
default_setup(cfg, args)
return cfg
def setup_cfg(args):
# load config from file and command-line arguments
cfg = get_cfg()
add_sparsercnn_config(cfg)
cfg.merge_from_file(args.config_file)
cfg.merge_from_list(args.opts)
# Set score_threshold for builtin models
cfg.MODEL.RETINANET.SCORE_THRESH_TEST = args.confidence_threshold
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = args.confidence_threshold
cfg.MODEL.PANOPTIC_FPN.COMBINE.INSTANCES_CONFIDENCE_THRESH = args.confidence_threshold
cfg.freeze()
return cfg
def setup(args):
"""
Create configs and perform basic setups.
"""
cfg = get_cfg()
add_sparsercnn_config(cfg)
cfg.merge_from_file(args.config_file)
cfg.DATASETS.TRAIN = ("bdd100k_train", )
cfg.DATASETS.TEST = ("bdd100k_test", )
cfg.SOLVER.CHECKPOINT_PERIOD = 500
cfg.TEST.EVAL_PERIOD = 500
os.makedirs(cfg.OUTPUT_DIR, exist_ok=True)
cfg.merge_from_list(args.opts)
cfg.freeze()
default_setup(cfg, args)
return cfg