Skip to content
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

Solving the issue of out of bounds caused by detection boxes less than 0 when filling colors in segmentation masks #1516

Merged
merged 7 commits into from
May 9, 2024
Merged

Solving the issue of out of bounds caused by detection boxes less than 0 when filling colors in segmentation masks #1516

Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
e8417b0
Merge pull request #1 from wang-xinyu/master
lindsayshuo Apr 30, 2024
e9e7adb
Add the generation of multi-class pose engines
May 2, 2024
e92bf89
Merge pull request #2 from lindsayshuo/lindsay
lindsayshuo May 2, 2024
6d36a3c
Merge pull request #3 from wang-xinyu/master
lindsayshuo May 7, 2024
46dd2c4
Merge pull request #4 from wang-xinyu/master
lindsayshuo May 9, 2024
4feb62e
Solving the issue of out of bounds caused by detection boxes less tha…
lindsayshuo May 9, 2024
1cf09f2
Solving the issue of out of bounds caused by detection boxes less tha…
lindsayshuo May 9, 2024
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
16 changes: 13 additions & 3 deletions yolov8/src/postprocess.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -25,11 +25,16 @@ cv::Rect get_rect(cv::Mat& img, float bbox[4]) {
t = t / r_h;
b = b / r_h;
}
return cv::Rect(round(l), round(t), round(r - l), round(b - t));
l = std::max(0.0f, l);
t = std::max(0.0f, t);
int width = std::max(0, std::min(int(round(r - l)), img.cols - int(round(l))));
int height = std::max(0, std::min(int(round(b - t)), img.rows - int(round(t))));

return cv::Rect(int(round(l)), int(round(t)), width, height);
}

cv::Rect get_rect_adapt_landmark(cv::Mat& img, float bbox[4], float lmk[kNumberOfPoints * 3]) {
int l, r, t, b;
float l, r, t, b;
float r_w = kInputW / (img.cols * 1.0);
float r_h = kInputH / (img.rows * 1.0);
if (r_h > r_w) {
Expand All @@ -53,7 +58,12 @@ cv::Rect get_rect_adapt_landmark(cv::Mat& img, float bbox[4], float lmk[kNumberO
// lmk[i + 2]
}
}
return cv::Rect(l, t, r - l, b - t);
l = std::max(0.0f, l);
t = std::max(0.0f, t);
int width = std::max(0, std::min(int(round(r - l)), img.cols - int(round(l))));
int height = std::max(0, std::min(int(round(b - t)), img.rows - int(round(t))));

return cv::Rect(int(round(l)), int(round(t)), width, height);
}

static float iou(float lbox[4], float rbox[4]) {
Expand Down
Loading