通过套接字在python中读取opencv图像

时间:2017-12-22 05:40:00

标签: python c++ sockets opencv

我正在尝试在从c ++发送的python套接字中读取opencv图像。

我能够将图像读入另一个c ++程序或VB程序并构建图像但是使用python我不了解发生了什么。

我发送mat.data的发送代码:

char *start_s = "<S><size>43434234<cols>64<rows>64<SE>";//plus I send the image size, cols, rows, which varies, not like the static char string shown
char *end_e = "<E>";
cv::Mat image_send = some_mat;
iResult = send( ConnectSocket, start_s, (int)strlen(start_s), 0 );
iResult = send( ConnectSocket, (const char *) image_send.data, i_buffer_size, 0 );
iResult = send( ConnectSocket, end_e, (int)strlen(end_e), 0 );

这是我用python试过的,但还没有取得任何成功。 image_cols和Image_rows是从套接字过滤的,这里没有显示,只有来自c ++ mat的image_mat.data位于我试图放入图像的套接字中:

data = conn.recv(4757560)
        if(i_Read_Image == 2) & (image_cols != 0) & (image_rows != 0):
            print ("Entering")
            #print(data)
            data2 = np.fromstring(data, dtype='uint8')
            img_np = cv2.imdecode(data2,cv2.IMREAD_COLOR )
            cv2.imshow('image',img_np)
            cv2.waitKey(0)

            #Also tried this
            #img = Image.new('RGB', (image_cols, image_rows))
            #img.putdata(data)

        #img5 = np.reshape(data2,(image_rows,image_cols))
        i_Read_Image = 0

1 个答案:

答案 0 :(得分:0)

在评论的帮助下,我得到了一个有效的答案。原始图像采用单个阵列RGB,这需要重新整形并放入“RGB&#39;图片,它可以在一行中完成:

img = Image.fromarray(data2.reshape(image_rows,image_cols,3), 'RGB')

当从套接字读取opencv数据数组时:这有效:

data = conn.recv(567667)
if(i_Read_Image == 2) & (image_cols != 0) & (image_rows != 0):
    data2 = np.fromstring(data, dtype='uint8')
    img = Image.fromarray(data2.reshape(image_rows,image_cols,3), 'RGB')
    img.show()