Update build_cartoon.yaml

pull/814/head
Bin Lu 2021-06-09 16:25:09 +08:00 committed by GitHub
parent 58aee31567
commit 97763fd95e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 35 deletions

View File

@ -1,15 +1,6 @@
Global:
infer_imgs: "./dataset/iCartoonFace/val2/0000000.jpg"
det_inference_model_dir: "./output/det"
rec_inference_model_dir: "./output/"
rec_inference_model_dir: "./cartoon/model"
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
@ -20,21 +11,6 @@ Global:
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:
@ -50,18 +26,12 @@ RecPreProcess:
RecPostProcess: null
# indexing engine config
IndexProcess:
build:
enable: False
index_path: "./icartoon_index/"
image_root: "./dataset/iCartoonFace"
data_file: "./dataset/iCartoonFace/gallery_pesudo.txt"
spacer: "\t"
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
infer:
index_path: "./icartoon_index/"
search_budget: 100
return_k: 10