如何在python中从numpy数组创建png映像文件-对象而不保存到磁盘（用于http请求）

Question

1. 生成图像
2. 用scipy.misc.toImage将图像保存到磁盘
3. 从磁盘打开保存的图像文件
4. 使用请求模块发送包含image opened image file对象的http请求

这非常好，但是我想摆脱步骤2和3，所以我不需要先将对象保存到磁盘，然后再加载它。相反，我希望在一个与http服务器兼容的文件对象中转换我的numpy数组，并直接发送它。（就像从open（）中得到的一样）

例如，我知道用PIL将numpy数组转换为PNG-image很容易，但我只找到了如何在一个函数中结合保存到磁盘来实现这一点。

import numpy as np
import requests
from scipy.misc import toimage

arr = generate64x64x3ImageWithNumpy()
toimage(arr, cmin=0.0, cmax=255.0).save('tmp.png')
d = {'key':API_KEY}
f= {'image': open('tmp.png', 'rb')}
result = requests.post(SERVER_URL, files=f, data=d)

我想要这个：

arr = generate64x64x3ImageWithNumpy()

not_on_disk = numpyArrayToPNGImageWithoutSavingOnDisk(arr)

d = {'key':API_KEY}
f = {'image': not_on_disk}
result = requests.post(SERVER_URL, files=f, data=d)

Answer 1

您可以在内存中使用带有savefig的iostream(https://docs.python.org/3/library/io.html#io.bytesio)

import io
tmpFile = io.BytesIO()
savefig(tmpFile, format='png')

为了验证是否可以将这个工作的tmpfile与保存到磁盘的实际文件进行比较。

# Get contents of tmpFile
tmpFile.seek(0)
not_on_disk = tmpFile.read(-1)

# Save to and load from disk
fname = 'tmp.png'
savefig(fname)
on_disk = open(fname, 'rb').read(-1)

>>>not_on_disk == on_disk
True

您正在使用scipy和pil而不是matplotlib进行编辑，但答案应该是相同的，包括format关键字用于保存。