| 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 | |