pybind11 C++读取图像

pybind11 C++读取图像

#include 
#include
#include 
#include
using namespace std;
//using namespace cv;
namespace py = pybind11;

double getimg(py::array_t input) {
	// read input arrays buffer_info
	py::buffer_info buf = input.request();
	// allocate the output buffer
	py::array_t result = py::array_t(buf.size);
	double* ptr = (double*)buf.ptr;
	double high = buf.shape[0];
	double width = buf.shape[1];
	cout << "high" << high << endl;
	cout << "width" << width << endl;
	cout << "ptr" << ptr << endl;
	//Mat mat(high, width, CV_8UC3, ptr);
	// Add both arrays

	return 1;
}

PYBIND11_MODULE(meiyan, m) {
	m.doc() = "Add two vectors using pybind11"; // optional module docstring
	m.def("getimg", &getimg, "Add two NumPy arrays");
}

import numpy as np
import cv2
import meiyan
img = cv2.imread('1.jpg')
print(np.shape(img))
print(type(img))
c = meiyan.getimg(img)
print(c)