从 URL 读取图像并将其保存在内存中

Question

我正在使用 Python 和请求库。例如，我只想将图像下载到 numpy 数组中，并且有多个问题可以在其中找到不同的组合（使用 opencv、PIL、requests、urllib...）

它们都不适合我的情况。当我尝试下载图像时，我基本上收到此错误：

cannot identify image file <_io.BytesIO object at 0x7f6a9734da98>

我的代码的一个简单示例可以是：

import requests
from PIL import Image

response = requests.get(url, stream=True)
response.raw.decode_content = True
image = Image.open(response.raw)
image.show()

让我发疯的主要原因是，如果我将图像下载到文件（使用 urllib），整个过程运行没有任何问题！

import urllib
urllib.request.urlretrieve(garment.url, os.path.join(download_folder, garment.get_path()))

我做错了什么？

编辑：

我的错误最终与 URL 形成有关，而不是与请求有关 或 PIL 库。如果 URL 正确，我之前的代码示例应该可以完美运行。

Answer 1

我认为您在将它们保存在 Image 之前以某种方式使用来自 requests.raw 对象的数据，但请求响应原始对象不可搜索，您只能从中读取一次：

>>> response.raw.seekable()
False

第一次打开就可以了：

>>> response.raw.tell()
0
>>> image = Image.open(response.raw)

第二次打开抛出错误（流位置已经在文件末尾）：

>>> response.raw.tell()
695  # this file length https://docs.python.org/3/_static/py.png

>>> image = Image.open(response.raw)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/lib/python3/dist-packages/PIL/Image.py", line 2295, in open
    % (filename if filename else fp))
OSError: cannot identify image file <_io.BytesIO object at 0x7f11850074c0>

如果您想多次使用它们，您应该将请求响应中的数据保存在类似文件的对象（当然也可以是文件）中：

import io
image_data = io.BytesIO(response.raw.read())

现在您可以根据需要读取图像流并多次回退：

>>> image_data.seekable()
True

image = Image.open(image_data)
image1 = Image.open(image_data)