From 2de1e4d3e0f1fc36db83a759bce2f6cf6c9b649a Mon Sep 17 00:00:00 2001
From: dongshuilong <dongshuilong@baidu.com>
Date: Tue, 8 Jun 2021 18:59:24 +0800
Subject: [PATCH] add vehicle build and inference yaml for deply

---
 deploy/configs/build_vehicle.yaml     | 35 +++++++++++++++++
 deploy/configs/inference_vehicle.yaml | 56 +++++++++++++++++++++++++++
 2 files changed, 91 insertions(+)
 create mode 100644 deploy/configs/build_vehicle.yaml
 create mode 100644 deploy/configs/inference_vehicle.yaml

diff --git a/deploy/configs/build_vehicle.yaml b/deploy/configs/build_vehicle.yaml
new file mode 100644
index 00000000..ba45ae97
--- /dev/null
+++ b/deploy/configs/build_vehicle.yaml
@@ -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
diff --git a/deploy/configs/inference_vehicle.yaml b/deploy/configs/inference_vehicle.yaml
new file mode 100644
index 00000000..7bc3b889
--- /dev/null
+++ b/deploy/configs/inference_vehicle.yaml
@@ -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"
-- 
GitLab