python - “cv2.imdecode(numpyArray, cv2.CV_LOAD_IMAGE_COLOR)”返回无

我正在尝试将图像转换为Opencv(转换为numpy数组)，并使用该数组通过ROS节点发布消息。我尝试通过以下代码进行相同的操作

    fig.canvas.draw()
    nparr = np.fromstring ( fig.canvas.tostring_argb(), np.uint8 )
    print nparr
    img_np = cv2.imdecode(nparr, cv2.CV_LOAD_IMAGE_COLOR)
    print img_np
    image_message = bridge.cv2_to_imgmsg(img_np, encoding="passthrough")
    pub.publish(image_message)

但是，当我尝试这样做时，我收到一条错误消息

AttributeError: 'NoneType' object has no attribute 'shape'

因此，我尝试打印两个值为[255 191 191 ..., 191 191 191]的numpy数组的值。我不明白的是img_np的值是None。我不知道我哪里出了错。任何帮助表示赞赏。

最佳答案

我最近也遇到过类似的问题。

不论原始资源如何，np.fromstring()方法都会从参数字符串返回 1-D np.array。要将np.array用作OpenCV中的图像数组，您可能需要根据图像的宽度和高度对其进行重塑。

试试这个:

img_str = np.fromstring ( fig.canvas.tostring_argb(), np.uint8 )
ncols, nrows = fig.canvas.get_width_height()
nparr = np.fromstring(img_str, dtype=np.uint8).reshape(nrows, ncols, 3)
img_np = cv2.imdecode(nparr, cv2.CV_LOAD_IMAGE_COLOR)