未验证 提交 91665f56 编写于 作者: W Wei Shengyu 提交者: GitHub

Merge pull request #814 from Intsigstephon/develop_reg

separate cartoon build and infer sciprt
Global:
rec_inference_model_dir: "./cartoon/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:
resize_short: 256
- CropImage:
size: 224
- NormalizeImage:
scale: 0.00392157
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
order: ''
- ToCHWImage:
RecPostProcess: null
IndexProcess:
index_path: "./cartoon/index/"
image_root: "./cartoon/dataset/"
data_file: "./cartoon/data_file.txt"
delimiter: "\t"
dist_type: "IP"
pq_size: 100
embedding_size: 2048
Global: Global:
infer_imgs: "./dataset/iCartoonFace/val2/0000000.jpg" infer_imgs: "./dataset/iCartoonFace/val2/0000000.jpg"
det_inference_model_dir: "./output/det" det_inference_model_dir: "./cartoon/det"
rec_inference_model_dir: "./output/" rec_inference_model_dir: "./cartoon/rec"
batch_size: 1 batch_size: 1
image_shape: [3, 640, 640] image_shape: [3, 640, 640]
threshold: 0.5 threshold: 0.5
...@@ -9,7 +9,6 @@ Global: ...@@ -9,7 +9,6 @@ Global:
labe_list: labe_list:
- foreground - foreground
# inference engine config
use_gpu: True use_gpu: True
enable_mkldnn: True enable_mkldnn: True
cpu_num_threads: 100 cpu_num_threads: 100
...@@ -34,7 +33,6 @@ DetPreProcess: ...@@ -34,7 +33,6 @@ DetPreProcess:
DetPostProcess: {} DetPostProcess: {}
RecPreProcess: RecPreProcess:
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
...@@ -50,18 +48,7 @@ RecPreProcess: ...@@ -50,18 +48,7 @@ RecPreProcess:
RecPostProcess: null RecPostProcess: null
# indexing engine config
IndexProcess: IndexProcess:
build: index_path: "./cartoon/index/"
enable: False
index_path: "./icartoon_index/"
image_root: "./dataset/iCartoonFace"
data_file: "./dataset/iCartoonFace/gallery_pesudo.txt"
spacer: "\t"
dist_type: "IP"
pq_size: 100
embedding_size: 2048
infer:
index_path: "./icartoon_index/"
search_budget: 100 search_budget: 100
return_k: 10 return_k: 10
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册