grabcut_3d.cpp File Reference

#include "precomp.hpp"
#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/imgproc/imgproc_c.h"
#include "opencv2/core/internal.hpp"
#include <math.h>
#include <assert.h>
#include <string.h>
#include <stdlib.h>
#include <stdio.h>
#include <limits.h>
#include <float.h>
#include "grabcut_3d/grabcut_3d.h"
#include <limits>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/core/core.hpp>

Include dependency graph for grabcut_3d.cpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	GMM
Defines
#define	ALPHA   1.0
#define	BETA   1.0
Functions
void	assignGMMsComponents (const Mat &img, const Mat &depth_img, const Mat &mask, const GMM &bgdGMM, const GMM &fgdGMM, Mat &compIdxs)
float	calcBeta (const Mat &img, const Mat &depth_img)
void	calcNWeights (const Mat &img, const Mat &depth_img, Mat &leftW, Mat &upleftW, Mat &upW, Mat &uprightW, float beta, float gamma)
void	checkMask (const Mat &img, const Mat &mask)
void	constructGCGraph (const Mat &img, const Mat &depth_img, const Mat &mask, const GMM &bgdGMM, const GMM &fgdGMM, float lambda, const Mat &leftW, const Mat &upleftW, const Mat &upW, const Mat &uprightW, GCGraph< float > &graph)
void	estimateSegmentation (GCGraph< float > &graph, Mat &mask)
Vec3f	getDataTerm (const Mat &img, const Mat &depth_img, Point &p)
Vec3f	getDataTerm (const Mat &img, const Mat &depth_img, int row, int col)
void	initGMMs (const Mat &img, const Mat &depth_img, const Mat &mask, GMM &bgdGMM, GMM &fgdGMM)
void	initMaskWithRect (Mat &mask, Size imgSize, Rect rect)
void	learnGMMs (const Mat &img, const Mat &depth_img, const Mat &mask, const Mat &compIdxs, GMM &bgdGMM, GMM &fgdGMM)

---

Define Documentation

#define ALPHA   1.0

Definition at line 225 of file grabcut_3d.cpp.

#define BETA   1.0

Definition at line 226 of file grabcut_3d.cpp.

---

Function Documentation

void assignGMMsComponents	(	const Mat &	img,
		const Mat &	depth_img,
		const Mat &	mask,
		const GMM &	bgdGMM,
		const GMM &	fgdGMM,
		Mat &	compIdxs
	)

Definition at line 425 of file grabcut_3d.cpp.

float calcBeta	(	const Mat &	img,
		const Mat &	depth_img
	)

Definition at line 248 of file grabcut_3d.cpp.

void calcNWeights	(	const Mat &	img,
		const Mat &	depth_img,
		Mat &	leftW,
		Mat &	upleftW,
		Mat &	upW,
		Mat &	uprightW,
		float	beta,
		float	gamma
	)

Definition at line 291 of file grabcut_3d.cpp.

void checkMask	(	const Mat &	img,
		const Mat &	mask
	)

Definition at line 343 of file grabcut_3d.cpp.

void constructGCGraph	(	const Mat &	img,
		const Mat &	depth_img,
		const Mat &	mask,
		const GMM &	bgdGMM,
		const GMM &	fgdGMM,
		float	lambda,
		const Mat &	leftW,
		const Mat &	upleftW,
		const Mat &	upW,
		const Mat &	uprightW,
		GCGraph< float > &	graph
	)

Definition at line 476 of file grabcut_3d.cpp.

void estimateSegmentation	(	GCGraph< float > &	graph,
		Mat &	mask
	)

Definition at line 539 of file grabcut_3d.cpp.

Vec3f getDataTerm	(	const Mat &	img,
		const Mat &	depth_img,
		Point &	p
	)

Definition at line 236 of file grabcut_3d.cpp.

Vec3f getDataTerm	(	const Mat &	img,
		const Mat &	depth_img,
		int	row,
		int	col
	)

Definition at line 228 of file grabcut_3d.cpp.

void initGMMs	(	const Mat &	img,
		const Mat &	depth_img,
		const Mat &	mask,
		GMM &	bgdGMM,
		GMM &	fgdGMM
	)

Definition at line 382 of file grabcut_3d.cpp.

void initMaskWithRect	(	Mat &	mask,
		Size	imgSize,
		Rect	rect
	)

Definition at line 366 of file grabcut_3d.cpp.

void learnGMMs	(	const Mat &	img,
		const Mat &	depth_img,
		const Mat &	mask,
		const Mat &	compIdxs,
		GMM &	bgdGMM,
		GMM &	fgdGMM
	)

Definition at line 443 of file grabcut_3d.cpp.