Grabcuts图割算法

来源:互联网 发布:linux中国时区 编辑:程序博客网 时间:2024/04/30 08:31

原文地址:http://blog.csdn.net/wangyaninglm/article/details/44151213

1.绪论

图切割算法是组合图论的经典算法之一。近年来,许多学者将其应用到图像和视频分割中,取得了很好的效果。本文简单介绍了图切算法和交互式图像分割技术,以及图切算法在交互式图像分割中的应用。

 

图像分割指图像分成各具特性的区域并提取出感兴趣目标的技术和过程,它是由图像处理到图像分析的关键步骤,是一种基本的计算机视觉技术。只有在图像分割的基础上才能对目标进行特征提取和参数测量,使得更高层的图像分析和理解成为可能。因此对图像分割方法的研究具有十分重要的意义。

 

图像分割技术的研究已有几十年的历史,但至今人们并不能找到通用的方法能够适合于所有类型的图像。常用的图像分割技术可划分为四类:特征阈值或聚类、边缘检测、区域生长或区域提取。虽然这些方法分割灰度图像效果较好,但用于彩色图像的分割往往达不到理想的效果。

 

交互式图像分割是指,首先由用户以某种交互手段指定图像的部分前景与部分背景,然后算法以用户的输入作为分割的约束条件自动地计算出满足约束条件下的最佳分割。典型的交互手段包括用一把画刷在前景和背景处各画几笔(如[1][4]等)以及在前景的周围画一个方框(如[2])等。

 

基于图切算法的图像分割技术是近年来国际上图像分割领域的一个新的研究热点。该类方法将图像映射为赋权无向图,把像素视作节点,利用最小切割得到图像的最佳分割。


 2.几种改进算法

 

  • Graph Cut[1]算法是一种直接基于图切算法的图像分割技术。它仅需要在前景和背景处各画几笔作为输入,算法将建立各个像素点与前景背景相似度的赋权图,并通过求解最小切割区分前景和背景。

 

  • Grabcut[2]算法方法的用户交互量很少,仅仅需要指定一个包含前景的矩形,随后用基于图切算法在图像中提取前景。

 

  • Lazy Snapping[4]系统则是对[1]的改进。通过预计算和聚类技术,该方法提供了一个即时反馈的平台,方便用户进行交互分割。

 

文档说明:

http://download.csdn.net/detail/wangyaninglm/8484301

 

3.代码实现效果


 

 

 

 

 

graphcuts代码:

http://download.csdn.net/detail/wangyaninglm/8484243

 

 

 

 

ICCV'2001论文"Interactive graph cuts for optimal boundary and region segmentation of objects in N-D images"。

Graph Cut方法是基于颜色统计采样的方法,因此对前背景相差较大的图像效果较佳。

同时,比例系数lambda的调节直接影响到最终的分割效果。

 

 

grabcut代码:

 

