椭圆拟合方法

作者总结工作中椭圆拟合问题的踩坑经验与使用方法。介绍了opencv3.2方法,其有时会失效;matlab的BFisher方法拟合效果好;还有c++版本的BFisher方法,但lapack配置复杂;最新opencv4.0以上版本可直接调用相关库,不过升级版本可能影响原有程序编译。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

最近工作中遇到了有关椭圆拟合的问题,把在这一过程中踩坑和最后使用的方法进行了总结。
参考链接:https://github.com/seisgo/EllipseFit
https://github.com/xiamenwcy/EllipseFitting

opencv3.2的方法

首先使用的是opencv的方法,该方法有时候会失效。

//待拟合的点
vector<Point> vpfinalfit = dangerzone;

//把点的数量限制到20个
int interval = floor(vpfinalfit.size() / 20.0);
if (interval > 1)
{
	vector<Point> vpsample;
	for (int i = 0; i < vpfinalfit.size(); i += interval)
	{
		vpsample.push_back(vpfinalfit[i]);
	}
	vpfinalfit = vpsample;
}

//把坐标输出的文件中,在matlab中拟合
ofstream fout3;
fout3.open("data3.txt");
for (unsigned long i = 0; i < vpfinalfit.size(); i++)
{
	fout3 << vpfinalfit[i].x << " " << vpfinalfit[i].y << endl;
}
fout3.close();

//使用椭圆拟合的方法
RotatedRect rr;
if (vpfinalfit.size() > 5)
{
	rr = fitEllipse(vpfinalfit);
	ellipse(src, rr, Scalar(0, 0, 0), 4);
}

拟合较好的结果
在这里插入图片描述
拟合不对的结果
在这里插入图片描述

matlab的方法(BFisher方法)

参考链接 http://homepages.inf.ed.ac.uk/rbf/CVonline/LOCAL_COPIES/FITZGIBBON/ELLIPSE/
参考文献:Andrew W. Fitzgibbon, Maurizio Pilu, and Robert B. Fisher Direct least-squares fitting of ellipses, IEEE Transactions on Pattern Analysis and Machine Intelligence, 21(5), 476–480, May 1999

同样的点,在matlab中进行拟合,则可以实现较好的效果。

x = load('data3.mat');
p0=[0.005 0.005 0.005 0.005 0.005 0.005];
warning off
F=@(p,x)p(1)*x(:,1).^2+p(2)*x(:,1).*x(:,2)+p(3)*x(:,2).^2+p(4)*x(:,1)+p(5)*x(:,2)+p(6);
% ???????????
p=nlinfit(x,zeros(size(x,1),1),F,p0);

A=p(1)/p(6);
B=p(2)/p(6);
C=p(3)/p(6);
D=p(4)/p(6);
E=p(5)/p(6);

X_center = (B*E-2*C*D)/(4*A*C - B^2);
Y_center = (B*D-2*A*E)/(4*A*C - B^2);
fprintf(' X_center=%g, Y_center=%g\n',X_center,Y_center);

a= 2*sqrt((2*A*(X_center^2)+2*C*(Y_center^2)+2*B*X_center*Y_center-2)/(A+C+sqrt(((A-C)^2+B^2))));
b= 2*sqrt((2*A*(X_center^2)+2*C*(Y_center^2)+2*B*X_center*Y_center-2)/(A+C-sqrt(((A-C)^2+B^2))));
 
 
q=0.5 * atan(B/(A-C));
fprintf(' q=%g\n',q);
 
 
fprintf(' a=%g, b=%g\n',a,b);
plot(x(:,1),x(:,2),'ro');
 
 
hold on;
xmin=min(x(:,1));
xmax=max(x(:,1));
ymin=min(x(:,2));
ymax=max(x(:,2));
% 作图
 
ezplot(@(x,y)F(p,[x,y]),[xmin,xmax,ymin,ymax]);
title('曲线拟合');
legend('采样点','拟合结果')

在这里插入图片描述

c++版本(BFisher方法)

该方法也有c++的版本,但是lapack的配置比较复杂。
DirectEllipseFit拟合的类定义,具体的代码可以去优快云下载,稍后会上传,命名分别为myEllipse.h和myEllipse.cpp

template <typename T>
class DirectEllipseFit
{
public:
	DirectEllipseFit(const vector<T>& xData, const vector<T>& yData);
	Ellipse doEllipseFit();

private:
	T getMeanValue(const vector<T>& data);
	T getMaxValue(const vector<T>& data);
	T getMinValue(const vector<T>& data);
	T getScaleValue(const vector<T>& data);
	vector<T> symmetricNormalize(const vector<T>& data);
	//Make sure xData and yData are of same size
	vector<T> dotMultiply(const vector<T>& xData, const vector<T>& yData);
	
	//Get n*6 design matrix D, make sure xData and yData are of same size
	vector<vector<T> > getDesignMatrix(const vector<T>& xData,const vector<T>& yData);
	//Get 6*6 constraint matrix C
	vector<vector<T> > getConstraintMatrix();
	//Get 6*6 scatter matrix S from design matrix
	vector<vector<T> > getScatterMatrix(const vector<vector<T> >& dMtrx);
	//Transpose matrix
	vector<vector<T> > transposeMatrix(const vector<vector<T> >& mtrx);
	//Do matrix multiplication, mtrx1: j*l; mtrx2: l*i; return: j*i
	vector<vector<T> > doMtrxMul(const vector<vector<T> >& mtrx1,
		const vector<vector<T> >& mtrx2);

	/**
	 * @brief solveGeneralEigens:   Solve generalized eigensystem
	 * @note        For real eiginsystem solving.
	 * @param sMtrx:    6*6 square matrix in this application
	 * @param cMtrx:    6*6 square matrix in this application
	 * @param eigVV:    eigenvalues and eigenvectors, 6*7 matrix
	 * @return  success or failure status
	 */
	bool solveGeneralEigens(const vector<vector<T> >& sMtrx,
		const vector<vector<T> >& cMtrx,
		vector<vector<T> >& eigVV);
	//Convert matrix expression from nested QVector to 1-order array
	double* mtrx2array(const vector<vector<T> >& mtrx);

	/**
	 * @brief calcEllipsePara:  calculate ellipse parameter form eigen information
	 * @param eigVV:    eigenvalues and eigenvectors
	 * @return ellipse parameter
	 */
	Ellipse calcEllipsePara(const vector<vector<T> >& eigVV);

private:
	vector<T> m_xData, m_yData;
};

更简单的方法

如果你喜欢用最新版的软件,在opencv4.0以上的版本中,椭圆拟合函数增加了新的方法,其中有一个就是使用的上述方法,所有可以直接调用opencv4.0以上版本的库。

RotatedRect rr;
if (vpfinalfit.size() > 5)
{
	rr = fitEllipseAMS(vpfinalfit);
	//rr = fitEllipse(vpfinalfit);
	ellipse(src, rr, Scalar(0, 255, 255), 1);  //画椭圆
}

但是不能盲目升级,就是因为这个原因,笔者把原来开发的Android程序jni使用的opencv版本从3.2升级到了4.0,导致队友原来开发的视觉算法编译通不过来,不过没有什么,从opencv3到opencv4大部分是是简单的变量命名替换,改起来还是比较轻松。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

元点机智

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值