OPENCVGrabCut详细讲稿.docVIP

一、GrabCut函数使用 在OpenCV的源码目录的samples的文件夹下，有grabCut的使用例程，请参考： opencv\samples\cpp\grabcut.cpp。 而grabCut函数的API说明如下： void cv::grabCut( InputArray _img, InputOutputArray _mask, Rect rect, InputOutputArray _bgdModel, InputOutputArray _fgdModel, int iterCount, int mode ) 二、GrabCut源码解读 其中源码包含了gcgraph.hpp这个构建图和max flow/min cut算法的实现文件，这个文件暂时没有解读，后面再更新了。 [cpp] view plaincopyprint? #include precomp.hpp #include gcgraph.hpp #include using namespace cv; class GMM { public: static const int componentsCount = 5; GMM( Mat _model ); double operator()( const Vec3d color ) const; double operator()( int ci, const Vec3d color ) const; int whichComponent( const Vec3d color ) const; void initLearning(); void addSample( int ci, const Vec3d color ); void endLearning(); private: void calcInverseCovAndDeterm( int ci ); Mat model; double* coefs; double* mean; double* cov; double inverseCovs[componentsCount][3][3]; //协方差的逆矩阵 double covDeterms[componentsCount]; //协方差的行列式 double sums[componentsCount][3]; double prods[componentsCount][3][3]; int sampleCounts[componentsCount]; int totalSampleCount; }; //背景和前景各有一个对应的GMM（混合高斯模型） GMM::GMM( Mat _model ) { //一个像素的（唯一对应）高斯模型的参数个数或者说一个高斯模型的参数个数 //一个像素RGB三个通道值，故3个均值，3*3个协方差，共用一个权值 const int modelSize = 3 + 9 + 1; if( _model.empty() ) { //一个GMM共有componentsCount个高斯模型，一个高斯模型有modelSize个模型参数 _model.create( 1, modelSize*componentsCount, CV_64FC1 ); _model.setTo(Scalar(0)); } else if( (_model.type() != CV_64FC1) || (_model.rows != 1) || (_model.cols != modelSize*componentsCount) ) CV_Error( CV_StsBadArg, _model must have CV_64FC1 type, rows == 1 and cols == 13*componentsCount );