bug fix: 修复contrib中的det模型后处理在遇到没有检测框的图时导致同batch中接下来的图片结果全部为空的bug

PaddlePaddle · Feb 27, 2024 · 5ef5e2a · 5ef5e2a
1 parent cfd80e9
commit 5ef5e2a
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/fastdeploy/vision/detection/contrib/yolov5/postprocessor.cc b/fastdeploy/vision/detection/contrib/yolov5/postprocessor.cc
@@ -88,7 +88,7 @@ bool YOLOv5Postprocessor::Run(const std::vector<FDTensor>& tensors, std::vector<
     }
 
     if ((*results)[bs].boxes.size() == 0) {
-      return true;
+      continue;
     }
 
     utils::NMS(&((*results)[bs]), nms_threshold_);

diff --git a/fastdeploy/vision/detection/contrib/yolov5seg/postprocessor.cc b/fastdeploy/vision/detection/contrib/yolov5seg/postprocessor.cc
@@ -105,7 +105,7 @@ bool YOLOv5SegPostprocessor::Run(
     }
 
     if ((*results)[bs].boxes.size() == 0) {
-      return true;
+      continue;
     }
     // get box index after nms
     std::vector<int> index;

diff --git a/fastdeploy/vision/detection/contrib/yolov7/postprocessor.cc b/fastdeploy/vision/detection/contrib/yolov7/postprocessor.cc
@@ -61,7 +61,7 @@ bool YOLOv7Postprocessor::Run(const std::vector<FDTensor>& tensors, std::vector<
     }
 
     if ((*results)[bs].boxes.size() == 0) {
-      return true;
+      continue;
     }
 
     utils::NMS(&((*results)[bs]), nms_threshold_);

diff --git a/fastdeploy/vision/detection/contrib/yolov8/postprocessor.cc b/fastdeploy/vision/detection/contrib/yolov8/postprocessor.cc
@@ -93,7 +93,7 @@ bool YOLOv8Postprocessor::Run(
     }
 
     if ((*results)[bs].boxes.size() == 0) {
-      return true;
+      continue;
     }
 
     utils::NMS(&((*results)[bs]), nms_threshold_);

diff --git a/fastdeploy/vision/facedet/contrib/centerface/postprocessor.cc b/fastdeploy/vision/facedet/contrib/centerface/postprocessor.cc
@@ -114,7 +114,7 @@ bool CenterFacePostprocessor::Run(const std::vector<FDTensor>& infer_result,
     }
 
     if ((*results)[bs].boxes.size() == 0) {
-      return true;
+      continue;
     }
 
     utils::NMS(&((*results)[bs]), nms_threshold_);

diff --git a/fastdeploy/vision/facedet/contrib/yolov7face/postprocessor.cc b/fastdeploy/vision/facedet/contrib/yolov7face/postprocessor.cc
@@ -75,7 +75,7 @@ bool Yolov7FacePostprocessor::Run(const std::vector<FDTensor>& infer_result,
     }
 
     if ((*results)[bs].boxes.size() == 0) {
-      return true;
+      continue;
     }
 
     utils::NMS(&((*results)[bs]), nms_threshold_);