PaddleClas/deploy/configs/build_product.yaml

36 lines
852 B
YAML
Raw Normal View History

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
2021-06-19 02:10:53 +08:00
enable_mkldnn: False
2021-06-09 02:40:53 +00:00
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:
index_path: "./recognition_demo_data_v1.0/gallery_product/index"
image_root: "./recognition_demo_data_v1.0/gallery_product/"
data_file: "./recognition_demo_data_v1.0/gallery_product/data_file.txt"
delimiter: "\t"
2021-06-09 02:40:53 +00:00
dist_type: "IP"
pq_size: 100
embedding_size: 512