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

2025-10-15 13:10:55 +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/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;