全栈工程师开发手册 (作者:栾鹏)
架构系列文章
我是在ubuntu系统进行实验的,所以和window可能会有区别。
python调用C/C++有不少的方法,如boost.python, swig, ctypes, pybind11等,这些方法有繁有简,而pybind11的优点是对C++ 11支持很好,API比较简单,现在我们就简单记下Pybind11的入门操作。
cmake教程可以参考:https://blog.csdn.net/luanpeng825485697/article/details/81202136
进入pybind11的目录,
cd tests
cmake ..
cmake --build . --config Release --target check
如果所有测试用例都通过了,说明安装成功了。
#include
namespace py = pybind11;
int add(int i, int j)
{
return i + j;
}
PYBIND11_MODULE(example, m)
{
// optional module docstring
m.doc() = "pybind11 example plugin";
// expose add function, and add keyword arguments and default arguments
m.def("add", &add, "A function which adds two numbers", py::arg("i")=1, py::arg("j")=2);
// exporting variables
m.attr("the_answer") = 42;
py::object world = py::cast("World");
m.attr("what") = world;
}
2.1 使用window编译
我没有实验,所以可以参考别的教程
2.2 CMake的编译方法
当然,我们也可以使用CMake进行编译。首先写一个CMakeLists.txt
cmake_minimum_required(VERSION 2.8.12)
project(example)
add_subdirectory(pybind11)
pybind11_add_module(example example.cpp)
这里要求example.cpp放在和pybind11同一级的目录下,因为我们在CMakeLists.txt中调用了同目录pybind11和同目录的example.cpp文件。在当前目录下执行
cmake .
make
会生成example.cpython-36m-x86_64-linux-gnu.so文件。这个文件就是python可以调用的文件。还是在相同目录下运行python,进入python命令行
import example
example.add(3, 4)
[out]: 7
上面是一个简单的例子,有时我们需要的功能可能很复杂。
生成模型设计库调用问题。
比如你的cpp文件中引用了其他的第三方库,这个时候我们生成的so文件,可能是需要依赖第三方库的。
本地myopencv.cpp文件
#include
#include
#include
#include
#include
namespace py = pybind11;
void read_img(std::string img_path)
{
cv::Mat image = cv::imread(img_path, CV_LOAD_IMAGE_COLOR);
}
PYBIND11_MODULE(myopencv, m)
{
m.def("read_img", &read_img, "get image size");
}
CMakeLists.txt可以写成下面这样
cmake_minimum_required(VERSION 2.8.12)
project(myopencv)
add_subdirectory(pybind11)
pybind11_add_module(myopencv myopencv.cpp)
通过cmkae编译通过
cmake .
make
产生myopencv.cpython-36m-x86_64-linux-gnu.so文件
在python中调用
import myopencv
目前这里发现有问题,还没有调通。。。。。
数据类型不匹配的问题
比如opencv在python中是numpy的类型,但是在c++中cv.Mat的类型
对于opencv的类型不匹配问题,github上已经给出了解决方法,
cpp和h文件下载地址 https://github.com/edmBernard/pybind11_opencv_numpy
如何你需要什么可以去github上搜索,或者自己去实现。