如何将 Python numpy 数组转换为 base64 输出

Question

我基本上需要做this，但是用 Python 而不是 Javascript。我从 socketio 连接接收到一个 base64 编码的字符串，将其转换为 uint8 并处理它，然后需要将其转换为 base64 字符串，以便我可以将其发回。

所以，到目前为止，我已经得到了这个（我正在从 socketio 服务器获取data 字典）：

import pickle
import base64
from io import BytesIO
from PIL import Image

base64_image_string = data["image"]
image = Image.open(BytesIO(base64.b64decode(base64_image_string)))
img = np.array(image)

如何反转这个过程以从img 回到base64_image_string？

更新：
我已经通过以下方式解决了这个问题（继续上面的代码 sn-p）：

pil_img = Image.fromarray(img)
buff = BytesIO()
pil_img.save(buff, format="JPEG")
new_image_string = base64.b64encode(buff.getvalue()).decode("utf-8")

有点令人困惑，new_image_string 与 base64_image_string 不完全相同，但从 new_image_string 渲染的图像看起来相同，所以我很满意！

Answer 1

我相信既然numpy.arrays 支持缓冲协议，你只需要以下内容：

processed_string = base64.b64encode(img)

所以，例如：

>>> encoded = b"aGVsbG8sIHdvcmxk"
>>> img = np.frombuffer(base64.b64decode(encoded), np.uint8)
>>> img
array([104, 101, 108, 108, 111,  44,  32, 119, 111, 114, 108, 100], dtype=uint8)
>>> img.tobytes()
b'hello, world'
>>> base64.b64encode(img)
b'aGVsbG8sIHdvcmxk'
>>>

Answer 2

我也有同样的问题。经过一番搜索和尝试，我的最终解决方案与您的几乎相同。

唯一的区别是base64编码的字符串是png格式的数据，所以我需要把它从RGBA改成RGBchannels才转换成np.array：

image = image.convert ("RGB")
img = np.array(image)

在逆过程中，你将数据视为JPEG格式，这可能是new_image_string与base64_image_string不完全相同的原因？

Answer 3

来自http://www.programcreek.com/2013/09/convert-image-to-string-in-python/：

import base64

with open("t.png", "rb") as imageFile:
    str = base64.b64encode(imageFile.read())
    print str

是二进制读取

https://docs.python.org/2/library/base64.html