Merge pull request #808 from RainFrost1/develop_reg
add vehicle build and inference yaml for deplypull/809/head
commit
38201d2682
|
@ -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
|
|
@ -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"
|
Loading…
Reference in New Issue