2021-06-18 17:44:04 +08:00
|
|
|
Global:
|
|
|
|
infer_imgs: "./images/wangzai.jpg"
|
|
|
|
det_inference_model_dir: "./models/ppyolov2_r50vd_dcn_mainbody_v1.0_infer"
|
|
|
|
batch_size: 1
|
|
|
|
image_shape: [3, 640, 640]
|
|
|
|
threshold: 0.2
|
|
|
|
max_det_results: 1
|
|
|
|
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-18 17:44:04 +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: {}
|
2021-12-06 15:54:45 +08:00
|
|
|
|
2021-12-06 16:14:00 +08:00
|
|
|
DetPostProcess: {}
|