Как я могу сохранить изображение с помощью PIL?

Я только что выполнил некоторую обработку изображений с помощью библиотеки изображений Python (PIL), используя сообщение, которое я нашел ранее для выполнения преобразования Фурье изображений, и я не могу заставить работать функцию сохранения. Весь код работает нормально, но не сохраняет получившееся изображение:

from PIL import Image
import numpy as np

i = Image.open("C:/Users/User/Desktop/mesh.bmp")
i = i.convert("L")
a = np.asarray(i)
b = np.abs(np.fft.rfft2(a))
j = Image.fromarray(b)
j.save("C:/Users/User/Desktop/mesh_trans",".bmp")

Я получаю следующую ошибку:

save_handler = SAVE[string.upper(format)] # unknown format
    KeyError: '.BMP'

Как я могу сохранить изображение с помощью Pythons PIL?

Ошибка, касающаяся расширения файла, была обработана, вы либо используете BMP(без точки), либо уже передаете выходное имя с расширением. Теперь, чтобы обработать ошибку, вам необходимо правильно изменить данные в частотной области для сохранения в виде целочисленного изображения, PILкоторое сообщает вам, что не принимает данные с плавающей запятой для сохранения в формате BMP.

Вот предложение (с другими небольшими изменениями, такими как использование fftshiftи numpy.arrayвместо numpy.asarray) для преобразования для правильной визуализации:

import sys
import numpy
from PIL import Image

img = Image.open(sys.argv[1]).convert('L')

im = numpy.array(img)
fft_mag = numpy.abs(numpy.fft.fftshift(numpy.fft.fft2(im)))

visual = numpy.log(fft_mag)
visual = (visual - visual.min()) / (visual.max() - visual.min())

result = Image.fromarray((visual * 255).astype(numpy.uint8))
result.save('out.bmp')

Вы должны просто позволить PIL получить тип файла из расширения, то есть использовать:

j.save("C:/Users/User/Desktop/mesh_trans.bmp")

Попробуйте удалить .перед .bmp(он не соответствует BMPожидаемому). Как вы можете видеть из ошибки, save_handlerэто верхний регистр, который formatвы указали, а затем ищет совпадение SAVE. Однако соответствующий ключ в этом объекте BMP(вместо .BMP).

Я не очень разбираюсь в этом PIL, но при быстром поиске кажется, что это проблема с modeизображением. Изменение определения jна:

j = Image.fromarray(b, mode='RGB')

Казалось, у меня работает (однако обратите внимание, что я очень мало знаю PIL, поэтому я бы предложил использовать решение @ mmgp, поскольку он / она четко знает, что они делают :)). Для типов modeя использовал эту страницу - надеюсь, один из вариантов вам подойдет.

Я знаю, что это старый, но я обнаружил, что (при использовании Pillow) открытие файла с помощью, open(fp, 'w')а затем сохранение файла будет работать. Например:

with open(fp, 'w') as f:
    result.save(f)

fp конечно, путь к файлу.