使用OpenCv python库从内存中读取基本64位编码的图像

Question

我正在开发一个可通过网络摄像头流进行面部识别的应用程序。我得到画布的base64编码的数据uri，并希望使用它来执行以下操作：

cv2.imshow('image',img)

数据URI看起来像这样：

 data:image/gif;base64,R0lGODlhEAAQAMQAAORHHOVSKudfOulrSOp3WOyDZu6QdvCchPGolfO0o/XBs/fNwfjZ0frl3/zy7////wAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAACH5BAkAABAALAAAAAAQABAAAAVVICSOZGlCQAosJ6mu7fiyZeKqNKToQGDsM8hBADgUXoGAiqhSvp5QAnQKGIgUhwFUYLCVDFCrKUE1lBavAViFIDlTImbKC5Gm2hB0SlBCBMQiB0UjIQA7

因此，为清楚起见，我已经显示了图像的外观，因此base64字符串没有损坏。

<img src="data:image/gif;base64,R0lGODlhEAAQAMQAAORHHOVSKudfOulrSOp3WOyDZu6QdvCchPGolfO0o/XBs/fNwfjZ0frl3/zy7////wAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAACH5BAkAABAALAAAAAAQABAAAAVVICSOZGlCQAosJ6mu7fiyZeKqNKToQGDsM8hBADgUXoGAiqhSvp5QAnQKGIgUhwFUYLCVDFCrKUE1lBavAViFIDlTImbKC5Gm2hB0SlBCBMQiB0UjIQA7">

在官方文档说，是imread接受作为参数的文件路径。从这样的答案，如果我做类似的事情：

 import base64
 imgdata = base64.b64decode(imgstring) #I use imgdata as this variable itself in references below
 filename = 'some_image.jpg'
 with open(filename, 'wb') as f:
    f.write(imgdata)

上面的代码段有效，并且图像文件已正确生成。但是，考虑到我将对流的每个帧执行此操作，因此我认为没有太多文件IO操作是可行的。我希望能够将图像读取到直接创建img对象的内存中。

我尝试了两种似乎对某些人有用的解决方案。

1. 使用PIL参考：

   pilImage = Image.open(StringIO(imgdata))
   

   npImage = np.array(pilImage)
   matImage = cv.fromarray(npImage)
   

我cv没有定义，因为我安装了openCV3，可以作为cv2模块使用。我试过了img = cv2.imdecode(npImage,0)，这什么也没返回。

1. 从解码后的字符串中获取字节并将其转换为numpy数组

   file_bytes = numpy.asarray(bytearray(imgdata), dtype=numpy.uint8)
   

   img = cv2.imdecode(file_bytes, 0) #Here as well I get returned nothing

该文档没有真正提及imdecode函数返回的内容。但是，从我遇到的错误中，我猜想它是以anumpy array或ascalar作为第一个参数。我如何在内存中获取该图像的句柄，以便以后可以执行cv2.imshow('image',img)各种出色的操作。

我希望我能说清楚。

Answer 1

您可以像这样同时使用cv2和枕头：

import base64
from PIL import Image
import cv2
from StringIO import StringIO
import numpy as np

def readb64(base64_string):
    sbuf = StringIO()
    sbuf.write(base64.b64decode(base64_string))
    pimg = Image.open(sbuf)
    return cv2.cvtColor(np.array(pimg), cv2.COLOR_RGB2BGR)

cvimg = readb64('R0lGODlhEAAQAMQAAORHHOVSKudfOulrSOp3WOyDZu6QdvCchPGolfO0o/XBs/fNwfjZ0frl3/zy7////wAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAACH5BAkAABAALAAAAAAQABAAAAVVICSOZGlCQAosJ6mu7fiyZeKqNKToQGDsM8hBADgUXoGAiqhSvp5QAnQKGIgUhwFUYLCVDFCrKUE1lBavAViFIDlTImbKC5Gm2hB0SlBCBMQiB0UjIQA7')
cv2.imshow(cvimg)