python - PIL open() 方法不适用于 BytesIO

Question

出于某种原因，当我尝试从 BytesIO 流制作图像时，它无法识别图像。这是我的代码:

from PIL import Image, ImageGrab
from io import BytesIO

i = ImageGrab.grab()
i.resize((1280, 720))
output = BytesIO()
i.save(output, format = "JPEG")
output.flush()
print(isinstance(Image.open(output), Image.Image))

以及它抛出的错误的堆栈跟踪:

Traceback (most recent call last):
  File "C:/Users/Natecat/PycharmProjects/Python/test.py", line 9, in <module>
    print(isinstance(Image.open(output), Image.Image))
  File "C:\Python27\lib\site-packages\PIL\Image.py", line 2126, in open
    % (filename if filename else fp))
IOError: cannot identify image file <_io.BytesIO object at 0x02394DB0>

我正在使用 PIL 的 Pillow 实现。

Answer 1

把 BytesIO 想象成一个文件对象，当你写完图像后，文件的光标在文件的末尾，所以当 Image.open() 尝试调用 output .read()，它立即得到一个 EOF。

在将 output 传递给 Image.open() 之前，您需要添加一个 output.seek(0)。