def load_train_config(args):
if args.config_file:
cfg.merge_from_file(args.config_file)
cfg.merge_from_list(args.opts)
if args.log_step != -1:
cfg.TRAIN.LOG_STEP = args.log_step
if args.save_step != -1:
cfg.TRAIN.SAVE_STEP = args.save_step
if args.eval_step != -1:
cfg.TRAIN.EVAL_STEP = args.eval_step
if args.resume:
cfg.TRAIN.RESUME = True
if not args.use_tensorboard:
cfg.TRAIN.USE_TENSORBOARD = False
if args.gpus != -1:
cfg.NUM_GPUS = args.gpus
if args.nodes != -1:
cfg.NUM_NODES = args.nodes
if args.nr != -1:
cfg.RANK_ID = args.nr
cfg.freeze()
if not os.path.exists(cfg.OUTPUT_DIR):
os.makedirs(cfg.OUTPUT_DIR)
return cfg
def load_config(args):
if args.config_file:
cfg.merge_from_file(args.config_file)
if args.pretrained:
cfg.MODEL.RECOGNIZER.PRELOADED = args.pretrained
if args.output_dir:
cfg.OUTPUT_DIR = args.output_dir
if args.log_step != -1:
cfg.TRAIN.LOG_STEP = args.log_step
if args.save_step != -1:
cfg.TRAIN.SAVE_STEP = args.save_step
if args.eval_step != -1:
cfg.TRAIN.EVAL_STEP = args.eval_step
if args.resume:
cfg.TRAIN.RESUME = True
if not args.use_tensorboard:
cfg.TRAIN.USE_TENSORBOARD = False
if args.gpus != -1:
cfg.NUM_GPUS = args.gpus
if args.nodes != -1:
cfg.NUM_NODES = args.nodes
if args.nr != -1:
cfg.RANK_ID = args.nr
if args.init_method:
cfg.INIT_METHOD = args.init_method
cfg.merge_from_list(args.opts)
cfg.freeze()
return cfg
def load_test_config(args):
if not os.path.isfile(args.config_file) or not os.path.isfile(
args.pretrained):
raise ValueError('需要输入配置文件和预训练模型路径')
cfg.merge_from_file(args.config_file)
cfg.MODEL.RECOGNIZER.PRELOADED = args.pretrained
cfg.OUTPUT_DIR = args.output
if args.gpus != -1:
cfg.NUM_GPUS = args.gpus
if args.nodes != -1:
cfg.NODES = args.nodes
if args.nr != -1:
cfg.RANK = args.nr
cfg.freeze()
if not os.path.exists(cfg.OUTPUT_DIR):
os.makedirs(cfg.OUTPUT_DIR)
return cfg
def main():
args = parse_args()
print(args)
cfg.merge_from_file(args.config_file)
cfg.merge_from_list(args.opts)
if args.ckpt is not None:
cfg.MODEL.RECOGNIZER.PRELOADED = args.ckpt
cfg.freeze()
if args.config_file:
print("Loaded configuration file {}".format(args.config_file))
with open(args.config_file, "r") as cf:
config_str = "\n" + cf.read()
print(config_str)
# print("Running with config:\n{}".format(cfg))
run_demo(cfg=cfg,
images_dir=args.images_dir,
rotate_dir=args.rotate_dir,
output_dir=args.output_dir)