OpenCV—椭圆拟合fitEllipse

本文的主要参考为官方文档OpenCV249-fitEllipse和博客-opencv中的椭圆拟合

以及《Learning OpenCV 3》page424-425

OpenCV中提供的椭圆拟合API如下：

RotatedRect fitEllipse(InputArray points)

输入：二维点集， 要求拟合的点至少为6个点。存储在std::vector<> or Mat

处理：该函数使用的是最小二乘的方法进行拟合的。参考论文：《Direct least square fitting of ellipses》

输出：RotatedRect 类型的矩形，是拟合出椭圆的最小外接矩形。

绘制椭圆用到的API如下，官方文档OpenCV249-ellipse

void ellipse(Mat& img, Point center, Size axes, double angle, double startAngle, double endAngle, const Scalar& color, int thickness=1, int lineType=8, int shift=0)
void ellipse(Mat& img, const RotatedRect& box, const Scalar& color, int thickness=1, int lineType=8)

对应的参数可参见下图：

示例代码如下：

#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/highgui/highgui.hpp"
#include <iostream>
using namespace cv;
using namespace std;

int main( )
{
	const char* filename = "rect-45度.bmp";
	Mat src_image = imread(filename, 0);
	if( src_image.empty() )
	{
		cout << "Couldn't open image!" << filename;
		return 0;
	}

	imshow("原图", src_image);
	
	//轮廓
	vector<vector<Point>> contours;

	//使用canny检测出边缘
	Mat edge_image;
	Canny(src_image,edge_image,30,70);
	imshow("canny边缘",edge_image);

	//边缘追踪，没有存储边缘的组织结构
	findContours(edge_image, contours, CV_RETR_EXTERNAL, CV_CHAIN_APPROX_NONE);
	Mat cimage = Mat::zeros(edge_image.size(), CV_8UC3);

	for(size_t i = 0; i < contours.size(); i++)
	{
		//拟合的点至少为6
		size_t count = contours[i].size();
		if( count < 6 )
			continue;

		//椭圆拟合
		RotatedRect box = fitEllipse(contours[i]);

		//如果长宽比大于30，则排除，不做拟合
		if( MAX(box.size.width, box.size.height) > MIN(box.size.width, box.size.height)*30 )
			continue;

		//画出追踪出的轮廓
		drawContours(cimage, contours, (int)i, Scalar::all(255), 1, 8);
		
		//画出拟合的椭圆
		ellipse(cimage, box, Scalar(0,0,255), 1, CV_AA);
	}
	imshow("拟合结果", cimage);

	waitKey();
	return 0;
}