opencv c++计算两个矩形的iou
时间: 2023-08-11 18:03:51 浏览: 349
在 OpenCV 中,我们可以使用 `cv::intersectConvexConvex()` 函数来计算两个矩形的交集,并使用 `cv::contourArea()` 函数计算它们的并集,从而计算它们的 IoU(Intersection over Union)。
以下是一个示例代码:
``` c++
#include <opencv2/opencv.hpp>
int main()
{
// 定义两个矩形
cv::Rect rect1(50, 50, 100, 100);
cv::Rect rect2(70, 70, 120, 80);
// 计算交集和并集
cv::Mat intersection;
cv::intersectConvexConvex(cv::Mat(cv::boxPoints(rect1)), cv::Mat(cv::boxPoints(rect2)), intersection);
std::vector<cv::Point> hull;
cv::convexHull(cv::Mat(cv::vconcat(cv::Mat(cv::boxPoints(rect1)), cv::Mat(cv::boxPoints(rect2)))), hull);
double unionArea = cv::contourArea(hull);
// 计算 IoU
double iou = cv::contourArea(intersection) / unionArea;
std::cout << "IoU: " << iou << std::endl;
return 0;
}
```
在上面的代码中,我们首先定义了两个矩形。然后,我们使用 `cv::Mat` 类型将它们转换为其对应的四个顶点坐标,并使用 `cv::intersectConvexConvex()` 函数计算它们的交集。接下来,我们使用 `cv::vconcat()` 函数将两个矩形的顶点坐标垂直拼接,并使用 `cv::convexHull()` 函数计算它们的并集。最后,我们使用 `cv::contourArea()` 函数计算交集和并集的面积,并计算 IoU 值。
请注意,在 C++ 中,`cv::Rect` 类型的构造函数参数依次为 `(x, y, width, height)`。
阅读全文