From 1314f3267e8e35b467eb75cb815858155ea16c10 Mon Sep 17 00:00:00 2001 From: Albin Date: Tue, 27 Feb 2024 19:21:17 +0800 Subject: [PATCH] =?UTF-8?q?bug=20fix:=20=E4=BF=AE=E5=A4=8Dcontrib=E4=B8=AD?= =?UTF-8?q?=E7=9A=84det=E6=A8=A1=E5=9E=8B=E5=90=8E=E5=A4=84=E7=90=86?= =?UTF-8?q?=E5=9C=A8=E9=81=87=E5=88=B0=E6=B2=A1=E6=9C=89=E6=A3=80=E6=B5=8B?= =?UTF-8?q?=E6=A1=86=E7=9A=84=E5=9B=BE=E6=97=B6=E5=AF=BC=E8=87=B4=E5=90=8C?= =?UTF-8?q?batch=E4=B8=AD=E6=8E=A5=E4=B8=8B=E6=9D=A5=E7=9A=84=E5=9B=BE?= =?UTF-8?q?=E7=89=87=E7=BB=93=E6=9E=9C=E5=85=A8=E9=83=A8=E4=B8=BA=E7=A9=BA?= =?UTF-8?q?=E7=9A=84bug=20(#2378)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- fastdeploy/vision/detection/contrib/yolov5/postprocessor.cc | 2 +- fastdeploy/vision/detection/contrib/yolov5seg/postprocessor.cc | 2 +- fastdeploy/vision/detection/contrib/yolov7/postprocessor.cc | 2 +- fastdeploy/vision/detection/contrib/yolov8/postprocessor.cc | 2 +- fastdeploy/vision/facedet/contrib/centerface/postprocessor.cc | 2 +- fastdeploy/vision/facedet/contrib/yolov7face/postprocessor.cc | 2 +- 6 files changed, 6 insertions(+), 6 deletions(-) diff --git a/fastdeploy/vision/detection/contrib/yolov5/postprocessor.cc b/fastdeploy/vision/detection/contrib/yolov5/postprocessor.cc index 4fe01dfeb8..5e1e6da9dc 100755 --- 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& 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 index 50bcaba5c0..535f77cd32 100755 --- 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 index; diff --git a/fastdeploy/vision/detection/contrib/yolov7/postprocessor.cc b/fastdeploy/vision/detection/contrib/yolov7/postprocessor.cc index 01d657adb3..f5d6b41740 100755 --- 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& 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 index 77974cdd68..550271c40e 100755 --- 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 index 98931956b6..17aad8eba5 100644 --- 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& 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 index 624aa3403c..9f023faa09 100644 --- 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& infer_result, } if ((*results)[bs].boxes.size() == 0) { - return true; + continue; } utils::NMS(&((*results)[bs]), nms_threshold_);