def setup(args):
"""
Create configs and perform basic setups.
"""
cfg = get_cfg()
# automatic OUTPUT dir
cfg.merge_from_file(args.config_file)
config_file_name = args.config_file.split('/')
for i, x in enumerate(config_file_name):
if x == 'configs':
config_file_name[i] = 'logs'
if '.yml' in x:
config_file_name[i] = config_file_name[i][:-4]
cfg.OUTPUT_DIR = '/'.join(config_file_name)
cfg.merge_from_list(args.opts)
cfg.freeze()
if args.eval_only or args.dist_only or args.tsne_only or args.domain_only:
if args.eval_only:
tmp = 'eval'
if args.dist_only:
tmp = 'dist'
if args.tsne_only:
tmp = 'tsne'
if args.domain_only:
tmp = 'domain'
default_setup(cfg, args, tmp=tmp)
else:
default_setup(cfg, args)
return cfg
def setup(args):
"""
Create configs and perform basic setups.
"""
cfg.merge_from_file(args.config_file)
cfg.merge_from_list(args.opts)
cfg.freeze()
default_setup(cfg, args)
return cfg
def setup(args):
"""
Create configs and perform basic setups.
"""
cfg = get_cfg()
add_partialreid_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(args):
"""
Create configs and perform basic setups.
"""
cfg = get_cfg()
add_cross_domain_baseline_config(cfg)
add_memory_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(args):
"""
Create configs and perform basic setups.
"""
cfg = get_cfg()
cfg.merge_from_file(args.config_file)
cfg.merge_from_list(args.opts)
cfg.SPECIFIC_DATASET = args.specific_dataset
if args.specific_dataset is not None and not args.eval_only:
cfg.OUTPUT_DIR = osp.join(cfg.OUTPUT_DIR, args.specific_dataset)
cfg.freeze()
default_setup(cfg, args)
return cfg