使用BOOST.python从C++将结构返回给Python

Question

我编写了一个C++方法,我需要将结构返回给Python.我已经能够mat使用BOOST按照此链接中描述的方法将OpenCV 从Python 发送到C++ .

现在我需要走另一条路; 从C++返回到Python,然后在Python中访问该结构.可以吗？

任何样本或参考链接都是好的.我在发布这个问题之前尝试了谷歌搜索,我无法获得任何样本或解释链接.

Answer 1

你可以使用modules/python/src2/cv2.cpp中的另一个函数:

PyObject* pyopencv_from(const cv::Mat& m)
{
  if( !m.data )
      Py_RETURN_NONE;
  cv::Mat temp, *p = (cv::Mat*)&m;
  if(!p->refcount || p->allocator != &g_numpyAllocator)
  {
      temp.allocator = &g_numpyAllocator;
      m.copyTo(temp);
      p = &temp;
  }
  p->addref();
  return pyObjectFromRefcount(p->refcount);
}

然后Boost Python包装器将如下所示:

boost::python::object toPython( const cv::Mat &frame )
{
    PyObject* pyObjFrame = pyopencv_from( frame );
    boost::python::object boostPyObjFrame(boost::python::handle<>((PyObject*)pyObjFrame));

    return boostPyObjFrame;
}

请查看以下链接:https://wiki.python.org/moin/boost.python/handle,以确保为您的案例使用适当的boost :: python :: handle <>函数.

如果你需要不需要返回一个cv :: Mat但是你可以考虑使用不同的数据来使用boost :: python :: list或boost :: python :: dict.例如,如果要将2D点的向量返回到Python,可以执行以下操作:

boost::python::dict toPython( std::vector<cv::Point> newPoints, std::vector<cv::Point> oldPoints )
{
    boost::python::dict pointsDict;
    boost::python::list oldPointsList;
    boost::python::list newPointsList;

    for( size_t ii = 0; ii < oldPoints.size( ); ++ii )
    {
        oldPointsList.append( boost::python::make_tuple( oldPoints[ii].x, oldPoints[ii].y ) );
    }

    for( size_t ii = 0; ii < newPoints.size( ); ++ii )
    {
        newPointsList.append( boost::python::make_tuple( newPoints[ii].x, newPoints[ii].y ) );
    }

    pointsDict["oldPoints"] = oldPointsList;
    pointsDict["newPoints"] = newPointsList;
    return pointsDict
}

最后是Boost Python包装器:

BOOST_PYTHON_MODULE( myWrapper )
{
    // necessary only if array (cv::Mat) is returned
    import_array();
    boost::python::converter::registry::insert( &extract_pyarray, type_id<PyArrayObject>());

    def toPython("toPython", &toPython);
}

我没有测试过这个特定的解决方案,但原则上应该可行.