Как выполнить сжатие JPEG в Python без записи/чтения

Я хотел бы работать непосредственно со сжатыми изображениями JPEG. Я знаю, что с PIL/Pillow я могу сжать изображение при его сохранении, а затем прочитать сжатое изображение - например,

from PIL import Image
im1 = Image.open(IMAGE_FILE)
IMAGE_10 = os.path.join('./images/dog10.jpeg')
im1.save(IMAGE_10,"JPEG", quality=10)
im10 = Image.open(IMAGE_10)

но, я бы хотел сделать это без посторонней записи и чтения. Есть ли какой-нибудь пакет Python с функцией, которая будет принимать изображение и качество в качестве входных данных и вернуть jpeg-версию этого изображения с заданным качеством?

Ответ 1

Для файловых файлов в памяти вы можете использовать StringIO. Посмотрите:

from io import StringIO # "import StringIO" directly in python2
from PIL import Image
im1 = Image.open(IMAGE_FILE)

# here, we create an empty string buffer    
buffer = StringIO.StringIO()
im1.save(buffer, "JPEG", quality=10)

# ... do something else ...

# write the buffer to a file to make sure it worked
with open("./photo-quality10.jpg", "w") as handle:
    handle.write(buffer.contents())

Если вы проверите файл photo-quality10.jpg, это должно быть то же изображение, но с 10% качеством, как при настройке сжатия JPEG.

Ответ 2

Использование BytesIO

try:
    from cStringIO import StringIO as BytesIO
except ImportError:
    from io import BytesIO

def generate(self, image, format='jpeg'):
    im = self.generate_image(image)
    out = BytesIO()
    im.save(out, format=format,quality=75)
    out.seek(0)
    return out

В Python3.0 отсутствует StringIO, ссылка на StringIO в python3