PaddleClas/deploy/python/ppshitu_v2/configs/test_search_config.yaml

16 lines
305 B
YAML

Global:
Engine: POPEngine
infer_imgs: "./vector.npy"
Modules:
- name:
type: AlgoMod
processors:
- name: Searcher
type: searcher
index_dir: "./index"
dist_type: "IP"
embedding_size: 512
batch_size: 32
return_k: 5
score_thres: 0.5