vs2015+opencv3.3.1+ maxflow-v3.01 c++实现Yuri Boykov 的Interactive Graph Cuts

出的结果不理想。 感觉是tlink的权重的计算有问题，以及参数的设置。三个可设置参数是后面的 i j k  。如果你找到了一组好参数请告诉我。

下载地址 http://download.csdn.net/download/li_haoren/10232283

Interactive Graph Cuts  for Optimal Boundary & Region Segmentation of Objects in N-D Images  code

#include "opencv2\imgproc\imgproc.hpp"  // Gaussian Blur
#include "opencv2\core\core.hpp"        // Basic OpenCV structures (cv::Mat, Scalar)
#include <opencv2\highgui\highgui.hpp> // OpenCV window I/O
#include <iostream> // for standard I/O
#include <string>   // for strings
#include <iomanip>  // for controlling float print precision
#include <sstream>  // string to number conversion
#include<vector>
#include<cmath>
#include "graph.h"  //maxflow-v3.01   下载地址 http://vision.csd.uwo.ca/code/maxflow-v3.01.zip 
typedef Graph<int, int, int> GraphType;
using namespace  cv;
using namespace  std;
 int ofexp = 5;//Lambda
double BSigma = 2;//the Sigma of B{p,q}   小边缘越清
double RSigma = 3;//the Sigma of Rp		小号
const string filename = "1.jpg";//照片
bool rButtonDown = false;
bool lButtonDown = false;
int numUsedBins = 0;
float varianceSquared = 0;
int scribbleRadius = 10;
Mat inputImg = imread(filename, 3), showImg = imread(filename, 3);

Mat fgScribbleMask;
Mat bgScribbleMask;

static void onMouse(int event, int x, int y, int, void*)
{
	//cout << "On Mouse: (" << x << "," << y << ")" <<endl;

	if (event == CV_EVENT_LBUTTONDOWN)
	{
		lButtonDown = true;

	}
	else if (event == CV_EVENT_RBUTTONDOWN)
	{
		rButtonDown = true;

	}
	else if (event == CV_EVENT_LBUTTONUP)
	{
		lButtonDown = false;
	}
	else if (event == CV_EVENT_RBUTTONUP)
	{
		rButtonDown = false;
	}
	else if (event == CV_EVENT_MOUSEMOVE)
	{
		if (rButtonDown)
		{
			// scribble the background
			circle(bgScribbleMask, Point(x, y), scribbleRadius, 255, -1);
			circle(showImg, Point(x, y), scribbleRadius, CV_RGB(0, 0, 255), -1);

		}
		else if (lButtonDown)
		{
			// scribble the foreground
			circle(showImg, Point(x, y), scribbleRadius, CV_RGB(255, 0, 0), -1);
			circle(fgScribbleMask, Point(x, y), scribbleRadius, 255, -1);
			//fgScribbleMask.at<char>(y,x)=(char)255;
			// set variables using mask
			//showImg.setTo(redColorElement,fgScribbleMask);

			//showImg.at<Vec3b>(y,x)[0] = 0;
			//showImg.at<Vec3b>(y,x)[1] = 0;
			//showImg.at<Vec3b>(y,x)[2] = 255;
		}

	}

	imshow("draw", showImg);
	imshow("fg mask", fgScribbleMask);
	imshow("bg mask", bgScribbleMask);
}
vector<double> slink(inputImg.cols*inputImg.rows),tlink(inputImg.cols*inputImg.rows);
double k=0;

