| from . import matchers | |
| from . import readers | |
| from . import evaluators | |
| from . import extractors | |
| def load_component(compo_name,model_name,config): | |
| if compo_name=='extractor': | |
| component=load_extractor(model_name,config) | |
| elif compo_name=='reader': | |
| component=load_reader(model_name,config) | |
| elif compo_name=='matcher': | |
| component=load_matcher(model_name,config) | |
| elif compo_name=='evaluator': | |
| component=load_evaluator(model_name,config) | |
| else: | |
| raise NotImplementedError | |
| return component | |
| def load_extractor(model_name,config): | |
| if model_name=='root': | |
| extractor =extractors.ExtractSIFT(config) | |
| elif model_name=='sp': | |
| extractor=extractors.ExtractSuperpoint(config) | |
| else: | |
| raise NotImplementedError | |
| return extractor | |
| def load_matcher(model_name,config): | |
| if model_name=='SGM': | |
| matcher=matchers.GNN_Matcher(config,'SGM') | |
| elif model_name=='SG': | |
| matcher=matchers.GNN_Matcher(config,'SG') | |
| elif model_name=='NN': | |
| matcher=matchers.NN_Matcher(config) | |
| else: | |
| raise NotImplementedError | |
| return matcher | |
| def load_reader(model_name,config): | |
| if model_name=='standard': | |
| reader=readers.standard_reader(config) | |
| else: | |
| raise NotImplementedError | |
| return reader | |
| def load_evaluator(model_name,config): | |
| if model_name=='AUC': | |
| evaluator=evaluators.auc_eval(config) | |
| elif model_name=='FM': | |
| evaluator=evaluators.FMbench_eval(config) | |
| return evaluator | |