计算两个旋转椭圆的交集/并集?

时间:2017-10-09 05:13:45

标签: python opencv geometry object-detection

我在SE和其他地方看到了很多关于计算交叉路口的帖子。两个矩形的联合,但我有两个任意的椭圆:旋转,平移,不同的半长圆和&半轴。

有没有人建议计算他们的交叉联合价值?

我正在使用OpenCV& Python,它的价值。

如果我/我们不能提出任何聪明和快速的东西,那么我将使用基于掩码的方法,我只需要定义一个零数组,用1' s填充它在第一个椭圆定义的区域上,并为第二个椭圆添加1,然后我知道联合是数组非零的任何地方,并且交集将在数组为2的任何地方。但这听起来有点慢。

编辑: Re。 "听起来有点慢":如同,我想评估这个' jillions'每秒一次,所以我更喜欢用解析方程来计算它(例如,就像非旋转矩形一样),而不仅仅是基于像素的掩码实现。

1 个答案:

答案 0 :(得分:0)

如果您不想制作额外的蒙版,则必须在椭圆中实现布尔函数(使用一般椭圆方程式):

https://math.stackexchange.com/a/434482/

然后,只有在方程式满足时才能对像素进行着色。下面我附上了c ++示例(我应该很容易将其转换为python):

bool ifInEllipse(cv::Size _axes, float _angle, cv::Point _center, cv::Point _point)
{

_angle =CV_PI/180*_angle;


float cosine = cos(_angle);
float sine = sin(_angle);


float res1 = (_point.x - _center.x)*cosine + (_point.y - _center.y)*sine;
res1 = res1*res1;
float res2 = (_point.x - _center.x)*sine - (_point.y - _center.y)*cosine;
res2 = res2*res2;

float a = _axes.width;
float b = _axes.height;
return ((res1 / (a*a)) + (res2 / (b*b))) < 1;
}

和示例用法:

cv::Mat A1,A2, B1,B2, AA,BB;
A1 = cv::Mat::zeros(500, 500, CV_8UC1);
A2 = cv::Mat::zeros(500, 500, CV_8UC1);

cv::Point center1(200, 200);
cv::Size axes1(100, 50);
float angle1 =30;

cv::ellipse(A1, center1, axes1, angle1, 0, 360,cv::Scalar(255,255,255),-1);


B1 = cv::Mat::zeros(500, 500, CV_8UC1);
B2 = cv::Mat::zeros(500, 500, CV_8UC1);
for(int i=0;i<500;i++)
{
    for(int j=0;j<500;j++)
    {
        if(ifInEllipse(axes1,angle1,center1,cv::Point(i,j)))
        {
            B1.at<uchar>(j, i) = 255;
        }
    }
}

cv::Point center2(350, 300);
cv::Size axes2(60, 120);
float angle2 = -45; 


cv::ellipse(A2, center2, axes2, angle2, 0, 360, cv::Scalar(255, 255, 255), -1);


B2 = cv::Mat::zeros(500, 500, CV_8UC1);
for (int i = 0; i<500; i++)
{
    for (int j = 0; j<500; j++)
    {
        if (ifInEllipse(axes2, angle2, center2, cv::Point(i, j)))
        {
            B2.at<uchar>(j, i) = 255;
        }
    }
}


AA = cv::Mat::zeros(500, 500, CV_8UC1);
BB = cv::Mat::zeros(500, 500, CV_8UC1);

cv::bitwise_and(A1, A2,AA);



for (int i = 0; i<500; i++)
{
    for (int j = 0; j<500; j++)
    {
        if (ifInEllipse(axes1, angle1, center1, cv::Point(i, j)))
        {
            if (ifInEllipse(axes2, angle2, center2, cv::Point(i, j)))
            {
                BB.at<uchar>(j, i) = 255;
            }
        }
    }
}

结果:

enter image description here

相关问题