diff --git a/deploy/configs/build_cartoon.yaml b/deploy/configs/build_cartoon.yaml new file mode 100644 index 0000000000000000000000000000000000000000..72d40b50d71b8486bd8cedd4414d6bb0c4ab0c51 --- /dev/null +++ b/deploy/configs/build_cartoon.yaml @@ -0,0 +1,36 @@ +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 diff --git a/deploy/configs/inference_icartoon.py b/deploy/configs/inference_cartoon.yaml similarity index 69% rename from deploy/configs/inference_icartoon.py rename to deploy/configs/inference_cartoon.yaml index 8bd5aeeec2123088cce3aea249e8072e075cfc2a..5b166bf4b9e061b00162c1fcef680b4b5d8257b0 100644 --- a/deploy/configs/inference_icartoon.py +++ b/deploy/configs/inference_cartoon.yaml @@ -1,7 +1,7 @@ Global: infer_imgs: "./dataset/iCartoonFace/val2/0000000.jpg" - det_inference_model_dir: "./output/det" - rec_inference_model_dir: "./output/" + det_inference_model_dir: "./cartoon/det" + rec_inference_model_dir: "./cartoon/rec" batch_size: 1 image_shape: [3, 640, 640] threshold: 0.5 @@ -9,7 +9,6 @@ Global: labe_list: - foreground - # inference engine config use_gpu: True enable_mkldnn: True cpu_num_threads: 100 @@ -34,7 +33,6 @@ DetPreProcess: DetPostProcess: {} - RecPreProcess: transform_ops: - ResizeImage: @@ -50,18 +48,7 @@ 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" - dist_type: "IP" - pq_size: 100 - embedding_size: 2048 - infer: - index_path: "./icartoon_index/" + index_path: "./cartoon/index/" search_budget: 100 return_k: 10