From 0126e953b3f293b111179e4777103c64f778870c Mon Sep 17 00:00:00 2001 From: Scheaven <xuepengqiang> Date: 星期四, 17 六月 2021 09:57:28 +0800 Subject: [PATCH] bug --- lib/core/detecter_manager.cpp | 9 +++------ 1 files changed, 3 insertions(+), 6 deletions(-) diff --git a/lib/core/detecter_manager.cpp b/lib/core/detecter_manager.cpp index 7364815..badcd72 100644 --- a/lib/core/detecter_manager.cpp +++ b/lib/core/detecter_manager.cpp @@ -43,7 +43,7 @@ { std::vector<bbox_t> result_vec; - result_vec = Detector::getInstance()->detect(mat_image); // 闈欐�佸崟渚嬫ā寮忥紝璋冪敤鍗曚緥鎸囬拡瀵硅薄杩涜浜轰綋妫�娴� + result_vec = Detector::getInstance()->detect(mat_image); encoder_features(result_vec, t_result); //灏嗘娴嬬殑缁撴灉杞寲鎴愭寚瀹氱殑鏍煎紡 @@ -51,7 +51,7 @@ // draw_boxes(mat_image, result_vec); } -// 灏嗘娴嬬粨鏋滅殑result_box淇℃伅铻嶅悎鍒癲etections涓� + void DetecterManager::encoder_features(std::vector<bbox_t> boxes, TResult &t_result) { t_result.targets = (Target*)malloc(sizeof(Target) * boxes.size()); // 缁樺埗浜鸿劯淇℃伅鐨� @@ -62,13 +62,11 @@ int result_index=0; - //NMS 鍘绘帀閲嶅彔鐨勬淇℃伅 start std::vector<bbox_t> n_boxes; float threshold = 0.99; - // 闇�瑕佷娇鐢ㄨ凯浠e櫒鐨勬柟寮忚繘琛屾摝闄� for (vector<bbox_t>::iterator it = boxes.begin();it !=boxes.end();) { - if(it->obj_id!=0) //灏嗛潪浜轰綋淇℃伅杞Щ鍒皀_boxes + if(it->obj_id!=0) { n_boxes.push_back(*it); it = boxes.erase(it); @@ -97,7 +95,6 @@ } boxes.erase(boxes.begin()); } - //NMS 鍘绘帀閲嶅彔鐨勬淇℃伅 end for (const auto &result_box:n_boxes) -- Gitblit v1.8.0