apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-15 21:20:53 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Albin
2024-02-27 19:21:17 +08:00
committed by GitHub
parent cfd80e95a1
commit 1314f3267e
6 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
fastdeploy/vision/detection/contrib/yolov5/postprocessor.cc
View File

@@ -88,7 +88,7 @@ bool YOLOv5Postprocessor::Run(const std::vector<FDTensor>& tensors, std::vector<
} }
if ((*results)[bs].boxes.size() == 0) { if ((*results)[bs].boxes.size() == 0) {
return true; continue;
} }
utils::NMS(&((*results)[bs]), nms_threshold_); utils::NMS(&((*results)[bs]), nms_threshold_);

2
fastdeploy/vision/detection/contrib/yolov5seg/postprocessor.cc
View File

@@ -105,7 +105,7 @@ bool YOLOv5SegPostprocessor::Run(
} }
if ((*results)[bs].boxes.size() == 0) { if ((*results)[bs].boxes.size() == 0) {
return true; continue;
} }
// get box index after nms // get box index after nms
std::vector<int> index; std::vector<int> index;

2
fastdeploy/vision/detection/contrib/yolov7/postprocessor.cc
View File

@@ -61,7 +61,7 @@ bool YOLOv7Postprocessor::Run(const std::vector<FDTensor>& tensors, std::vector<
} }
if ((*results)[bs].boxes.size() == 0) { if ((*results)[bs].boxes.size() == 0) {
return true; continue;
} }
utils::NMS(&((*results)[bs]), nms_threshold_); utils::NMS(&((*results)[bs]), nms_threshold_);

2
fastdeploy/vision/detection/contrib/yolov8/postprocessor.cc
View File

@@ -93,7 +93,7 @@ bool YOLOv8Postprocessor::Run(
} }
if ((*results)[bs].boxes.size() == 0) { if ((*results)[bs].boxes.size() == 0) {
return true; continue;
} }
utils::NMS(&((*results)[bs]), nms_threshold_); utils::NMS(&((*results)[bs]), nms_threshold_);

2
fastdeploy/vision/facedet/contrib/centerface/postprocessor.cc
View File

@@ -114,7 +114,7 @@ bool CenterFacePostprocessor::Run(const std::vector<FDTensor>& infer_result,
} }
if ((*results)[bs].boxes.size() == 0) { if ((*results)[bs].boxes.size() == 0) {
return true; continue;
} }
utils::NMS(&((*results)[bs]), nms_threshold_); utils::NMS(&((*results)[bs]), nms_threshold_);

2
fastdeploy/vision/facedet/contrib/yolov7face/postprocessor.cc
View File

@@ -75,7 +75,7 @@ bool Yolov7FacePostprocessor::Run(const std::vector<FDTensor>& infer_result,
} }
if ((*results)[bs].boxes.size() == 0) { if ((*results)[bs].boxes.size() == 0) {
return true; continue;
} }
utils::NMS(&((*results)[bs]), nms_threshold_); utils::NMS(&((*results)[bs]), nms_threshold_);