Python 成像 : load jpeg from memory

Question

问题是从内存中加载 jpeg 编码的图像。

我从套接字收到一个字符串:

jpgdata = self.rfile.read(sz)

我知道这是 jpeg 编码的图像。

我需要解码它。最愚蠢的解决方案是:

o = open("Output/1.jpg","wb")
o.write(jpgdata)
o.close()
dt = Image.open("Output/1.jpg")

问题是如何在内存中做同样的事情？

Answer 1

PIL 的 Image.open 对象接受任何类似文件的对象。这意味着您可以将 Image 数据包装在 StringIO 对象上，并将其传递给 Image.Open

from io import BytesIO
file_jpgdata = BytesIO(jpgdata)
dt = Image.open(file_jpgdata)

或者，尝试将 self.rfile 作为参数传递给 Image.open - 它也可以正常工作。 (这适用于 Python 3 - 对于 Python 2 使用 from cStringIO import StringIO as BytesIO)