C++ 调用 Python 脚本,并把 cv::Mat 类型传参到 Python 端

本文介绍了如何在C++中调用Python脚本并将cv::Mat类型的数据传递到Python端。首先确保C++与Python交互正常,以及各自独立处理opencv图像无误。然后参考发送Mat的方法,在C++中利用numpy头文件实现数据转换并传递。
摘要由CSDN通过智能技术生成

C++ 调用 Python 脚本,并把 cv::Mat 类型传参到 Python 端

前言

查看了很多参考,最后找到一个完整的示例并且跑通,在开始这个任务之前,首先应该确保你的环境是没有问题的,比如:

  • C++ 和 Python 之间可以传递简单类型的参数
  • C++ 端独立通过 opencv 加载图像是没有问题的
  • Python 端 import cv2 独立加载图像是没有问题的

具备上面这些条件后,可以参考下面的代码,将 cv::Mat 类型的参数传递到 Python 端。

代码

这部分主要参考 sending Mat

C++ 端需要借助 numpy,所以也需要配置好 numpy 的头文件位置。

C++ code

#include "Python.h"
#include "numpy/arrayobject.h"
#include <opencv2/opencv.hpp>

#include <iostream>

using namespace std;
// for the references to all the functions
PyObject *m_PyDict, *m_PyFooBar;

// for the reference to the Pyhton module
PyObject* m_PyModule;

int main() {

Python调用C++函数并传递cv::Mat数据,可以使用OpenCV库将cv::Mat数据转换C++中的Mat类型。下面是一个示例代码,展示了如何在Python调用一个接受cv::Mat数据的C++函数: 假设我们有一个C++函数,它接受一个cv::Mat类型的图像数据: ```c++ #include <iostream> #include <opencv2/core.hpp> #include <opencv2/imgcodecs.hpp> void my_function(cv::Mat& image) { cv::imwrite("output.jpg", image); } ``` 现在,我们可以在Python中使用OpenCV库来读取一张图像,并将其转换cv::Mat类型的数据,然后将其传递给这个C++函数。下面是示例代码: ```python import cv2 import numpy as np import ctypes # 加载C++编译后的动态链接库 lib = ctypes.cdll.LoadLibrary('./libexample.so') # 定义函数参数类型 lib.my_function.argtypes = [ctypes.c_void_p] # 读取一张图像,并将其转换cv::Mat类型的数据 image = cv2.imread('input.jpg') image_data = np.ascontiguousarray(image) image_data_ptr = image_data.ctypes.data_as(ctypes.c_void_p) # 将cv::Mat类型的数据传递C++函数 lib.my_function(image_data_ptr) ``` 在上面的示例代码中,我们首先使用`cdll.LoadLibrary()`函数加载C++编译后的动态链接库。接着,我们使用`argtypes`属性定义了C++函数的参数类型,其中`ctypes.c_void_p`表示无类型指针。然后,我们使用OpenCV库读取一张图像,并使用`np.ascontiguousarray()`函数将图像数据转换为连续的内存块。最后,我们使用`ctypes.c_void_p`类型将图像数据的内存地址传递C++函数`my_function()`。 需要注意的是,在传递cv::Mat数据时,需要确保图像数据是连续的。如果图像数据不是连续的,需要使用`np.ascontiguousarray()`函数将其转换为连续的内存块。
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值