opencv的多边形拟合函数cv::approxPolyDP(),不能指定想要拟合的多边形的边数,然后往往很多时候,我们是已知多边形的边数的,这里在approxPolyDP的基础之上通过二分法去寻找满足要拟合的多边形的epsilon,然后得到要拟合的多边形。
//利用二分法逼近epsilon的四边形(多边形)拟合
bool myapproxPolyDP(std::vector<cv::Point> contours, std::vector<cv::Point>& rects,double minepsilon = 1, double maxepsilon = 20,int sides=4);
myapproxPolyDP(std::vector<cv::Point> contours, std::vector<cv::Point>& rects, double minepsilon, double maxepsilon, int sides)
{
std::vector<cv::Point> rect1;
std::vector<cv::Point> rect2;
cv::approxPolyDP(contours, rect1, minepsilon, true);
cv::approxPolyDP(contours, rect2, maxepsilon, true);
if (rect1.size() > sides && rect2.size() > sides)
{
rects = contours;
return false;
}
if (rect1.size() < sides && rect2.size() < sides)
{
rects = contours;
return false;
}
else
{
if (rect1.size() == sides)
{
rects.resize(sides);
for (int i = 0; i < sides; i++)
{
rects[i] = rect1[i];
}
return true;
}
else if (rect2.size() == sides)
{
rects.resize(sides);
for (int i = 0; i < sides; i++)
{
rects[i] = rect2[i];
}
return true;
}
else
{
double midepsilon = (minepsilon + maxepsilon) / 2.0;
std::vector<cv::Point> rect3;
cv::approxPolyDP(contours, rect3, midepsilon, true);
if (rect3.size() == sides)
{
rects.resize(sides);
for (int i = 0; i < sides; i++)
{
rects[i] = rect3[i];
}
return true;
}
else if (rect3.size() < sides)
{
return myapproxPolyDP(contours, rects, minepsilon, midepsilon);
}
else
{
return myapproxPolyDP(contours, rects, midepsilon, minepsilon);
}
}
}
return true;
}
效果图。