Skip to content

Fix bug while the inference result is empty with YOLOv5 #29

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
jiangjiajun merged 9 commits into from
Jul 19, 2022
Merged

Fix bug while the inference result is empty with YOLOv5 #29

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
cb5c585
Add multi-label function for yolov5
felixhjh Jul 12, 2022
5ac4976
Update README.md
felixhjh Jul 12, 2022
4c1f986
Update fastdeploy_runtime.cc
felixhjh Jul 14, 2022
6410251
Update runtime_option.md
felixhjh Jul 14, 2022
443f3a7
Merge pull request #1 from PaddlePaddle/develop
felixhjh Jul 14, 2022
7594559
Fix bug when inference result boxes are empty
felixhjh Jul 19, 2022
4c88b9e
Merge pull request #2 from PaddlePaddle/develop
felixhjh Jul 19, 2022
9332cf2
Delete detection.py
felixhjh Jul 19, 2022
67d3086
Merge branch 'yolov5' of https://github.com/felixhjh/FastDeploy into …
felixhjh Jul 19, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions fastdeploy/vision/ultralytics/yolov5.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -198,6 +198,11 @@ bool YOLOv5::Postprocess(
result->scores.push_back(confidence);
}
}

if (result->boxes.size() == 0) {
return true;
}

utils::NMS(result, nms_iou_threshold);

// scale the boxes to the origin image shape
Expand Down
6 changes: 5 additions & 1 deletion fastdeploy/vision/utils/sort_det_res.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,11 @@ void MergeSort(DetectionResult* result, size_t low, size_t high) {

void SortDetectionResult(DetectionResult* result) {
size_t low = 0;
size_t high = result->scores.size() - 1;
size_t high = result->scores.size();
if (high == 0) {
return;
}
high = high - 1;
MergeSort(result, low, high);
}

Expand Down