获得两个矩形的交集,该交集类型也是矩形
这种情况就范围交集的矩形,蓝色部分;
这种情况返回 cv::Rect(cv::Point(0,0),cv::Point(0,0) );
欢迎测试,提BUG;
/** @brief 获得两个矩形的交集
@param rect1; rect2 is lds range.
@output cv::Rect 交集
*/
cv::Rect get_IOU(cv::Rect rect1, cv::Rect rect2){
// 横向和纵向距离
cout <<"input " << rect1 << rect2 << endl;
cv::Point center_rect1 = (rect1.tl()+rect1.br()) * 0.5;
cv::Point center_rect2 = (rect2.tl()+rect2.br()) * 0.5;
cout <<"input center" << rect1 << rect2 << endl;
auto verti_dis = abs(center_rect1.x - center_rect2.x);
auto horiz_dis = abs(center_rect1.y - center_rect2.y);
// 高度和宽度之和的一半
auto half_w = (rect1.width + rect2.width) / 2;
auto half_y = (rect1.height + rect2.height) / 2;
cout << verti_dis << " " << half_y << " " << verti_dis << " " << half_w << endl;
if ( horiz_dis < half_y && verti_dis < half_w) {
int x_tl , x_br, y_tl, y_br;
if(rect1.tl().x > rect2.tl().x){
if(rect1.br().x > rect2.br().x){
x_tl = rect1.tl().x;
x_br = rect2.br().x;
} else{
x_tl = rect1.tl().x;
x_br = rect1.br().x;
}
} else{
if(rect1.br().x > rect2.br().x){
x_tl = rect2.tl().x;
x_br = rect2.br().x;
} else{
x_tl = rect2.tl().x;
x_br = rect1.br().x;
}
}
if(rect1.tl().y > rect2.tl().y){
if(rect1.br().y > rect2.br().y){
y_tl = rect1.tl().y;
y_br = rect2.br().y;
} else{
y_tl = rect1.tl().y;
y_br = rect1.br().y;
}
} else{
if(rect1.br().x > rect2.br().y){
y_tl = rect2.tl().y;
y_br = rect2.br().y;
} else{
y_tl = rect2.tl().y;
y_br = rect1.br().y;
}
}
cv::Point tl(x_tl, y_tl);
cv::Point br(x_br, y_br);
return cv::Rect(tl, br);
}
return cv::Rect(cv::Point(0,0),cv::Point(0,0) );
}
版本2
/** @brief 获得两个矩形的交集
@param rect1;rect2 is lds range.
@output cv::Rect 交集
*/
cv::Rect get_IOU(cv::Rect rect1, cv::Rect rect2){
// 横向和纵向距离
std::cout <<"input " << rect1 << " " << rect2 << std::endl;
cv::Point center_rect1 = (rect1.tl() + rect1.br()) * 0.5;
cv::Point center_rect2 = (rect2.tl() + rect2.br()) * 0.5;
std::cout <<"input center" << center_rect1 << " " << center_rect2 << std::endl;
auto verti_dis = std::abs(center_rect1.x - center_rect2.x);
auto horiz_dis = std::abs(center_rect1.y - center_rect2.y);
// 高度和宽度之和的一半
auto half_w = (rect1.width + rect2.width) / 2;
auto half_h = (rect1.height + rect2.height) / 2;
std::cout << verti_dis << " " << half_h << " " << horiz_dis << " " << half_w << std::endl;
if (horiz_dis < half_w && verti_dis < half_h) {
int x_tl = std::max(rect1.tl().x, rect2.tl().x);
int y_tl = std::max(rect1.tl().y, rect2.tl().y);
int x_br = std::min(rect1.br().x, rect2.br().x);
int y_br = std::min(rect1.br().y, rect2.br().y);
if (x_tl < x_br && y_tl < y_br) {
return cv::Rect(cv::Point(x_tl, y_tl), cv::Point(x_br, y_br));
}
}
return cv::Rect(cv::Point(0, 0), cv::Point(0, 0));
}