generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

grabCut fist impl and test

Browse Source
This commit is contained in:
James Bowman 2010-07-22 23:39:42 +00:00
parent f572047496
commit 389c002eea
2 changed files with 35 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
modules/python/api
View File

@ -1644,6 +1644,15 @@ HOGDetectMultiScale CvSeq*
int nbins 9 int nbins 9
int gammaCorrection 1 int gammaCorrection 1
grabCut
CvArr image
CvArr mask
CvRect rect
CvArr bgdModel
CvArr fgdModel
int iterCount
int mode
# These functions are handwritten in cv.cpp; they appear here as 'doconly' declarations # These functions are handwritten in cv.cpp; they appear here as 'doconly' declarations
# so that their documentation can be auto-generated # so that their documentation can be auto-generated
ApproxPoly /doconly ApproxPoly /doconly

26
modules/python/cv.cpp
View File

@ -3763,6 +3763,21 @@ static CvSeq* cvHOGDetectMultiScale( const CvArr* image, CvMemStorage* storage,
return seq; return seq;
} }
static void cvgrabCut(CvArr *image,
CvArr *mask,
CvRect rect,
CvArr *bgdModel,
CvArr *fgdModel,
int iterCount,
int mode)
{
cv::Mat _image = cv::cvarrToMat(image);
cv::Mat _mask = cv::cvarrToMat(mask);
cv::Mat _bgdModel = cv::cvarrToMat(bgdModel);
cv::Mat _fgdModel = cv::cvarrToMat(fgdModel);
grabCut(_image, _mask, rect, _bgdModel, _fgdModel, iterCount, mode);
}
static int zero = 0; static int zero = 0;
/************************************************************************/ /************************************************************************/
@ -3826,10 +3841,13 @@ static int to_ok(PyTypeObject *to)
#define MKTYPE(NAME) do { NAME##_specials(); if (!to_ok(&NAME##_Type)) return; } while (0) #define MKTYPE(NAME) do { NAME##_specials(); if (!to_ok(&NAME##_Type)) return; } while (0)
using namespace cv;
extern "C" extern "C"
#if defined WIN32 || defined _WIN32 #if defined WIN32 || defined _WIN32
__declspec(dllexport) __declspec(dllexport)
#endif #endif
void initcv() void initcv()
{ {
PyObject *m, *d; PyObject *m, *d;
@ -3939,6 +3957,14 @@ void initcv()
PUBLISH(CV_PTLOC_VERTEX); PUBLISH(CV_PTLOC_VERTEX);
PUBLISH(CV_PTLOC_OUTSIDE_RECT); PUBLISH(CV_PTLOC_OUTSIDE_RECT);
PUBLISH(GC_BGD);
PUBLISH(GC_FGD);
PUBLISH(GC_PR_BGD);
PUBLISH(GC_PR_FGD);
PUBLISH(GC_INIT_WITH_RECT);
PUBLISH(GC_INIT_WITH_MASK);
PUBLISH(GC_EVAL);
#include "generated2.i" #include "generated2.i"
#if 0 #if 0