def build_siamfcpp_tester(task_cfg): # build model model = model_builder.build("track", task_cfg.model) # build pipeline pipeline = pipeline_builder.build("track", task_cfg.pipeline, model) # build tester testers = tester_builder("track", task_cfg.tester, "tester", pipeline) return testers
def build_siamfcpp_tester(task_cfg): parsed_args = parser.parse_args() # build model model = model_builder.build("track", task_cfg.model) # build pipeline pipeline = pipeline_builder.build("track", task_cfg.pipeline, model) # build tester testers = tester_builder(parsed_args.video, "track", task_cfg.tester, "tester", pipeline) return testers
def build_sat_tester(task_cfg): # build model tracker_model = model_builder.build("track", task_cfg.tracker_model) tracker = pipeline_builder.build("track", task_cfg.tracker_pipeline, model=tracker_model) segmenter = model_builder.build('vos', task_cfg.segmenter) # build pipeline pipeline = pipeline_builder.build('vos', task_cfg.pipeline, segmenter=segmenter, tracker=tracker) # build tester testers = tester_builder('vos', task_cfg.tester, "tester", pipeline) return testers
help='experiment configuration') return parser if __name__ == '__main__': # parsing parser = make_parser() parsed_args = parser.parse_args() # experiment config exp_cfg_path = osp.realpath(parsed_args.config) root_cfg.merge_from_file(exp_cfg_path) logger.info("Load experiment config. at: %s" % exp_cfg_path) # resolve config task, task_cfg = specify_task(root_cfg) task_cfg.freeze() # build model model = model_builder.build_model(task, task_cfg.model) # build pipeline pipeline = pipeline_builder.build_pipeline('track', task_cfg.pipeline) pipeline.set_model(model) # build tester testers = tester_builder(task, task_cfg, "tester", pipeline=pipeline) # start engine for tester in testers: tester.test()
# resolve config root_cfg = complete_path_wt_root_in_cfg(root_cfg, ROOT_PATH) root_cfg = root_cfg.test task, task_cfg = specify_task(root_cfg) task_cfg.freeze() if task == 'track': # build model model = model_builder.build(task, task_cfg.model) # build pipeline pipeline = pipeline_builder.build('track', task_cfg.pipeline, model=model) # build tester testers = tester_builder(task, task_cfg.tester, "tester", pipeline) elif task == 'vos': # build model tracker = model_builder.build("track_vos", task_cfg.tracker) segmenter = model_builder.build('vos', task_cfg.segmenter) # build pipeline pipeline = pipeline_builder.build('vos', task_cfg.pipeline, segmenter=segmenter, tracker=tracker) # build tester testers = tester_builder(task, task_cfg.tester, "tester", pipeline) for tester in testers: tester.test()