Merge pull request #808 from RainFrost1/develop_reg

add vehicle build and inference yaml for deply
pull/809/head
cuicheng01 2021-06-08 19:19:20 +08:00 committed by GitHub
commit 38201d2682
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 91 additions and 0 deletions

View File

@ -0,0 +1,35 @@
Global:
rec_inference_model_dir: "./vehicle/model/"
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:
index_path: "./vehilce/vehicle_index/"
image_root: "./vehicle/dataset/"
data_file: "./vehilce/demo_gallery.txt"
delimiter: " "
dist_type: "IP"
pq_size: 100
embedding_size: 512

View File

@ -0,0 +1,56 @@
Global:
infer_imgs: "./vehicle/demo/2e3521935c280c.jpg"
det_inference_model_dir: "./det/"
rec_inference_model_dir: "./vehicle/rec/"
batch_size: 1
image_shape: [3, 640, 640]
threshold: 0.5
max_det_results: 1
labe_list:
- foreground
# inference engine config
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
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:
index_path: "./vehicle_index/"
search_budget: 100
return_k: 10
dist_type: "IP"