From 97763fd95e428c79e27bbb3b2214b2ec03abde85 Mon Sep 17 00:00:00 2001 From: Bin Lu Date: Wed, 9 Jun 2021 16:25:09 +0800 Subject: [PATCH] Update build_cartoon.yaml --- deploy/configs/build_cartoon.yaml | 40 ++++--------------------------- 1 file changed, 5 insertions(+), 35 deletions(-) diff --git a/deploy/configs/build_cartoon.yaml b/deploy/configs/build_cartoon.yaml index 8bd5aeee..0a254233 100644 --- a/deploy/configs/build_cartoon.yaml +++ b/deploy/configs/build_cartoon.yaml @@ -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 -- GitLab