opencv EM

 void myEM(int clusterNum, vector<Point>samplesPtVec, int cov_mat_type, vector<int> &LabelsVec,Mat &mean,vector<Mat> &varMatVec)
 {
	 int nsamplesNum = samplesPtVec.size();
	 Mat samples( nsamplesNum, 2, CV_32FC1 );

	 for( int i = 0; i < nsamplesNum; i++ )
	 {
		 samples.at<float>(i,0) = samplesPtVec[i].x;
		 samples.at<float>(i,1) = samplesPtVec[i].y;
	 }

	 Mat labels;
	 CvEM em_model;
	 CvEMParams params;
	 // initialize model parameters
	 params.covs      = NULL;
	 params.means     = NULL;
	 params.weights   = NULL;
	 params.probs     = NULL;
	 params.nclusters = clusterNum;
	 params.cov_mat_type  = cov_mat_type;//CvEM::COV_MAT_GENERIC;//COV_MAT_DIAGONAL COV_MAT_GENERIC  COV_MAT_SPHERICAL
	 params.start_step         = CvEM::START_AUTO_STEP;
	 params.term_crit.max_iter = 500;
	 params.term_crit.epsilon   = 0.01;
	 params.term_crit.type       = CV_TERMCRIT_ITER|CV_TERMCRIT_EPS;

	 // cluster the data
	 em_model.train(samples, Mat(), params, &labels);
	 mean = em_model.getMeans();

	 //cout<<" ++++++++++++++++++"<<endl;
	 //cout<<" mean = "<<mean<<endl;
	 //vector<Mat> varMatVec;
	 em_model.getCovs(varMatVec);
	 // 	for (int i=0;i<varMatVec.size();i++)
	 // 	{
	 // 		cout<<" var = "<<varMatVec[i]<<endl;
	 // 		Mat ss,uu,vt;
	 // 		cv::SVDecomp(varMatVec[i],ss,uu,vt,SVD::FULL_UV);  
	 // 		cout<<" ss = "<<ss<<endl;
	 // 		cout<<" uu = "<<uu<<endl;
	 // 		cout<<" vt  = "<<vt<<endl;
	 // 	}

	 LabelsVec.clear();
	 LabelsVec.resize(nsamplesNum);

	 for( int i = 0; i < nsamplesNum; i++ )
	 {
		 LabelsVec[i] = labels.at<int>(i);
	 }
 }


 void drawEM(Mat rgbImg, Mat mean, vector<Mat> varMatVec,Scalar color)
 {
	 int clusterNum = varMatVec.size();

	 for (int i=0;i<clusterNum;i++)
	 {
		 Point cPt = Point(mean.at<double>(i,0),mean.at<double>(i,1));
		 cv::circle(rgbImg,cPt,4,color/*colors[i]*/,2);

		 Mat ss,uu,vt;
		 cv::SVDecomp(varMatVec[i],ss,uu,vt,SVD::FULL_UV);  

		 Size sss = Size(sqrt(ss.at<double>(0,0))*1.5, sqrt(ss.at<double>(1,0))*1.5);
		 double angle = atan(uu.at<double>(1,0)/(uu.at<double>(0,0)+1/1000000))*180/CV_PI;

		 // 		if (sss.width > sss.height*1.2)
		 // 		{
		 // 			cv::ellipse(rgbImg, cPt,sss, angle, 0, 360, color, 3);
		 // 		}
		 // 		else
		 {
			 cv::ellipse(rgbImg, cPt,sss, angle, 0, 360, color, 2);//CV_AA
		 }
		 cout<<" sss = "<<sss<<" angle = "<<angle<<endl;
	 }
 }


void drawPtsEM(Mat rgbImg,vector<Point> ptVec,Scalar color)
{
	Mat mean;
	vector<Mat> varMatVec;
	vector<int>labelsVec;
	//COV_MAT_SPHERICAL  COV_MAT_DIAGONAL COV_MAT_GENERIC
	myEM(1, ptVec,CvEM::COV_MAT_GENERIC, labelsVec, mean, varMatVec);
	drawEM(rgbImg,mean,varMatVec,color);
	//cout<<" varMatVec = "<<varMatVec[0]<<endl;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

小新识图

你的鼓励是我最大的分享动力。

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

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

打赏作者

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

抵扣说明:

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

余额充值