pybind11 numpy转cv::mat;eigen

最新推荐文章于 2024-08-16 07:40:57 发布

bea_tree

最新推荐文章于 2024-08-16 07:40:57 发布

阅读量2.9k

点赞数

文章标签： pybind11 numpy转mat

本文链接：https://blog.csdn.net/bea_tree/article/details/88676725

版权

python语言灵活易用，但是有时效率较低，c++效率较高，pybind可以较好的结合两者。
在c++代码中引入pybind11相关库，pybind可以利用buffer将python中的一些数据类型转化为c++可用的类型。如下代码：

#include<iostream>
#include<vector>
#include<opencv2/core/core.hpp>
#include<opencv2/imgproc/imgproc.hpp>
#include<opencv2/highgui/highgui.hpp>
#include<pybind11/pybind11.h>
#include<pybind11/numpy.h>
#include<pybind11/stl.h>
#include"mat_warper.h"
#include<pybind11/eigen.h>
#include<eigen3/Eigen/Dense>

namespace py = pybind11;
using namespace std;
using namespace cv;

Mat numpy_uint8_1c_to_cv_mat(py::array_t<unsigned char>& input) {
    if (input.ndim() != 2)
        throw std::runtime_error("1-channel image must be 2 dims ");
    py::buffer_info buf = input.request();
    cv::Mat mat(buf.shape[0], buf.shape[1], CV_8UC1, (unsigned char*)buf.ptr);
    retur