mirror of https://github.com/PyRetri/PyRetri.git
69 lines
1.3 KiB
YAML
69 lines
1.3 KiB
YAML
datasets:
|
|
batch_size: 16
|
|
|
|
collate_fn:
|
|
name: "CollateFn"
|
|
|
|
folder:
|
|
name: "Folder"
|
|
|
|
transformers:
|
|
names: ["DirectResize", "ToTensor", "Normalize"]
|
|
DirectResize:
|
|
size: (224, 224) #(448, 448) #(224, 224)
|
|
Normalize:
|
|
mean: [0.485, 0.456, 0.406]
|
|
std: [0.229, 0.224, 0.225]
|
|
|
|
model:
|
|
name: "resnet50"
|
|
resnet50:
|
|
load_checkpoint: "/data/places365_model/res50_places365.pt"
|
|
|
|
extract:
|
|
assemble: 0
|
|
|
|
extractor:
|
|
name: "ResSeries"
|
|
ResSeries:
|
|
extract_features: ["pool5"]
|
|
|
|
splitter:
|
|
name: "Identity"
|
|
|
|
aggregators:
|
|
names: ["Crow"]
|
|
|
|
index:
|
|
query_fea_dir: "/data/features/best_features/indoor/query"
|
|
gallery_fea_dir: "/data/features/best_features/indoor/gallery"
|
|
|
|
feature_names: ["pool5_Crow"]
|
|
|
|
dim_processors:
|
|
names: ["L2Normalize", "PCA", "L2Normalize"]
|
|
PCA:
|
|
proj_dim: 512
|
|
whiten: False
|
|
train_fea_dir: "/data/features/best_features/indoor/gallery"
|
|
l2: True
|
|
|
|
feature_enhancer:
|
|
name: "DBA"
|
|
DBA:
|
|
enhance_k: 10 # number of the nearest points to be calculated.
|
|
|
|
metric:
|
|
name: "KNN"
|
|
|
|
re_ranker:
|
|
name: "QE"
|
|
QE:
|
|
qe_times: 1 # number of query expansion times.
|
|
qe_k: 10 # number of the neighbors to be combined.
|
|
|
|
evaluate:
|
|
evaluator:
|
|
name: "OverAll"
|
|
|