使用OpenCV检测组件的主方向

Question

我有一张像A图的图片，我想要检测出B图所显示的主方向。在我的代码中，我首先检测边缘，然后尝试找到轮廓。然后我计划使用矩来通过"angle = 180*atan((2*m11)/(m20 - m02))/(PI*2.0);"来得到角度。但是结果是错误的！

// My function
Mat edge_Processing(Mat Input_image,int Para_canny_1,int Para_canny_2)
{
Mat edge;

//blur gray
blur(Input_image, Input_image, Size(3,3));

// Canny edge
Canny(Input_image,edge, Para_canny_1,Para_canny_2 , 3);

//imshow("Before contour",edge);

std::vector<std::vector<cv::Point>> contours;
cv::findContours(edge, 
    contours, // a vector of contours 
    CV_RETR_EXTERNAL, // retrieve the ezxternal contours
    CV_CHAIN_APPROX_NONE); // all pixels of each contours

// Draw black contours on a white image
cv::Mat result(edge.size(),CV_8U,cv::Scalar(0));


std::vector<std::vector<cv::Point>>::
    const_iterator itc= contours.begin();

//Get the moments
vector<Moments> mu(contours.size() );
int moment_count=0;
while (itc!=contours.end())
{
    mu[moment_count] = moments(contours[moment_count], false );

double m11=mu[moment_count].m11;
double m20=mu[moment_count].m20;
double m02=mu[moment_count].m02;

double angle=90;
    // In case of the situation of dividing by 0        
if (abs(m20 - m02)>0.001)
{
angle = 180*atan((2*m11)/(m20 - m02))/(PI*2.0);
}

cout<<angle<<endl; // Here! The angle are wrong!


++itc;
moment_count++;

}
    // Draw the contour
cv::drawContours(result,contours,
    -1, // draw all contours
    cv::Scalar(255), // in black
    1); // with a thickness of 1

//imshow("contour after filtering",result);

return result;
};

这里展示了一些结果，我们可以看到这些组件的角度几乎是90度。但是这些结果都是错误的。 感谢！

- David Ding

Answer 1

根据您的需求，您也可以使用RotatedRect。

//you got your contour
RotatedRect minRect = minAreaRect(contour);
minRect.angle //angle from 0-90

请查看此线程，了解如何从中计算出正确的角度。

如果您想使用RotatedRect并且轮廓接近矩形，则这可能是唯一可行的方法。

- Sebastian Schmitz