将图像从python传递到C ++并返回

时间:2019-07-12 09:22:37

标签: python c++ opencv pipe

我需要在Python中读取图像(使用OpenCV),将其通过管道传输到C ++程序,然后再通过管道传输回Python。 到目前为止,这是我的代码:

C ++

#include <iostream>
#include <fstream>
#include <string>
#include <vector>
#include <cv.h>
#include <highgui.h>
#include <cstdio>

#include <sys/stat.h>

using namespace std;
using namespace cv;

int main(int argc, char *argv[]) {
    const char *fifo_name = "fifo";
    mknod(fifo_name, S_IFIFO | 0666, 0);
    ifstream f(fifo_name);
    string line;
    getline(f, line);
    auto data_size = stoi(line);

    char *buf = new char[data_size];
    f.read(buf, data_size);

    Mat matimg;
    matimg = imdecode(Mat(1, data_size, CV_8UC1, buf), CV_LOAD_IMAGE_UNCHANGED);

    imshow("display", matimg);
    waitKey(0);

    return 0;
}

Python

import os
import cv2

fifo_name = 'fifo'

def main():
    data = cv2.imread('testimage.jpg').tobytes()
    try:
        os.mkfifo(fifo_name)
    except FileExistsError:
        pass
    with open(fifo_name, 'wb') as f:
        f.write('{}\n'.format(len(data)).encode())
        f.write(data)

if __name__ == '__main__':
    main()

当C ++尝试打印到图像时,引发异常。我已经调试了代码,并且buf已填充,但是matimg为空。

1 个答案:

答案 0 :(得分:0)

在代码中,C ++读取器调用mknod,而它应该只打开由Python写入器创建的现有命名管道。

如果读者尝试打开该管道时该管道不存在,则它可能会失败,也可能会继续尝试以超时超时打开命名管道。例如:

const char *fifo_name = "fifo";
std::ifstream f;
for(;;) { // Wait till the named pipe is available.
    f.open(fifo_name, std::ios_base::in);
    if(f.is_open())
        break;
    std::this_thread::sleep_for(std::chrono::seconds(3));
}