Python调用C++代码（含OpenCV cv::Mat转换传输数据）

最新推荐文章于 2024-01-18 10:06:29 发布

YixuanWWW

最新推荐文章于 2024-01-18 10:06:29 发布

阅读量4.5k

点赞数 5

分类专栏：环境配置

本文链接：https://blog.csdn.net/weixin_35754503/article/details/102938254

版权

本文介绍了如何在Python项目中调用C++代码，特别是涉及OpenCV图像处理的场景。利用pyboostcvconverter库，可以将Python的cv::Mat对象转换为C++类型，反之亦然。文章详细讲解了如何在C++中添加自定义函数，以及如何处理转换和返回的数据。在编译过程中，确保正确链接OpenCV和其他依赖库以避免运行时错误。

摘要由CSDN通过智能技术生成

文章目录

简介

Python因为编写简单、易用而被广泛使用，但是其效率比较低下。尤其是在计算机视觉、图像处理领域，C++调用GPU，相对于Python可能是10倍以上的提速。一些开源的算法通常是使用C++来实现。经常会有需求需要快速用Python实现一个项目，然后其中要调用一段C++代码。

本方法使用了一个开源库，pyboostcvconverter，这个C++库被编译后，会生成.so，在.so路径下，就可以用Python通过import pbcvt来调用其中编写的转换方法。

因项目需求，在Python中会读取到图像，图像是numpy.ndarray类型的，而C++的函数的参数是用cv::Mat类型的，怎么进行这两者的转换也是关键。这个库将Python的numpy.ndarray转成C++的cv::Mat，C++函数返回的cv::Mat又会被转成Python的numpy.ndarray

库结构 + 添加你的函数

在pyboostcvconverter/src目录下有4个文件：

pyboost_cv2_converter.cpp
pyboost_cv3_converter.cpp
pyboost_cv4_converter.cpp
python_module.cpp （关键代码，需要添加代码的地方）

在python_module.cpp中添加相应内容，便可以添加Python可以调用的C++函数。

#define PY_ARRAY_UNIQUE_SYMBOL pbcvt_ARRAY_API
 
#include <boost/python.hpp>
#include <pyboostcvconverter/pyboostcvconverter.hpp>
#include <opencv2/opencv.hpp>
//添加相应用到的头文件

namespace pbcvt {
   
 
    using namespace boost::python;
 
/**
 * @brief Example function. Basic inner matrix product using explicit matrix conversion.
 * @param left left-hand matrix operand (NdArray required)
 * @param right right-hand matrix operand (NdArray required)
 * @return an NdArray representing the dot-product of the left and right operands
 */
    PyObject *dot(PyObject *left, PyObject *right) {
   
 
        cv::Mat leftMat, rightMat;
        leftMat = pbcvt::fromNDArrayToMat(left);
        rightMat = pbcvt::fromNDArrayToMat(right);
        auto c1 = leftMat.cols, r2 = rightMat.rows;
        // Check that the 2-D matrices can be legally multiplied.
        if (c1 != r2)