修改锥桶阈值

yolov3/extra/api.cpp

@@ -20,7 +20,7 @@ void InitCnrt(bool useConfig,std::vector<std::pair<std::string,std::string>> mod
   //config_v3.file_model_weights = "../configs/yolov3-spp_final.weights";
   //config_v3.calibration_image_list_file_txt = "../configs/calibration_images.txt";
   config_v3.inference_precison =INT8;
-  config_v3.detect_thresh = 0.5;
+  config_v3.detect_thresh = 0.05;
 
   if(useConfig)
   {
@@ -76,6 +76,11 @@ void inference(std::map<int,Results*>* results_map,vector<cv::Mat>* input_mats)
             MyResult result;
             result.location = {r.rect.x, r.rect.y, r.rect.x + r.rect.width, r.rect.y + r.rect.height};
             result.prob = r.prob;
+            if(r.id != 4){
+              if(r.prob < 0.5){
+                continue;
+              }
+            }
             switch (r.id) {
                 case 0: {
                     result.facility_name = "person";