GraphType *g = new GraphType(inputImg.cols*inputImg.rows, inputImg.cols*inputImg.rows*4);
inline int node1n(int i, int j) { return i*inputImg.cols + j; }
void computnlink() { 

	double weight0=0; double weight1=0; double weight2=0; double weight3=0;
	for (int i = 0; i < inputImg.rows; ++i)
		for (int j = 0; j < inputImg.cols; ++j)
		{
			g->add_node();
		}
	for (int i = 0; i < inputImg.rows; ++i)
		for (int j = 0; j < inputImg.cols; ++j)
		{	

			if (i  == 0);
			else {
				 weight0 =
						ofexp*exp(-(pow(inputImg.at<Vec3b>(i, j)[0] - inputImg.at<Vec3b>(i - 1, j)[0], 2) +
						pow(inputImg.at<Vec3b>(i, j)[1] - inputImg.at<Vec3b>(i - 1, j)[1], 2) +
						pow(inputImg.at<Vec3b>(i, j)[2] - inputImg.at<Vec3b>(i - 1, j)[2], 2)) / 6.0 / BSigma / BSigma);
				 g->add_edge(node1n(i, j), node1n(i - 1, j), weight0, weight0);
			}
			if (i + 1 ==inputImg.rows);
			else {
				weight2 =
					ofexp*exp(-(pow(inputImg.at<Vec3b>(i, j)[0] - inputImg.at<Vec3b>(i + 1, j)[0], 2) +
						pow(inputImg.at<Vec3b>(i, j)[1] - inputImg.at<Vec3b>(i + 1, j)[1], 2) +
						pow(inputImg.at<Vec3b>(i, j)[2] - inputImg.at<Vec3b>(i + 1, j)[2], 2)) / 6.0 / BSigma / BSigma);
				g->add_edge(node1n(i, j), node1n(i + 1, j), weight2, weight2);
			}
			if (j==0);
			else {
				weight3 =
					ofexp*exp(-(pow(inputImg.at<Vec3b>(i, j)[0] - inputImg.at<Vec3b>(i , j-1)[0], 2) +
						pow(inputImg.at<Vec3b>(i, j)[1] - inputImg.at<Vec3b>(i , j-1)[1], 2) +
						pow(inputImg.at<Vec3b>(i, j)[2] - inputImg.at<Vec3b>(i , j-1)[2], 2)) / 6.0 / BSigma / BSigma);
				g->add_edge(node1n(i, j), node1n(i , j-1), weight3, weight3);
			}
			if (j + 1 == inputImg.cols);
			else {
				weight1 =
					ofexp*exp(-(pow(inputImg.at<Vec3b>(i, j)[0] - inputImg.at<Vec3b>(i, j +1)[0], 2) +
						pow(inputImg.at<Vec3b>(i, j)[1] - inputImg.at<Vec3b>(i, j + 1)[1], 2) +
						pow(inputImg.at<Vec3b>(i, j)[2] - inputImg.at<Vec3b>(i, j + 1)[2], 2)) / 6.0 / BSigma / BSigma);
				g->add_edge(node1n(i, j), node1n(i, j + 1), weight1, weight1);
			}
			if (k < weight1+ weight0+ weight3+ weight2)
				k = weight1 + weight0 + weight3 + weight2;
		}
	k += 1;
}

Mat greyimg = imread(filename, 0); double gray[256]; double Bgray[256]; double Fgray[256]; double weight1 = 0; double weight2 = 0;
/*weight有问题改动*/void computtlink() {

	for (int i = 0; i < greyimg.rows*greyimg.cols; i++) {
		++gray[greyimg.at<uchar>(i)];
		if(bgScribbleMask.at<uchar>(i))
		++Bgray[greyimg.at<uchar>(i)];
		if (fgScribbleMask.at<uchar>(i))
		++Fgray[greyimg.at<uchar>(i)];
	}
	cout << greyimg.rows*greyimg.cols<<endl;
	for (int i = 0; i < greyimg.rows*greyimg.cols; i++) {

		weight1 = 0; weight2 = 0;
		for (int j = 0; j < 256; j++) {
			weight1 += (!gray[j]) ? 0 : ((Fgray[j] / gray[j])*exp(-1 * (pow(greyimg.at<uchar>(i) - j, 2))/2 / RSigma));
			weight2 += (!gray[j]) ? 0 : ((Bgray[j] / gray[j])*exp(-1 * (pow(greyimg.at<uchar>(i) - j, 2))/2/ RSigma));

		}
		if (bgScribbleMask.at<uchar>(i))
			{g->add_tweights(i,   /* capacities */ k, 0); continue;}
		if (fgScribbleMask.at<uchar>(i))
			{g->add_tweights(i,   /* capacities */ 0, k); continue;}

		g->add_tweights(i,   /* capacities */weight2 ,weight1   );

	}
}

int main() {
	cout << "在draw上画前景背景，左键红前景，右键蓝背景。然后在input输入s 开始运算" << endl;
	bgScribbleMask.create(2, inputImg.size, CV_8UC1);
	bgScribbleMask = 0;
	fgScribbleMask.create(2, inputImg.size, CV_8UC1);
	Mat result;
	result.create(2, inputImg.size, CV_8UC1);
	result = 0;
	fgScribbleMask = 0;
	namedWindow("draw");
	imshow("draw", showImg);
	namedWindow("input");
	imshow("input", inputImg); string ii[250];
	for (int i = 1; i < 250; i++)
		ii[i] = i;

		setMouseCallback("draw", onMouse, 0);
		while (1) {
			char key = waitKey(0);
			if (key == 's') {
				double j = 5; int k = 13;for (int i = 1; i < 10; ++++i)
				{
					ofexp = i;//Lambda                               三个可变动参数
					BSigma = j;//the Sigma of B{p,q}   小边缘越清
					RSigma = k;
					computnlink(); computtlink(); int flow = g->maxflow(); printf("Flow = %d\n", flow);
					cout << k;
					for (int i = 0; i < inputImg.rows*inputImg.cols; i++) {
						if (g->what_segment(i)) {

							result.at<uchar>(i) = 255; result.at<uchar>(i) = 255; result.at<uchar>(i) = 255;
						}

					}

					namedWindow(ii[i+k*6]);
					imshow(ii[i+k * 6], result);
				}
			}
		}

		return 0;

}