From aefef5995647bfb0383788c52532dccc8e0ee8ad Mon Sep 17 00:00:00 2001 From: littletomatodonkey Date: Fri, 18 Jun 2021 19:26:55 +0000 Subject: [PATCH] fix yaml --- deploy/configs/inference_cartoon.yaml | 5 +++-- deploy/configs/inference_logo.yaml | 5 +++-- deploy/configs/inference_product.yaml | 3 ++- deploy/configs/inference_vehicle.yaml | 5 +++-- 4 files changed, 11 insertions(+), 7 deletions(-) diff --git a/deploy/configs/inference_cartoon.yaml b/deploy/configs/inference_cartoon.yaml index e47aca31d..d81440b1f 100644 --- a/deploy/configs/inference_cartoon.yaml +++ b/deploy/configs/inference_cartoon.yaml @@ -4,8 +4,8 @@ Global: rec_inference_model_dir: "./models/cartoon_rec_ResNet50_iCartoon_v1.0_infer/" batch_size: 1 image_shape: [3, 640, 640] - threshold: 0.5 - max_det_results: 1 + threshold: 0.2 + max_det_results: 5 labe_list: - foreground @@ -53,3 +53,4 @@ IndexProcess: search_budget: 100 return_k: 5 dist_type: "IP" + score_thres: 0.5 diff --git a/deploy/configs/inference_logo.yaml b/deploy/configs/inference_logo.yaml index 6f183f105..ba452b9c2 100644 --- a/deploy/configs/inference_logo.yaml +++ b/deploy/configs/inference_logo.yaml @@ -4,8 +4,8 @@ Global: rec_inference_model_dir: "./models/logo_rec_ResNet50_Logo3K_v1.0_infer/" batch_size: 1 image_shape: [3, 640, 640] - threshold: 0.5 - max_det_results: 1 + threshold: 0.2 + max_det_results: 5 labe_list: - foreground @@ -52,3 +52,4 @@ IndexProcess: search_budget: 100 return_k: 5 dist_type: "IP" + score_thres: 0.5 diff --git a/deploy/configs/inference_product.yaml b/deploy/configs/inference_product.yaml index 4a6aa572c..975aba323 100644 --- a/deploy/configs/inference_product.yaml +++ b/deploy/configs/inference_product.yaml @@ -5,7 +5,7 @@ Global: batch_size: 1 image_shape: [3, 640, 640] threshold: 0.2 - max_det_results: 1 + max_det_results: 5 labe_list: - foreground @@ -52,3 +52,4 @@ IndexProcess: search_budget: 100 return_k: 5 dist_type: "IP" + score_thres: 0.5 diff --git a/deploy/configs/inference_vehicle.yaml b/deploy/configs/inference_vehicle.yaml index 357b722d5..671018a0f 100644 --- a/deploy/configs/inference_vehicle.yaml +++ b/deploy/configs/inference_vehicle.yaml @@ -4,8 +4,8 @@ Global: rec_inference_model_dir: "./models/vehicle_cls_ResNet50_CompCars_v1.0_infer/" batch_size: 1 image_shape: [3, 640, 640] - threshold: 0.5 - max_det_results: 1 + threshold: 0.2 + max_det_results: 5 labe_list: - foreground @@ -54,3 +54,4 @@ IndexProcess: search_budget: 100 return_k: 5 dist_type: "IP" + score_thres: 0.5