提交 6c5ff1e9 编写于 作者: C cuicheng01

Update inshop inference config

上级 958727f3
Global: Global:
rec_inference_model_dir: "./inshop/rec/" rec_inference_model_dir: "./models/product_ResNet50_vd_Inshop_v1.0_infer"
batch_size: 1 batch_size: 1
use_gpu: True use_gpu: True
enable_mkldnn: True enable_mkldnn: True
...@@ -26,9 +26,9 @@ RecPostProcess: null ...@@ -26,9 +26,9 @@ RecPostProcess: null
# indexing engine config # indexing engine config
IndexProcess: IndexProcess:
index_path: "./inshop/inshop_index/" index_path: "./dataset/product_demo_data_v1.0/index"
image_root: "./inshop/dataset/" image_root: "./dataset/product_demo_data_v1.0"
data_file: "./inshop/inshop_gallery_demo.txt" data_file: "./dataset/product_demo_data_v1.0/data_file.txt"
delimiter: " " delimiter: " "
dist_type: "IP" dist_type: "IP"
pq_size: 100 pq_size: 100
......
Global: Global:
infer_imgs: "./inshop/demo/01_3_back.jpg" infer_imgs: "./dataset/product_demo_data_v1.0/query"
det_inference_model_dir: "./inshop/det/" det_inference_model_dir: "./models/ppyolov2_r50vd_dcn_mainbody_v1.0_infer"
rec_inference_model_dir: "./inshop/rec/" rec_inference_model_dir: "./models/product_ResNet50_vd_Inshop_v1.0_infer"
batch_size: 1 batch_size: 1
image_shape: [3, 640, 640] image_shape: [3, 640, 640]
threshold: 0.0 threshold: 0.0
max_det_results: 3 max_det_results: 1
labe_list: labe_list:
- foreground - foreground
...@@ -48,7 +48,7 @@ RecPostProcess: null ...@@ -48,7 +48,7 @@ RecPostProcess: null
# indexing engine config # indexing engine config
IndexProcess: IndexProcess:
index_path: "./inshop/inshop_index" index_path: "./dataset/product_demo_data_v1.0/index"
search_budget: 100 search_budget: 100
return_k: 10 return_k: 10
dist_type: "IP" dist_type: "IP"
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册