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

AlbinZhu · web-flow · commit 1314f3267e8e · 2024-02-27T19:21:17.000+08:00
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_);

Original file line number	Diff line number	Diff line change
`@@ -88,7 +88,7 @@ bool YOLOv5Postprocessor::Run(const std::vector<FDTensor>& tensors, std::vector<`
`88`	`88`	`}`
`89`	`89`
`90`	`90`	`if ((*results)[bs].boxes.size() == 0) {`
`91`		`- return true;`
	`91`	`+ continue;`
`92`	`92`	`}`
`93`	`93`
`94`	`94`	`utils::NMS(&((*results)[bs]), nms_threshold_);`
Original file line number	Diff line number	Diff line change
`@@ -105,7 +105,7 @@ bool YOLOv5SegPostprocessor::Run(`
`105`	`105`	`}`
`106`	`106`
`107`	`107`	`if ((*results)[bs].boxes.size() == 0) {`
`108`		`- return true;`
	`108`	`+ continue;`
`109`	`109`	`}`
`110`	`110`	`// get box index after nms`
`111`	`111`	`std::vector<int> index;`
Original file line number	Diff line number	Diff line change
`@@ -61,7 +61,7 @@ bool YOLOv7Postprocessor::Run(const std::vector<FDTensor>& tensors, std::vector<`
`61`	`61`	`}`
`62`	`62`
`63`	`63`	`if ((*results)[bs].boxes.size() == 0) {`
`64`		`- return true;`
	`64`	`+ continue;`
`65`	`65`	`}`
`66`	`66`
`67`	`67`	`utils::NMS(&((*results)[bs]), nms_threshold_);`
Original file line number	Diff line number	Diff line change
`@@ -93,7 +93,7 @@ bool YOLOv8Postprocessor::Run(`
`93`	`93`	`}`
`94`	`94`
`95`	`95`	`if ((*results)[bs].boxes.size() == 0) {`
`96`		`- return true;`
	`96`	`+ continue;`
`97`	`97`	`}`
`98`	`98`
`99`	`99`	`utils::NMS(&((*results)[bs]), nms_threshold_);`
Original file line number	Diff line number	Diff line change
`@@ -114,7 +114,7 @@ bool CenterFacePostprocessor::Run(const std::vector<FDTensor>& infer_result,`
`114`	`114`	`}`
`115`	`115`
`116`	`116`	`if ((*results)[bs].boxes.size() == 0) {`
`117`		`- return true;`
	`117`	`+ continue;`
`118`	`118`	`}`
`119`	`119`
`120`	`120`	`utils::NMS(&((*results)[bs]), nms_threshold_);`
Original file line number	Diff line number	Diff line change
`@@ -75,7 +75,7 @@ bool Yolov7FacePostprocessor::Run(const std::vector<FDTensor>& infer_result,`
`75`	`75`	`}`
`76`	`76`
`77`	`77`	`if ((*results)[bs].boxes.size() == 0) {`
`78`		`- return true;`
	`78`	`+ continue;`
`79`	`79`	`}`
`80`	`80`
`81`	`81`	`utils::NMS(&((*results)[bs]), nms_threshold_);`