Як зберегти зображення за допомогою 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я використовував цю сторінку - сподіваюся, один із варіантів, який ви там буде, допоможе вам.

Я знаю, що це старе, але я виявив, що (під час використання Подушки) відкриття файлу за допомогою open(fp, 'w')та збереження файлу спрацює. Наприклад:

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

fp будучи шлях файлу, звичайно.