mirror of https://github.com/PyRetri/PyRetri.git
159 lines
2.8 KiB
Python
159 lines
2.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from utils.search_modules import SearchModules
|
|
from pyretri.config import get_defaults_cfg
|
|
|
|
indexes = SearchModules()
|
|
evaluates = SearchModules()
|
|
|
|
indexes.add(
|
|
"pca_wo_whiten",
|
|
{
|
|
"gallery_fea_dir": "",
|
|
"query_fea_dir": "",
|
|
|
|
"feature_names": [],
|
|
|
|
"dim_processors": {
|
|
"names": ["L2Normalize", "PCA", "L2Normalize"],
|
|
"PCA": {
|
|
"whiten": False,
|
|
"train_fea_dir": "",
|
|
"proj_dim": 512,
|
|
"l2": True,
|
|
}
|
|
},
|
|
|
|
"feature_enhancer": {
|
|
"name": "Identity"
|
|
},
|
|
|
|
"metric": {
|
|
"name": "KNN"
|
|
},
|
|
|
|
"re_ranker": {
|
|
"name": "Identity"
|
|
}
|
|
}
|
|
)
|
|
|
|
indexes.add(
|
|
"pca_whiten",
|
|
{
|
|
"gallery_fea_dir": "",
|
|
"query_fea_dir": "",
|
|
|
|
"feature_names": [],
|
|
|
|
"dim_processors": {
|
|
"names": ["L2Normalize", "PCA", "L2Normalize"],
|
|
"PCA": {
|
|
"whiten": True,
|
|
"train_fea_dir": "",
|
|
"proj_dim": 512,
|
|
"l2": True,
|
|
}
|
|
},
|
|
|
|
"feature_enhancer": {
|
|
"name": "Identity"
|
|
},
|
|
|
|
"metric": {
|
|
"name": "KNN"
|
|
},
|
|
|
|
"re_ranker": {
|
|
"name": "Identity"
|
|
}
|
|
}
|
|
)
|
|
|
|
indexes.add(
|
|
"svd_wo_whiten",
|
|
{
|
|
"gallery_fea_dir": "",
|
|
"query_fea_dir": "",
|
|
|
|
"feature_names": [],
|
|
|
|
"dim_processors": {
|
|
"names": ["L2Normalize", "SVD", "L2Normalize"],
|
|
"SVD": {
|
|
"whiten": False,
|
|
"train_fea_dir": "",
|
|
"proj_dim": 511,
|
|
"l2": True,
|
|
}
|
|
},
|
|
|
|
"feature_enhancer": {
|
|
"name": "Identity"
|
|
},
|
|
|
|
"metric": {
|
|
"name": "KNN"
|
|
},
|
|
|
|
"re_ranker": {
|
|
"name": "Identity"
|
|
}
|
|
}
|
|
)
|
|
|
|
indexes.add(
|
|
"svd_whiten",
|
|
{
|
|
"gallery_fea_dir": "",
|
|
"query_fea_dir": "",
|
|
|
|
"feature_names": [],
|
|
|
|
"dim_processors": {
|
|
"names": ["L2Normalize", "SVD", "L2Normalize"],
|
|
"SVD": {
|
|
"whiten": True,
|
|
"train_fea_dir": "",
|
|
"proj_dim": 511,
|
|
"l2": True,
|
|
}
|
|
},
|
|
|
|
"feature_enhancer": {
|
|
"name": "Identity"
|
|
},
|
|
|
|
"metric": {
|
|
"name": "KNN"
|
|
},
|
|
|
|
"re_ranker": {
|
|
"name": "Identity"
|
|
}
|
|
}
|
|
)
|
|
|
|
evaluates.add(
|
|
"overall",
|
|
{
|
|
"evaluator": {
|
|
"name": "OverAll"
|
|
}
|
|
}
|
|
)
|
|
|
|
evaluates.add(
|
|
"oxford_overall",
|
|
{
|
|
"evaluator": {
|
|
"name": "OxfordOverAll"
|
|
}
|
|
}
|
|
)
|
|
|
|
cfg = get_defaults_cfg()
|
|
|
|
indexes.check_valid(cfg["index"])
|
|
evaluates.check_valid(cfg["evaluate"])
|