Function maxflow2d_wrapper

maxflow_python/maxflow.cpp:10–59

example to use numpy object: http://blog.debao.me/2013/04/my-first-c-extension-to-numpy/ write a c extension ot Numpy: http://folk.uio.no/hpl/scripting/doc/python/NumPy/Numeric/numpy-13.html

// example to use numpy object: http://blog.debao.me/2013/04/my-first-c-extension-to-numpy/
// write a c extension ot Numpy: http://folk.uio.no/hpl/scripting/doc/python/NumPy/Numeric/numpy-13.html
static PyObject *
maxflow2d_wrapper(PyObject *self, PyObject *args)
{
    PyObject *I=NULL, *P=NULL, *param=NULL;
    PyArrayObject *arr_I=NULL, *arr_P=NULL;
    if (!PyArg_ParseTuple(args, "OOO", &I, &P, &param)) return NULL;
    
    arr_I = (PyArrayObject*)PyArray_FROM_OTF(I, NPY_FLOAT32, NPY_IN_ARRAY);
    arr_P = (PyArrayObject*)PyArray_FROM_OTF(P, NPY_FLOAT32, NPY_IN_ARRAY);
    if (arr_I == NULL || arr_P == NULL) return NULL;
    float lambda = PyFloat_AsDouble(PyTuple_GET_ITEM(param, 0));
    float sigma  = PyFloat_AsDouble(PyTuple_GET_ITEM(param, 1));

    /*vv* code that makes use of arguments *vv*/
    int dimI = PyArray_NDIM(arr_I);             // number of dimensions
    int dimP = PyArray_NDIM(arr_P); 
    npy_intp * shapeI = PyArray_DIMS(arr_I);    // npy_intp array of length nd showing length in each dim
    npy_intp * shapeP = PyArray_DIMS(arr_P);

    if(dimI > 3){
        cout << "the input dimension can only be 2 or 3"<<endl;
        return NULL;
    }
    if(dimP != 3){
        cout << "dimension of probabilily map should be 3"<<endl;
        return NULL;
    }
    if(shapeI[0] != shapeP[0] || shapeI[1] != shapeP[1]){
        cout << "image and probability map have different sizes"<<endl;
        return NULL;
    }
    if(shapeP[2] != 2){
        cout << "probabilily map should have two channels"<<endl;
        return NULL;
    }

    int chns = 1;
    if(dimI == 3) chns = shapeI[2];
    npy_intp outshape[2];
    outshape[0]=shapeI[0];
    outshape[1]=shapeI[1];
    PyArrayObject * arr_L = (PyArrayObject*)  PyArray_SimpleNew(2, outshape, NPY_INT8);
    maxflow_inference((unsigned char *) arr_L->data, (const float *) arr_I->data, (const float *) arr_P->data, NULL,
        shapeI[0], shapeI[1], chns, 2, lambda, sigma);
  
    Py_DECREF(arr_I);
    Py_DECREF(arr_P);
    Py_INCREF(arr_L);
    return PyArray_Return(arr_L);
}

static PyObject *
interactive_maxflow2d_wrapper(PyObject *self, PyObject *args)