如何以最佳方式将PIL图像保存为JSON?

5

我正在尝试发送包含Pillow图像的json字典作为其中一个字段,为此必须将图像转换为字符串。 我尝试使用Pillow函数: image.toString() 但仍然得到了字节,所以我尝试对其进行编码:

buff = BytesIO()
image.save(buff, format="JPEG")
img_str = base64.b64encode(buff.getvalue())

但仍然将其作为字节获取。如何将Pillow图像转换为可保存在JSON文件中的格式?


2
我不确定我理解这个问题。您现在已经将数据作为字符串,想要将数据保存在JSON文件中,而JSON文件可以保存字符串。那么问题是什么?将您的字符串保存到文件中即可。 - Kevin
也许这个 jstr = json.dumps({"image": base64.b64encode(buff.getvalue()).decode('ascii')}) - Mark Setchell
2个回答

7
我使用以下方法通过json交换Pillow图像。
import json
from PIL import Image
import numpy as np

filename = "filename.jpeg"
image = Image.open(filename)
json_data = json.dumps(np.array(image).tolist())
new_image = Image.fromarray(np.array(json.loads(json_data), dtype='uint8'))

你好!你能解释一下为什么这是一个好的回答吗? - Cullub

6
在评论中,Mark Setchell建议在b64encode调用的结果上调用.decode('ascii')。我同意这会起作用,但我认为一开始进行base64编码引入了不必要的额外步骤,使您的代码更加复杂。
相反,我建议直接解码image.tostring返回的字节。唯一的复杂之处在于字节对象可能包含大于128的值,因此您无法使用ascii对其进行解码。尝试使用可以处理高达256值的编码,如latin1
from PIL import Image
import json

#create sample file. You don't have to do this in your real code.
img = Image.new("RGB", (10,10), "red")

#decode.
s = img.tobytes().decode("latin1")

#serialize.
with open("outputfile.json", "w") as file:
    json.dump(s, file)

(* 但是令我惊讶的是,使用UTF-8编码生成的JSON文件仍然比使用Latin1编码生成的文件要小,至少对于我的示例文件来说。根据自己的判断确定文件大小或程序清晰度哪个更重要。)


我并不想与你争论或不同意你的看法,因为我来这里也是要像其他人一样学习。我认为JPEG编码可能比任何关于base64效率的问题更重要,以减小文件大小。一个3000x4000像素的彩色图像有36MB,但可能只有300kB经过JPEG编码,而额外30%的base64效率使其达到了400kB,这相对于省略JPEG步骤来说微不足道。正如我所说,我来这里是为了学习,可能我是错的 - 我对Python和JSON都很新,并愿意看到解决方案的另一面并进行学习。 - Mark Setchell
@MarkSetchell 观察得很好。我同意将图像保存为JPEG格式可能会显着改善内存。我的示例代码使用 tobytes() 而不是 image.save(buff, format="JPEG"),因为它稍微更加简洁。但这对于我的答案并非必要;无论他们选择使用 b64-和-ascii 编码还是 latin1 编码,OP都应该随意继续使用自己的 BytesIO 策略。 - Kevin
很酷 - 谢谢你抽出时间澄清。你已经得到了我的投票。 - Mark Setchell

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接