Python PIL: как записать изображение PNG в строку

Question

Я сгенерировал изображение, используя PIL. Как я могу сохранить его в строку в памяти? Image.save() метод требует файл.

Я хотел бы иметь несколько таких изображений в словаре.

Answer 1

Вы можете использовать BytesIO класс, чтобы получить обертку вокруг строк, которая ведет себя как файл. BytesIO Объект предоставляет тот же интерфейс, что и файл, но сохраняет содержимое только в памяти:

import io

with io.BytesIO() as output:
    image.save(output, format="GIF")
    contents = output.getvalue()

Вы должны явно указать выходной формат с format параметр, в противном случае PIL выдаст ошибку при попытке его автоматического обнаружения.

Если вы загрузили изображение из файла, оно имеет format параметр, который содержит исходный формат файла, поэтому в этом случае вы можете использовать format=image.format,

В старых версиях Python 2 до введения io модуль, который вы бы использовали StringIO модуль вместо.

Answer 2

Для Python3 требуется использовать BytesIO:

from io import BytesIO
from PIL import Image, ImageDraw

image = Image.new("RGB", (300, 50))
draw = ImageDraw.Draw(image)
draw.text((0, 0), "This text is drawn on image")

byte_io = BytesIO()

image.save(byte_io, 'PNG')

Читать подробнее: http://fadeit.dk/blog/post/python3-flask-pil-in-memory-image

Answer 3

Решение STH не работает для меня
потому что в...

Imaging / PIL / Image.pyc строка 1423 -> поднять KeyError(ext) # неизвестное расширение

Он пытался определить формат по расширению в имени файла, которого нет в случае StringIO

Вы можете обойти обнаружение формата, установив формат самостоятельно в параметре

import StringIO
output = StringIO.StringIO()
format = 'PNG' # or 'JPEG' or whatever you want
image.save(output, format)
contents = output.getvalue()
output.close()

Answer 4

save() может принимать как файловый объект, так и путь, поэтому вы можете использовать буфер в памяти, такой как StringIO:

buf= StringIO.StringIO()
im.save(buf, format= 'JPEG')
jpeg= buf.getvalue()

Answer 5

С современными (по состоянию на середину 2017 года Python 3.5 и Pillow 4.0):

StringIO больше не работает, как раньше. Класс BytesIO - правильный способ справиться с этим. Функция сохранения подушек ожидает строку в качестве первого аргумента и неожиданно не видит StringIO как таковой. Следующее похоже на старые решения StringIO, но на его месте стоит BytesIO.

from io import BytesIO
from PIL import Image

image = Image.open("a_file.png")
faux_file = BytesIO()
image_data = faux_file.getvalue()
image.save(faux_file, 'png')

Answer 6

Когда вы говорите: "Я бы хотел, чтобы количество таких изображений сохранялось в словаре", неясно, является ли это структурой в памяти или нет.

Вам не нужно делать что-либо из этого, чтобы смирить изображение в памяти. Просто держи image Объект в вашем словаре.

Если вы собираетесь записать свой словарь в файл, вы можете посмотреть на im.tostring() метод и Image.fromstring() функция

http://effbot.org/imagingbook/image.htm

im.tostring () => строка

Возвращает строку, содержащую данные пикселей, используя стандартный "сырой" кодировщик.

Image.fromstring (режим, размер, данные) => изображение

Создает память изображений из пиксельных данных в строке, используя стандартный "сырой" декодер.

"Формат" (.jpeg, .png и т. Д.) Имеет значение только на диске, когда вы обмениваетесь файлами. Если вы не обмениваетесь файлами, формат не имеет значения.