opencv基于混合高斯模型的图像分割

#include "stdafx.h"
#include <opencv\cv.h>
#include <opencv\highgui.h>
#include <opencv2\legacy\legacy.hpp>

int _tmain(int argc, _TCHAR* argv[])
{
    CvEM em_model;
    //CvEM em_model2;
    CvEMParams params;
    int N=;
    //设置模型参数
    params.covs      = NULL;
    params.means     = NULL;
    params.weights   = NULL;
    params.probs     = NULL;
    params.nclusters = N;
    params.cov_mat_type       = CvEM::COV_MAT_SPHERICAL;
    //params.cov_mat_type       = CvEM::COV_MAT_DIAGONAL;
    params.start_step         = CvEM::START_AUTO_STEP;
    params.term_crit.max_iter = ;
    params.term_crit.epsilon  = 0.1;
    params.term_crit.type     = CV_TERMCRIT_ITER|CV_TERMCRIT_EPS;

    IplImage* img=cvLoadImage("frame18843.jpg");//加载图像，图像放在Debug文件夹里，这里是相对路径

    cvNamedWindow( "原始图像",  ); //创建窗口
    cvShowImage( "原始图像", img  ); //显示图像
    //cvWaitKey(0); //等待按键

    int i,j;
    CvMat *samples=cvCreateMat((img->width)*(img->height),,CV_32FC1);//创建样本矩阵，CV_32FC3代表32位浮点3通道（彩色图像）
    CvMat *clusters=cvCreateMat((img->width)*(img->height),,CV_32SC1);//创建类别标记矩阵，CV_32SF1代表32位整型1通道
    cvReshape( samples, samples, ,  );
    int k=;
    for (i=;i<img->width;i++)
    {
        for (j=;j<img->height;j++)
        {
            CvScalar s;
            //获取图像各个像素点的三通道值（RGB）
            s.val[]=(float)cvGet2D(img,j,i).val[];
            s.val[]=(float)cvGet2D(img,j,i).val[];
            s.val[]=(float)cvGet2D(img,j,i).val[];
            cvSet2D(samples,k++,,s);//将像素点三通道的值按顺序排入样本矩阵
        }
    }
    int nCuster=;//聚类类别数，自己修改。
    //cvKMeans2(samples,nCuster,clusters,cvTermCriteria(CV_TERMCRIT_ITER,100,1.0));//开始聚类，迭代100次，终止误差1.0
    cvReshape( samples, samples, ,  );
    em_model.train( samples, , params, clusters );

    IplImage *bin=cvCreateImage(cvSize(img->width,img->height),IPL_DEPTH_8U,);//创建用于显示的图像，二值图像
    k=;
    int val=;
    float step=/(nCuster-);
    for (i=;i<img->width;i++)
    {
        for (j=;j<img->height;j++)
        {
            val=(int)clusters->data.i[k++];
            CvScalar s;
            s.val[]=-val*step;//这个是将不同类别取不同的像素值，
            cvSet2D(bin,j,i,s); //将每个像素点赋值
        }
    }
    cvNamedWindow( "聚类图像",  ); //创建窗口
    cvShowImage( "聚类图像", bin  ); //显示图像
    cvWaitKey(); //等待按键
    cvDestroyWindow( "原始图像" );//销毁窗口
    cvReleaseImage( &img ); //释放图像
    cvDestroyWindow( "聚类图像" );//销毁窗口
    cvReleaseImage( &bin ); //释放图像
    return ;
}