【问题标题】:Python C API PyObject_Repr Segmentation fault (core dumped)Python C API PyObject_Repr 分段错误(核心转储)
【发布时间】:2016-07-21 15:40:15
【问题描述】:

我正在尝试在我的 C++ 项目中使用 scikit learn。这是我正在使用的代码:

    #include <Python.h>
    PyObject* loadModule(char* name)//
    {
        PyObject* pName = PyString_FromString(name);
        PyObject* pModule = PyImport_Import(pName);
        Py_DECREF(pName);
        return pModule;
    }
    void displayPyObject(PyObject* object)
    {

    PyObject* objectsRepresentation = PyObject_Repr(object);
    PyErr_Print();
    const char* s = PyString_AsString(objectsRepresentation);
    PyErr_Print();
    std::cout << "[ PYOBJECT ]" << s << std::endl;

    }
    //load functions/ attributes from module
    PyObject* loadComponentFromModule(char* module, char* component) 
    {
       PyObject* pModule = loadModule(module);
       PyObject* pyComponent = PyObject_GetAttrString(pModule, component);
       Py_DECREF(pModule);
       return pyComponent;
    }


   //WRAPPER FOR KMEANS CLUSTERING FROM SCIKIT-LEARN
class KMeans
{
public:
    KMeans(int nClusters)
    {

        PyObject* KmeansClass = loadComponentFromModule("sklearn.cluster", "KMeans");
        PyObject* pName2 = PyInt_FromLong((long) nClusters);
        PyObject* pArgs = PyTuple_New(1);
        PyTuple_SetItem(pArgs, 0, pName2);
        _Kcluster = PyObject_CallObject(KmeansClass, pArgs);
        _closestor = loadComponentFromModule("sklearn.metrics","pairwise_distances_argmin_min");
        Py_DECREF(KmeansClass);
        Py_DECREF(pName2);
        Py_DECREF(pArgs);
    }
    ~KMeans()
    {
        Py_DECREF(_Kcluster);
       Py_DECREF(_closestor);
    }

    void setNumClusters(int nClusters)
    {
        std::cout << "change to number cluster: " << nClusters << "\n";
        PyObject* nCluster = PyInt_FromLong((long) nClusters);
        int code = PyObject_SetAttrString(_Kcluster,"n_clusters", nCluster);
        PyErr_Print();
        if (code == -1)
        {
            std::cout << "[Error] KMeans.setNumClusters() Failed!! - Number of clusters didn't change!!\n";
        }
        Py_DECREF(nCluster);
    }




    void info()
    {
        displayPyObject(_Kcluster);
    }

private:
    PyObject* _Kcluster;
    //PyObject* _result;
    PyObject* _closestor;
};

PyObject* loadClassifier()
{
    PyObject* loader = loadComponentFromModule("sklearn.externals.joblib", "load");
    PyObject* pName2 = PyString_FromString("lda.pkl");
    PyObject* pArgs = PyTuple_New(1);
    PyTuple_SetItem(pArgs, 0, pName2);
    PyObject* clf = PyObject_CallObject(loader, pArgs);
    Py_DECREF(loader);
    Py_DECREF(pName2);
    Py_DECREF(pArgs);
//    displayPyObject(clf);
    return clf;
}

void produce_error()
{
    std::cout << "============================= LINE 0 =========================================\n";
    PyObject* clf = loadClassifier();//"sklearn.externals.joblib", "load");
    std::cout << "============================= LINE 1 =========================================\n";
    KMeans cluster(8);
    std::cout << "============================= LINE 2 =========================================\n";
    cluster.setNumClusters(5);
    std::cout << "============================= LINE 3 =========================================\n";
    cluster.info();
    std::cout << "============================= LINE 4 =========================================\n";

}

int main(int argc, char *argv[])
{
    Py_Initialize();
    produce_error();
    Py_Finalize();
    return 0;
}

每次我运行这个程序,我都会得到错误:

============================= LINE 0 =========================================
============================= LINE 1 =========================================
Segmentation fault (core dumped)

据我了解,在不调用分类器的情况下,我没有收到任何错误,但是每当我调用它 (loadClassifier()) 并创建 KMeans 实例时,都会显示错误。有时错误在我加载模块的 KMeans 的构造函数内部,在另一个函数中使用相同的代码(不在此处),错误在 PyObject_Repr() 内部(在 displayPyObject() 内部)。

以前有人遇到过同样的问题吗?你知道怎么解决吗?提前致谢。

【问题讨论】:

    标签: c++ segmentation-fault python-c-api


    【解决方案1】:

    问题解决了。我想到了。因此,如果有人需要,请在此处回答。使用 Py_DECREF() 时要小心。问题出在 KMeans 的构造函数中的这一行:

    Py_DECREF(pName2);

    因为当你运行 Py_DECREF(pArgs);它会尝试释放已经被释放的 pName2。它将产生不可预测的行为。评论该行后,一切运行正常。

    【讨论】:

      猜你喜欢
      • 2015-02-27
      • 1970-01-01
      • 2022-01-14
      • 2017-02-25
      • 2016-07-12
      • 2018-03-16
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多