[cpp] view plain copy print?在CODE上查看代码片派生到我的代码片
  1. // Grabcut.cpp : 定义控制台应用程序的入口点。  
  2. //  
  3.   
  4. #include "stdafx.h"  
  5.   
  6.   
  7.   
  8.   
  9. #include "opencv2/highgui/highgui.hpp"  
  10. #include "opencv2/imgproc/imgproc.hpp"  
  11.   
  12. #include <iostream>  
  13.   
  14. #include "ComputeTime.h"  
  15. #include "windows.h"  
  16.   
  17. using namespace std;  
  18. using namespace cv;  
  19.   
  20. static void help()  
  21. {  
  22.     cout << "\nThis program demonstrates GrabCut segmentation -- select an object in a region\n"  
  23.         "and then grabcut will attempt to segment it out.\n"  
  24.         "Call:\n"  
  25.         "./grabcut <image_name>\n"  
  26.         "\nSelect a rectangular area around the object you want to segment\n" <<  
  27.         "\nHot keys: \n"  
  28.         "\tESC - quit the program\n"  
  29.         "\tr - restore the original image\n"  
  30.         "\tn - next iteration\n"  
  31.         "\n"  
  32.         "\tleft mouse button - set rectangle\n"  
  33.         "\n"  
  34.         "\tCTRL+left mouse button - set GC_BGD pixels\n"  
  35.         "\tSHIFT+left mouse button - set CG_FGD pixels\n"  
  36.         "\n"  
  37.         "\tCTRL+right mouse button - set GC_PR_BGD pixels\n"  
  38.         "\tSHIFT+right mouse button - set CG_PR_FGD pixels\n" << endl;  
  39. }  
  40.   
  41. const Scalar RED = Scalar(0,0,255);  
  42. const Scalar PINK = Scalar(230,130,255);  
  43. const Scalar BLUE = Scalar(255,0,0);  
  44. const Scalar LIGHTBLUE = Scalar(255,255,160);  
  45. const Scalar GREEN = Scalar(0,255,0);  
  46.   
  47. const int BGD_KEY = CV_EVENT_FLAG_CTRLKEY;  //Ctrl键  
  48. const int FGD_KEY = CV_EVENT_FLAG_SHIFTKEY; //Shift键  
  49.   
  50. static void getBinMask( const Mat& comMask, Mat& binMask )  
  51. {  
  52.     if( comMask.empty() || comMask.type()!=CV_8UC1 )  
  53.         CV_Error( CV_StsBadArg, "comMask is empty or has incorrect type (not CV_8UC1)" );  
  54.     if( binMask.empty() || binMask.rows!=comMask.rows || binMask.cols!=comMask.cols )  
  55.         binMask.create( comMask.size(), CV_8UC1 );  
  56.     binMask = comMask & 1;  //得到mask的最低位,实际上是只保留确定的或者有可能的前景点当做mask  
  57. }  
  58.   
  59. class GCApplication  
  60. {  
  61. public:  
  62.     enum{ NOT_SET = 0, IN_PROCESS = 1, SET = 2 };  
  63.     static const int radius = 2;  
  64.     static const int thickness = -1;  
  65.   
  66.     void reset();  
  67.     void setImageAndWinName( const Mat& _image, const string& _winName );  
  68.     void showImage() const;  
  69.     void mouseClick( int event, int x, int y, int flags, void* param );  
  70.     int nextIter();  
  71.     int getIterCount() const { return iterCount; }  
  72. private:  
  73.     void setRectInMask();  
  74.     void setLblsInMask( int flags, Point p, bool isPr );  
  75.   
  76.     const string* winName;  
  77.     const Mat* image;  
  78.     Mat mask;  
  79.     Mat bgdModel, fgdModel;  
  80.   
  81.     uchar rectState, lblsState, prLblsState;  
  82.     bool isInitialized;  
  83.   
  84.     Rect rect;  
  85.     vector<Point> fgdPxls, bgdPxls, prFgdPxls, prBgdPxls;  
  86.     int iterCount;  
  87. };  
  88.   
  89. /*给类的变量赋值*/  
  90. void GCApplication::reset()  
  91. {  
  92.     if( !mask.empty() )  
  93.         mask.setTo(Scalar::all(GC_BGD));  
  94.     bgdPxls.clear(); fgdPxls.clear();  
  95.     prBgdPxls.clear();  prFgdPxls.clear();  
  96.   
  97.     isInitialized = false;  
  98.     rectState = NOT_SET;    //NOT_SET == 0  
  99.     lblsState = NOT_SET;  
  100.     prLblsState = NOT_SET;  
  101.     iterCount = 0;  
  102. }  
  103.   
  104. /*给类的成员变量赋值而已*/  
  105. void GCApplication::setImageAndWinName( const Mat& _image, const string& _winName  )  
  106. {  
  107.     if( _image.empty() || _winName.empty() )  
  108.         return;  
  109.     image = &_image;  
  110.     winName = &_winName;  
  111.     mask.create( image->size(), CV_8UC1);  
  112.     reset();  
  113. }  
  114.   
  115. /*显示4个点,一个矩形和图像内容,因为后面的步骤很多地方都要用到这个函数,所以单独拿出来*/  
  116. void GCApplication::showImage() const  
  117. {  
  118.     if( image->empty() || winName->empty() )  
  119.         return;  
  120.   
  121.     Mat res;  
  122.     Mat binMask;  
  123.     if( !isInitialized )  
  124.         image->copyTo( res );  
  125.     else  
  126.     {  
  127.         getBinMask( mask, binMask );  
  128.         image->copyTo( res, binMask );  //按照最低位是0还是1来复制,只保留跟前景有关的图像,比如说可能的前景,可能的背景  
  129.     }  
  130.   
  131.     vector<Point>::const_iterator it;  
  132.     /*下面4句代码是将选中的4个点用不同的颜色显示出来*/  
  133.     for( it = bgdPxls.begin(); it != bgdPxls.end(); ++it )  //迭代器可以看成是一个指针  
  134.         circle( res, *it, radius, BLUE, thickness );  
  135.     for( it = fgdPxls.begin(); it != fgdPxls.end(); ++it )  //确定的前景用红色表示  
  136.         circle( res, *it, radius, RED, thickness );  
  137.     for( it = prBgdPxls.begin(); it != prBgdPxls.end(); ++it )  
  138.         circle( res, *it, radius, LIGHTBLUE, thickness );  
  139.     for( it = prFgdPxls.begin(); it != prFgdPxls.end(); ++it )  
  140.         circle( res, *it, radius, PINK, thickness );  
  141.   
  142.     /*画矩形*/  
  143.     if( rectState == IN_PROCESS || rectState == SET )  
  144.         rectangle( res, Point( rect.x, rect.y ), Point(rect.x + rect.width, rect.y + rect.height ), GREEN, 2);  
  145.   
  146.     imshow( *winName, res );  
  147. }  
  148.   
  149. /*该步骤完成后,mask图像中rect内部是3,外面全是0*/  
  150. void GCApplication::setRectInMask()  
  151. {  
  152.     assert( !mask.empty() );  
  153.     mask.setTo( GC_BGD );   //GC_BGD == 0  
  154.     rect.x = max(0, rect.x);  
  155.     rect.y = max(0, rect.y);  
  156.     rect.width = min(rect.width, image->cols-rect.x);  
  157.     rect.height = min(rect.height, image->rows-rect.y);  
  158.     (mask(rect)).setTo( Scalar(GC_PR_FGD) );    //GC_PR_FGD == 3,矩形内部,为可能的前景点  
  159. }  
  160.   
  161. void GCApplication::setLblsInMask( int flags, Point p, bool isPr )  
  162. {  
  163.     vector<Point> *bpxls, *fpxls;  
  164.     uchar bvalue, fvalue;  
  165.     if( !isPr ) //确定的点  
  166.     {  
  167.         bpxls = &bgdPxls;  
  168.         fpxls = &fgdPxls;  
  169.         bvalue = GC_BGD;    //0  
  170.         fvalue = GC_FGD;    //1  
  171.     }  
  172.     else    //概率点  
  173.     {  
  174.         bpxls = &prBgdPxls;  
  175.         fpxls = &prFgdPxls;  
  176.         bvalue = GC_PR_BGD; //2  
  177.         fvalue = GC_PR_FGD; //3  
  178.     }  
  179.     if( flags & BGD_KEY )  
  180.     {  
  181.         bpxls->push_back(p);  
  182.         circle( mask, p, radius, bvalue, thickness );   //该点处为2  
  183.     }  
  184.     if( flags & FGD_KEY )  
  185.     {  
  186.         fpxls->push_back(p);  
  187.         circle( mask, p, radius, fvalue, thickness );   //该点处为3  
  188.     }  
  189. }  
  190.   
  191. /*鼠标响应函数,参数flags为CV_EVENT_FLAG的组合*/  
  192. void GCApplication::mouseClick( int event, int x, int y, int flags, void* )  
  193. {  
  194.     // TODO add bad args check  
  195.     switch( event )  
  196.     {  
  197.     case CV_EVENT_LBUTTONDOWN: // set rect or GC_BGD(GC_FGD) labels  
  198.         {  
  199.             bool isb = (flags & BGD_KEY) != 0,  
  200.                 isf = (flags & FGD_KEY) != 0;  
  201.             if( rectState == NOT_SET && !isb && !isf )//只有左键按下时  
  202.             {  
  203.                 rectState = IN_PROCESS; //表示正在画矩形  
  204.                 rect = Rect( x, y, 1, 1 );  
  205.             }  
  206.             if ( (isb || isf) && rectState == SET ) //按下了alt键或者shift键,且画好了矩形,表示正在画前景背景点  
  207.                 lblsState = IN_PROCESS;  
  208.         }  
  209.         break;  
  210.     case CV_EVENT_RBUTTONDOWN: // set GC_PR_BGD(GC_PR_FGD) labels  
  211.         {  
  212.             bool isb = (flags & BGD_KEY) != 0,  
  213.                 isf = (flags & FGD_KEY) != 0;  
  214.             if ( (isb || isf) && rectState == SET ) //正在画可能的前景背景点  
  215.                 prLblsState = IN_PROCESS;  
  216.         }  
  217.         break;  
  218.     case CV_EVENT_LBUTTONUP:  
  219.         if( rectState == IN_PROCESS )  
  220.         {  
  221.             rect = Rect( Point(rect.x, rect.y), Point(x,y) );   //矩形结束  
  222.             rectState = SET;  
  223.             setRectInMask();  
  224.             assert( bgdPxls.empty() && fgdPxls.empty() && prBgdPxls.empty() && prFgdPxls.empty() );  
  225.             showImage();  
  226.         }  
  227.         if( lblsState == IN_PROCESS )   //已画了前后景点  
  228.         {  
  229.             setLblsInMask(flags, Point(x,y), false);    //画出前景点  
  230.             lblsState = SET;  
  231.             showImage();  
  232.         }  
  233.         break;  
  234.     case CV_EVENT_RBUTTONUP:  
  235.         if( prLblsState == IN_PROCESS )  
  236.         {  
  237.             setLblsInMask(flags, Point(x,y), true); //画出背景点  
  238.             prLblsState = SET;  
  239.             showImage();  
  240.         }  
  241.         break;  
  242.     case CV_EVENT_MOUSEMOVE:  
  243.         if( rectState == IN_PROCESS )  
  244.         {  
  245.             rect = Rect( Point(rect.x, rect.y), Point(x,y) );  
  246.             assert( bgdPxls.empty() && fgdPxls.empty() && prBgdPxls.empty() && prFgdPxls.empty() );  
  247.             showImage();    //不断的显示图片  
  248.         }  
  249.         else if( lblsState == IN_PROCESS )  
  250.         {  
  251.             setLblsInMask(flags, Point(x,y), false);  
  252.             showImage();  
  253.         }  
  254.         else if( prLblsState == IN_PROCESS )  
  255.         {  
  256.             setLblsInMask(flags, Point(x,y), true);  
  257.             showImage();  
  258.         }  
  259.         break;  
  260.     }  
  261. }  
  262.   
  263. /*该函数进行grabcut算法,并且返回算法运行迭代的次数*/  
  264. int GCApplication::nextIter()  
  265. {  
  266.     if( isInitialized )  
  267.         //使用grab算法进行一次迭代,参数2为mask,里面存的mask位是:矩形内部除掉那些可能是背景或者已经确定是背景后的所有的点,且mask同时也为输出  
  268.         //保存的是分割后的前景图像  
  269.         grabCut( *image, mask, rect, bgdModel, fgdModel, 1 );  
  270.     else  
  271.     {  
  272.         if( rectState != SET )  
  273.             return iterCount;  
  274.   
  275.         if( lblsState == SET || prLblsState == SET )  
  276.             grabCut( *image, mask, rect, bgdModel, fgdModel, 1, GC_INIT_WITH_MASK );  
  277.         else  
  278.             grabCut( *image, mask, rect, bgdModel, fgdModel, 1, GC_INIT_WITH_RECT );  
  279.   
  280.         isInitialized = true;  
  281.     }  
  282.     iterCount++;  
  283.   
  284.     bgdPxls.clear(); fgdPxls.clear();  
  285.     prBgdPxls.clear(); prFgdPxls.clear();  
  286.   
  287.     return iterCount;  
  288. }  
  289.   
  290. GCApplication gcapp;  
  291.   
  292. static void on_mouse( int event, int x, int y, int flags, void* param )  
  293. {  
  294.     gcapp.mouseClick( event, x, y, flags, param );  
  295. }  
  296.   
  297. int main( int argc, char** argv )  
  298. {  
  299.     string filename;  
  300.     cout<<" Grabcuts ! \n";  
  301.     cout<<"input image name:  "<<endl;  
  302.     cin>>filename;  
  303.   
  304.       
  305.     Mat image = imread( filename, 1 );  
  306.     if( image.empty() )  
  307.     {  
  308.         cout << "\n Durn, couldn't read image filename " << filename << endl;  
  309.         return 1;  
  310.     }  
  311.   
  312.     help();  
  313.   
  314.     const string winName = "image";  
  315.     cvNamedWindow( winName.c_str(), CV_WINDOW_AUTOSIZE );  
  316.     cvSetMouseCallback( winName.c_str(), on_mouse, 0 );  
  317.   
  318.     gcapp.setImageAndWinName( image, winName );  
  319.     gcapp.showImage();  
  320.   
  321.     for(;;)  
  322.     {  
  323.         int c = cvWaitKey(0);  
  324.         switch( (char) c )  
  325.         {  
  326.         case '\x1b':  
  327.             cout << "Exiting ..." << endl;  
  328.             goto exit_main;  
  329.         case 'r':  
  330.             cout << endl;  
  331.             gcapp.reset();  
  332.             gcapp.showImage();  
  333.             break;  
  334.         case 'n':  
  335.             ComputeTime ct ;  
  336.             ct.Begin();  
  337.               
  338.             int iterCount = gcapp.getIterCount();  
  339.             cout << "<" << iterCount << "... ";  
  340.             int newIterCount = gcapp.nextIter();  
  341.             if( newIterCount > iterCount )  
  342.             {  
  343.                 gcapp.showImage();  
  344.                 cout << iterCount << ">" << endl;  
  345.                 cout<<"运行时间:  "<<ct.End()<<endl;  
  346.             }  
  347.             else  
  348.                 cout << "rect must be determined>" << endl;  
  349.             break;  
  350.         }  
  351.     }  
  352.   
  353. exit_main:  
  354.     cvDestroyWindow( winName.c_str() );  
  355.     return 0;  
  356. }  


 lazy snapping代码实现:

 

