2021-06-09 02:40:53 +00:00
|
|
|
Global:
|
2021-06-16 09:29:08 +00:00
|
|
|
rec_inference_model_dir: "./models/product_ResNet50_vd_aliproduct_v1.0_infer"
|
2021-06-09 02:40:53 +00:00
|
|
|
batch_size: 1
|
|
|
|
|
use_gpu: True
|
|
|
|
|
enable_mkldnn: True
|
|
|
|
|
cpu_num_threads: 100
|
|
|
|
|
enable_benchmark: True
|
|
|
|
|
use_fp16: False
|
|
|
|
|
ir_optim: True
|
|
|
|
|
use_tensorrt: False
|
|
|
|
|
gpu_mem: 8000
|
|
|
|
|
enable_profile: False
|
|
|
|
|
|
|
|
|
|
RecPreProcess:
|
|
|
|
|
transform_ops:
|
|
|
|
|
- ResizeImage:
|
|
|
|
|
size: 224
|
|
|
|
|
- NormalizeImage:
|
|
|
|
|
scale: 0.00392157
|
|
|
|
|
mean: [0.485, 0.456, 0.406]
|
|
|
|
|
std: [0.229, 0.224, 0.225]
|
|
|
|
|
order: ''
|
|
|
|
|
- ToCHWImage:
|
|
|
|
|
|
|
|
|
|
RecPostProcess: null
|
|
|
|
|
|
|
|
|
|
# indexing engine config
|
|
|
|
|
IndexProcess:
|
2021-06-10 05:18:55 +00:00
|
|
|
index_path: "./dataset/product_demo_data_v1.0/index"
|
|
|
|
|
image_root: "./dataset/product_demo_data_v1.0"
|
|
|
|
|
data_file: "./dataset/product_demo_data_v1.0/data_file.txt"
|
2021-06-09 02:40:53 +00:00
|
|
|
delimiter: " "
|
|
|
|
|
dist_type: "IP"
|
|
|
|
|
pq_size: 100
|
|
|
|
|
embedding_size: 512
|
