mirror of https://github.com/PyRetri/PyRetri.git
44 lines
1.0 KiB
Python
44 lines
1.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from yacs.config import CfgNode
|
|
|
|
from .registry import ENHANCERS, METRICS, DIMPROCESSORS, RERANKERS
|
|
|
|
from ..utils import get_config_from_registry
|
|
|
|
|
|
def get_enhancer_cfg() -> CfgNode:
|
|
cfg = get_config_from_registry(ENHANCERS)
|
|
cfg["name"] = "unknown"
|
|
return cfg
|
|
|
|
|
|
def get_metric_cfg() -> CfgNode:
|
|
cfg = get_config_from_registry(METRICS)
|
|
cfg["name"] = "unknown"
|
|
return cfg
|
|
|
|
|
|
def get_processors_cfg() -> CfgNode:
|
|
cfg = get_config_from_registry(DIMPROCESSORS)
|
|
cfg["names"] = ["unknown"]
|
|
return cfg
|
|
|
|
|
|
def get_ranker_cfg() -> CfgNode:
|
|
cfg = get_config_from_registry(RERANKERS)
|
|
cfg["name"] = "unknown"
|
|
return cfg
|
|
|
|
|
|
def get_index_cfg() -> CfgNode:
|
|
cfg = CfgNode()
|
|
cfg["query_fea_dir"] = "unknown"
|
|
cfg["gallery_fea_dir"] = "unknown"
|
|
cfg["feature_names"] = ["all"]
|
|
cfg["dim_processors"] = get_processors_cfg()
|
|
cfg["feature_enhancer"] = get_enhancer_cfg()
|
|
cfg["metric"] = get_metric_cfg()
|
|
cfg["re_ranker"] = get_ranker_cfg()
|
|
return cfg
|