pybind11 cv :: MAT来自C＆＃x2B;＆＃x2B;到Python

Question

我想编写一个将图像作为参数并返回另一个图像并使用pybind11将其绑定到python中的函数。

关于如何接收图像作为参数的部分很好地求解，谢谢 to这个问题。

另一方面，返回的图像有点棘手。

在这里，我的代码（我尝试使用cv函数作为示例翻转图像）：

py::array_t<uint8_t> flipcvMat(py::array_t<uint8_t>& img)
{
    auto rows = img.shape(0);
    auto cols = img.shape(1);
    auto channels = img.shape(2);
    std::cout << "rows: " << rows << " cols: " << cols << " channels: " << channels << std::endl;
    auto type = CV_8UC3;

    cv::Mat cvimg2(rows, cols, type, (unsigned char*)img.data());

    cv::imwrite("/source/test.png", cvimg2); // OK

    cv::Mat cvimg3(rows, cols, type);
    cv::flip(cvimg2, cvimg3, 0);

    cv::imwrite("/source/testout.png", cvimg3); // OK

    py::array_t<uint8_t> output(
                                py::buffer_info(
                                cvimg3.data,
                                sizeof(uint8_t), //itemsize
                                py::format_descriptor<uint8_t>::format(),
                                3, // ndim
                                std::vector<size_t> {rows, cols , 3}, // shape
                                std::vector<size_t> {cols * sizeof(uint8_t), sizeof(uint8_t), 3} // strides
    )
    );
    return output;
}

我将其从python称为：

img = cv2.imread('/source/whatever/ubuntu-1.png')
img3= opencvtest.flipcvMat(img)

• 我的输入映像是RGB图像。
• 这两个用cv :: imwrite编写的图像都是正确的（原件与输入相同，第二个是正确翻转的）

问题在python方面，返回的图像似乎是错误的（我可以区分一些形状，但是像素不是正确的位置。

我的猜测是我在创建py :: buffer_info时遇到了问题，但是我不能找到它。

Answer 1

是的，确实是py :: buffer_info的问题，更精确而

​​不是：

{cols * sizeof（uint8_t），sizeof（uint8_t），3}

步伐应为：

{sizeof（uint8_t） * cols * 3，sizeof（uint8_t） * 3，sizeof（uint8_t）}