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

2025-10-15 05:01:00 +08:00 · 2024-02-27 19:21:17 +08:00
parent cfd80e95a1
commit 1314f3267e
6 changed files with 6 additions and 6 deletions
--- 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_);