Function dense_crf_wrapper

densecrf_python/densecrf3d.cpp:13–158

static PyObject *
dense_crf_wrapper(PyObject *self, PyObject *args)
{
    PyObject *I=NULL, *P=NULL, *param=NULL, *BilateralModsStds=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_UINT8, NPY_IN_ARRAY);
    if (arr_I == NULL) return NULL;
    
    arr_P = (PyArrayObject*)PyArray_FROM_OTF(P, NPY_FLOAT32, NPY_IN_ARRAY);
    if (arr_P == NULL) return NULL;

    
    /*vv* code that makes use of arguments *vv*/
    
    int nd_I = PyArray_NDIM(arr_I);   //number of dimensions
    int nd_P = PyArray_NDIM(arr_P);
    if(nd_I != nd_P)
    {
        cerr<<"data dimension "<<nd_I<<" probability dimension "<<nd_P<<" do not match"<<endl;
        return NULL;

    }
    npy_intp * shape_I = PyArray_DIMS(arr_I);  // npy_intp array of length nd showing length in each dim.
    npy_intp * shape_P = PyArray_DIMS(arr_P);
#ifdef DEBUG
    cout<<"input data shape: ";
    for(int i=0; i<nd_I; i++)
    {
        cout<<shape_I[i]<<" ";
    }
    cout<<std::endl;
    cout<<"input probability shape: ";
    for(int i=0; i<nd_P; i++)
    {
        cout<<shape_P[i]<<" ";
    }
    cout<<std::endl;
#endif
    for(int i=0; i<nd_I-1; i++)
    {
        if(shape_I[i] !=shape_P[i] || shape_I[i]!=shape_P[i])
        {
            cerr<<"input shape does not match"<<endl;
            return NULL;
        }
    }

    double MaxIterations = PyFloat_AsDouble(PyDict_GetItemString(param, "MaxIterations"));
    double PosRStd  = PyFloat_AsDouble(PyDict_GetItemString(param, "PosRStd"));
    double PosCStd  = PyFloat_AsDouble(PyDict_GetItemString(param, "PosCStd"));
    double PosZStd  = PyFloat_AsDouble(PyDict_GetItemString(param, "PosZStd"));
    double PosW     = PyFloat_AsDouble(PyDict_GetItemString(param, "PosW"));
    double BilateralRStd = PyFloat_AsDouble(PyDict_GetItemString(param, "BilateralRStd"));
    double BilateralCStd = PyFloat_AsDouble(PyDict_GetItemString(param, "BilateralCStd"));
    double BilateralZStd = PyFloat_AsDouble(PyDict_GetItemString(param, "BilateralZStd"));