[cpp] view plain copy print?在CODE上查看代码片派生到我的代码片
  1. // LazySnapping.cpp : 定义控制台应用程序的入口点。  
  2. //  
  3. /* author: zhijie Lee 
  4.  * home page: lzhj.me 
  5.  * 2012-02-06 
  6.  */  
  7. #include "stdafx.h"  
  8. #include <cv.h>  
  9. #include <highgui.h>  
  10. #include "graph.h"  
  11. #include <vector>  
  12. #include <iostream>  
  13. #include <cmath>  
  14. #include <string>  
  15.   
  16. using namespace std;  
  17.   
  18. typedef Graph<float,float,float> GraphType;  
  19.   
  20. class LasySnapping  
  21. {  
  22.       
  23. public :  
  24.     LasySnapping();  
  25.   
  26.     ~LasySnapping()  
  27.     {   
  28.         if(graph)  
  29.         {  
  30.             delete graph;  
  31.         }  
  32.     };  
  33. private :  
  34.     vector<CvPoint> forePts;  
  35.     vector<CvPoint> backPts;  
  36.     IplImage* image;  
  37.     // average color of foreground points  
  38.     unsigned char avgForeColor[3];  
  39.     // average color of background points  
  40.     unsigned char avgBackColor[3];  
  41. public :  
  42.     void setImage(IplImage* image)  
  43.     {  
  44.         this->image = image;  
  45.         graph = new GraphType(image->width*image->height,image->width*image->height*2);  
  46.     }  
  47.     // include-pen locus  
  48.     void setForegroundPoints(vector<CvPoint> pts)  
  49.     {  
  50.         forePts.clear();  
  51.         for(int i =0; i< pts.size(); i++)  
  52.         {  
  53.             if(!isPtInVector(pts[i],forePts))  
  54.             {  
  55.                 forePts.push_back(pts[i]);  
  56.             }  
  57.         }  
  58.         if(forePts.size() == 0)  
  59.         {  
  60.             return;  
  61.         }  
  62.         int sum[3] = {0};  
  63.         for(int i =0; i < forePts.size(); i++)  
  64.         {  
  65.             unsigned char* p = (unsigned char*)image->imageData + forePts[i].x * 3   
  66.                 + forePts[i].y*image->widthStep;  
  67.             sum[0] += p[0];  
  68.             sum[1] += p[1];  
  69.             sum[2] += p[2];              
  70.         }  
  71.         cout<<sum[0]<<" " <<forePts.size()<<endl;  
  72.         avgForeColor[0] = sum[0]/forePts.size();  
  73.         avgForeColor[1] = sum[1]/forePts.size();  
  74.         avgForeColor[2] = sum[2]/forePts.size();  
  75.     }  
  76.     // exclude-pen locus  
  77.     void setBackgroundPoints(vector<CvPoint> pts)  
  78.     {  
  79.         backPts.clear();  
  80.         for(int i =0; i< pts.size(); i++)  
  81.         {  
  82.             if(!isPtInVector(pts[i],backPts))  
  83.             {  
  84.                 backPts.push_back(pts[i]);  
  85.             }  
  86.         }  
  87.         if(backPts.size() == 0)  
  88.         {  
  89.             return;  
  90.         }  
  91.         int sum[3] = {0};  
  92.         for(int i =0; i < backPts.size(); i++)  
  93.         {  
  94.             unsigned char* p = (unsigned char*)image->imageData + backPts[i].x * 3 +   
  95.                 backPts[i].y*image->widthStep;  
  96.             sum[0] += p[0];  
  97.             sum[1] += p[1];  
  98.             sum[2] += p[2];              
  99.         }  
  100.         avgBackColor[0] = sum[0]/backPts.size();  
  101.         avgBackColor[1] = sum[1]/backPts.size();  
  102.         avgBackColor[2] = sum[2]/backPts.size();  
  103.     }  
  104.   
  105.     // return maxflow of graph  
  106.     int runMaxflow();  
  107.     // get result, a grayscale mast image indicating forground by 255 and background by 0  
  108.     IplImage* getImageMask();  
  109.   
  110. private :  
  111.   
  112.     float colorDistance(unsigned char* color1, unsigned char* color2);  
  113.     float minDistance(unsigned char* color, vector<CvPoint> points);  
  114.     bool isPtInVector(CvPoint pt, vector<CvPoint> points);  
  115.     void getE1(unsigned char* color,float* energy);  
  116.     float getE2(unsigned char* color1,unsigned char* color2);  
  117.       
  118.     GraphType *graph;      
  119. };  
  120.   
  121. LasySnapping::LasySnapping()  
  122. {  
  123.     graph = NULL;  
  124.     avgForeColor[0] = 0;  
  125.     avgForeColor[1] = 0;  
  126.     avgForeColor[2] = 0;  
  127.   
  128.     avgBackColor[0] = 0;  
  129.     avgBackColor[1] = 0;  
  130.     avgBackColor[2] = 0;  
  131.   
  132.       
  133. }  
  134.   
  135.   
  136.   
  137. float LasySnapping::colorDistance(unsigned char* color1, unsigned char* color2)  
  138. {  
  139.       
  140.     return sqrt(((float)color1[0]-(float)color2[0])*((float)color1[0]-(float)color2[0])+  
  141.         ((float)color1[1]-(float)color2[1])*((float)color1[1]-(float)color2[1])+  
  142.         ((float)color1[2]-(float)color2[2])*((float)color1[2]-(float)color2[2]));      
  143. }  
  144.   
  145. float LasySnapping::minDistance(unsigned char* color, vector<CvPoint> points)  
  146. {  
  147.     float distance = -1;  
  148.     for(int i =0 ; i < points.size(); i++)  
  149.     {  
  150.         unsigned char* p = (unsigned char*)image->imageData + points[i].y * image->widthStep +   
  151.             points[i].x * image->nChannels;  
  152.         float d = colorDistance(p,color);  
  153.         if(distance < 0 )  
  154.         {  
  155.             distance = d;  
  156.         }  
  157.         else  
  158.         {  
  159.             if(distance > d)  
  160.             {  
  161.                 distance = d;  
  162.             }  
  163.         }  
  164.     }  
  165.   
  166.     return distance;  
  167. }  
  168.   
  169. bool LasySnapping::isPtInVector(CvPoint pt, vector<CvPoint> points)  
  170. {  
  171.     for(int i =0 ; i < points.size(); i++)  
  172.     {  
  173.         if(pt.x == points[i].x && pt.y == points[i].y)  
  174.         {  
  175.             return true;  
  176.         }  
  177.     }  
  178.     return false;  
  179. }  
  180. void LasySnapping::getE1(unsigned char* color,float* energy)  
  181. {  
  182.     // average distance  
  183.     float df = colorDistance(color,avgForeColor);  
  184.     float db = colorDistance(color,avgBackColor);  
  185.     // min distance from background points and forground points  
  186.     // float df = minDistance(color,forePts);  
  187.     // float db = minDistance(color,backPts);  
  188.     energy[0] = df/(db+df);  
  189.     energy[1] = db/(db+df);  
  190. }  
  191.   
  192. float LasySnapping::getE2(unsigned char* color1,unsigned char* color2)  
  193. {  
  194.     const float EPSILON = 0.01;  
  195.     float lambda = 100;  
  196.     return lambda/(EPSILON+  
  197.         (color1[0]-color2[0])*(color1[0]-color2[0])+  
  198.         (color1[1]-color2[1])*(color1[1]-color2[1])+  
  199.         (color1[2]-color2[2])*(color1[2]-color2[2]));  
  200. }  
  201.   
  202. int LasySnapping::runMaxflow()  
  203. {     
  204.     const float INFINNITE_MAX = 1e10;  
  205.     int indexPt = 0;  
  206.     for(int h = 0; h < image->height; h ++)  
  207.     {  
  208.         unsigned char* p = (unsigned char*)image->imageData + h *image->widthStep;  
  209.         for(int w = 0; w < image->width; w ++)  
  210.         {  
  211.             // calculate energe E1  
  212.             float e1[2]={0};  
  213.             if(isPtInVector(cvPoint(w,h),forePts))  
  214.             {  
  215.                 e1[0] =0;  
  216.                 e1[1] = INFINNITE_MAX;  
  217.             }  
  218.             else if  
  219.                 (isPtInVector(cvPoint(w,h),backPts))  
  220.             {  
  221.                 e1[0] = INFINNITE_MAX;  
  222.                 e1[1] = 0;  
  223.             }  
  224.             else   
  225.             {  
  226.                 getE1(p,e1);  
  227.             }  
  228.   
  229.             // add node  
  230.             graph->add_node();  
  231.             graph->add_tweights(indexPt, e1[0],e1[1]);  
  232.   
  233.             // add edge, 4-connect  
  234.             if(h > 0 && w > 0)  
  235.             {  
  236.                 float e2 = getE2(p,p-3);  
  237.                 graph->add_edge(indexPt,indexPt-1,e2,e2);  
  238.                 e2 = getE2(p,p-image->widthStep);  
  239.                 graph->add_edge(indexPt,indexPt-image->width,e2,e2);  
  240.             }  
  241.               
  242.             p+= 3;  
  243.             indexPt ++;              
  244.         }  
  245.     }  
  246.       
  247.     return graph->maxflow();  
  248. }  
  249.   
  250. IplImage* LasySnapping::getImageMask()  
  251. {  
  252.     IplImage* gray = cvCreateImage(cvGetSize(image),8,1);   
  253.     int indexPt =0;  
  254.     for(int h =0; h < image->height; h++)  
  255.     {  
  256.         unsigned char* p = (unsigned char*)gray->imageData + h*gray->widthStep;  
  257.         for(int w =0 ;w <image->width; w++)  
  258.         {  
  259.             if (graph->what_segment(indexPt) == GraphType::SOURCE)  
  260.             {  
  261.                 *p = 0;  
  262.             }  
  263.             else  
  264.             {  
  265.                 *p = 255;  
  266.             }  
  267.   
  268.             p++;  
  269.             indexPt ++;  
  270.         }  
  271.     }  
  272.     return gray;  
  273. }  
  274.   
  275. // global  
  276. vector<CvPoint> forePts;  
  277. vector<CvPoint> backPts;  
  278. int currentMode = 0;// indicate foreground or background, foreground as default  
  279. CvScalar paintColor[2] = {CV_RGB(0,0,255),CV_RGB(255,0,0)};  
  280.   
  281. IplImage* image = NULL;  
  282. char* winName = "lazySnapping";  
  283. IplImage* imageDraw = NULL;  
  284. const int SCALE = 4;  
  285.   
  286. void on_mouse( int event, int x, int y, int flags, void* )  
  287. {      
  288.     if( event == CV_EVENT_LBUTTONUP )  
  289.     {  
  290.         if(backPts.size() == 0 && forePts.size() == 0)  
  291.         {  
  292.             return;  
  293.         }  
  294.         LasySnapping ls;  
  295.         IplImage* imageLS = cvCreateImage(cvSize(image->width/SCALE,image->height/SCALE),  
  296.             8,3);  
  297.         cvResize(image,imageLS);  
  298.         ls.setImage(imageLS);  
  299.         ls.setBackgroundPoints(backPts);  
  300.         ls.setForegroundPoints(forePts);  
  301.         ls.runMaxflow();  
  302.         IplImage* mask = ls.getImageMask();  
  303.         IplImage* gray = cvCreateImage(cvGetSize(image),8,1);  
  304.         cvResize(mask,gray);  
  305.         // edge  
  306.         cvCanny(gray,gray,50,150,3);  
  307.           
  308.         IplImage* showImg = cvCloneImage(imageDraw);  
  309.         for(int h =0; h < image->height; h ++)  
  310.         {  
  311.             unsigned char* pgray = (unsigned char*)gray->imageData + gray->widthStep*h;  
  312.             unsigned char* pimage = (unsigned char*)showImg->imageData + showImg->widthStep*h;  
  313.             for(int width  =0; width < image->width; width++)  
  314.             {  
  315.                 if(*pgray++ != 0 )  
  316.                 {  
  317.                     pimage[0] = 0;  
  318.                     pimage[1] = 255;  
  319.                     pimage[2] = 0;  
  320.                 }  
  321.                 pimage+=3;                  
  322.             }  
  323.         }  
  324.         cvSaveImage("t.bmp",showImg);  
  325.         cvShowImage(winName,showImg);  
  326.         cvReleaseImage(&imageLS);  
  327.         cvReleaseImage(&mask);  
  328.         cvReleaseImage(&showImg);  
  329.         cvReleaseImage(&gray);  
  330.     }  
  331.     else if( event == CV_EVENT_LBUTTONDOWN )  
  332.     {  
  333.   
  334.     }  
  335.     else if( event == CV_EVENT_MOUSEMOVE && (flags & CV_EVENT_FLAG_LBUTTON))  
  336.     {  
  337.         CvPoint pt = cvPoint(x,y);  
  338.         if(currentMode == 0)  
  339.         {//foreground  
  340.             forePts.push_back(cvPoint(x/SCALE,y/SCALE));  
  341.         }  
  342.         else  
  343.         {//background  
  344.             backPts.push_back(cvPoint(x/SCALE,y/SCALE));  
  345.         }  
  346.         cvCircle(imageDraw,pt,2,paintColor[currentMode]);  
  347.         cvShowImage(winName,imageDraw);  
  348.     }  
  349. }  
  350. int main(int argc, char** argv)  
  351. {     
  352.     //if(argc != 2)  
  353.     //{  
  354.      //   cout<<"command : lazysnapping inputImage"<<endl;  
  355.      //   return 0;  
  356.    // }  
  357.   
  358.     string image_name;  
  359.     cout<<"input image name: "<<endl;  
  360.     cin>>image_name;  
  361.   
  362.     cvNamedWindow(winName,1);  
  363.     cvSetMouseCallback( winName, on_mouse, 0);  
  364.       
  365.     image = cvLoadImage(image_name.c_str(),CV_LOAD_IMAGE_COLOR);  
  366.     imageDraw = cvCloneImage(image);  
  367.     cvShowImage(winName, image);  
  368.     for(;;)  
  369.     {  
  370.         int c = cvWaitKey(0);  
  371.         c = (char)c;  
  372.         if(c == 27)  
  373.         {//exit  
  374.             break;  
  375.         }  
  376.         else if(c == 'r')  
  377.         {//reset  
  378.             image = cvLoadImage(image_name.c_str(),CV_LOAD_IMAGE_COLOR);  
  379.             imageDraw = cvCloneImage(image);  
  380.             forePts.clear();  
  381.             backPts.clear();  
  382.             currentMode = 0;  
  383.             cvShowImage(winName, image);  
  384.         }  
  385.         else if(c == 'b')  
  386.         {//change to background selection  
  387.             currentMode = 1;  
  388.         }else if(c == 'f')  
  389.         {//change to foreground selection  
  390.             currentMode = 0;  
  391.         }  
  392.     }  
  393.     cvReleaseImage(&image);  
  394.     cvReleaseImage(&imageDraw);  
  395.     return 0;  
  396. }  


 

 

 

 

