Commit 389c002e authored by James Bowman's avatar James Bowman

grabCut fist impl and test

parent f5720474
...@@ -1644,6 +1644,15 @@ HOGDetectMultiScale CvSeq* ...@@ -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
......
...@@ -3763,6 +3763,21 @@ static CvSeq* cvHOGDetectMultiScale( const CvArr* image, CvMemStorage* storage, ...@@ -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) ...@@ -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() ...@@ -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
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment