PaddleClas/deploy/configs/inference_product.yaml

56 lines
1.2 KiB
YAML
Raw Normal View History

2021-06-09 10:40:53 +08:00
Global:
2021-11-02 01:58:31 +08:00
infer_imgs: "./recognition_demo_data_v1.1/test_product/daoxiangcunjinzhubing_6.jpg"
det_inference_model_dir: "./models/ppyolov2_r50vd_dcn_mainbody_v1.0_infer"
rec_inference_model_dir: "./models/product_ResNet50_vd_aliproduct_v1.0_infer"
2021-06-20 08:37:14 +08:00
rec_nms_thresold: 0.05
2021-06-09 10:40:53 +08:00
batch_size: 1
image_shape: [3, 640, 640]
2021-06-16 17:29:08 +08:00
threshold: 0.2
2021-06-19 03:26:55 +08:00
max_det_results: 5
2021-06-09 10:40:53 +08:00
labe_list:
- foreground
# inference engine config
use_gpu: True
2021-07-14 13:58:45 +08:00
enable_mkldnn: True
cpu_num_threads: 10
2021-06-09 10:40:53 +08:00
enable_benchmark: True
use_fp16: False
ir_optim: True
use_tensorrt: False
gpu_mem: 8000
enable_profile: False
DetPreProcess:
transform_ops:
- DetResize:
interp: 2
keep_ratio: false
target_size: [640, 640]
- DetNormalizeImage:
is_scale: true
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
- DetPermute: {}
DetPostProcess: {}
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-11-02 01:58:31 +08:00
index_dir: "./recognition_demo_data_v1.1/gallery_product/index"
return_k: 5
2021-06-19 03:26:55 +08:00
score_thres: 0.5