参考文献

[1] Y. Boykov, and M. P. Jolly, “Interactive graph cuts for optimal boundary and region segmentation ofobjects in N-D images”,Proceeding ofIEEE International Conference on Computer Vision, 1:105~112, July 2001.

[2] C. Rother, A. Blake, and V. Kolmogorov, “Grabcut – interactive foreground extractionusing iterated graph cuts”,Proceedingsof ACM SIGGRAPH 2004, 23(3):307~312, August 2004.

[3] A. Agarwala, M. Dontcheva, M. Agrawala,et al, “Interactive digital photomontage”,Proceedings of ACM SIGGRAPH 2004, 23(3):294~302, August 2004.

[4] Y. Li, J. Sun, C. Tang,et al, “Interacting withimages: Lazy snapping”,Proceedingsof ACM SIGGRAPH 2004, 23(3):303~308, August 2004.

[5] A. Blake, C. Rother, M. Brown,et al, “Interactive ImageSegmentation using an adaptive GMMRF model”.Proceedings of European Conference on Computer Vision, pp. 428~441,May 2004.

[6] V. Kwatra, A. Schodl, I. Essa,et al, “Graphcut Textures:Image and Video Synthesis Using Graph Cuts”.Proceedings of ACM Siggraph 2003, pp.277~286, Augst 2003.




部分代码与文档是早些时候收集的,出处找不到了,还请原作者看到后联系注明。

 


0 0
原创粉丝点击