Jak mogę zapisać obraz za pomocą PIL?

Właśnie wykonałem trochę przetwarzania obrazu przy użyciu biblioteki obrazów Python (PIL), korzystając z postu, który znalazłem wcześniej, aby wykonać transformacje Fouriera obrazów i nie mogę uruchomić funkcji zapisywania. Cały kod działa dobrze, ale po prostu nie zapisze wynikowego obrazu:

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")

Otrzymuję następujący błąd:

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

Jak mogę zapisać obraz za pomocą Pythons PIL?

Błąd dotyczący rozszerzenia pliku został rozwiązany, albo użyj BMP(bez kropki), albo już przekazujesz nazwę wyjściową z rozszerzeniem. Teraz, aby poradzić sobie z błędem, musisz odpowiednio zmodyfikować dane w dziedzinie częstotliwości, aby zostały zapisane jako obraz całkowity, PILinformuje, że nie akceptuje danych zmiennoprzecinkowych do zapisania jako BMP.

Oto sugestia (z innymi drobnymi modyfikacjami, takimi jak użycie fftshifti numpy.arrayzamiast numpy.asarray), aby wykonać konwersję w celu właściwej wizualizacji:

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')

Powinieneś móc po prostu pozwolić PILowi pobrać typ pliku z rozszerzenia, tj. Użyj:

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

Spróbuj usunąć .przed .bmp(nie pasuje BMPzgodnie z oczekiwaniami). Jak widać z błędu, save_handlerjest to duże litery formatpodane przez Ciebie, a następnie szuka dopasowania w SAVE. Jednak odpowiadający mu klucz w tym obiekcie to BMP(zamiast .BMP).

Nie wiem zbyt wiele PIL, ale po szybkich poszukiwaniach wydaje się, że jest to problem z modeobrazem. Zmiana definicji jna:

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

Wydawało mi się, że działa dla mnie (jednak zauważ, że mam bardzo małą wiedzę PIL, więc sugerowałbym użycie rozwiązania @ mmgp, ponieważ on / on wyraźnie wie, co robią :)). Jeśli chodzi o typy mode, użyłem tej strony - mam nadzieję, że jeden z dostępnych wyborów będzie dla Ciebie odpowiedni.

Wiem, że to jest stare, ale odkryłem, że (podczas korzystania z Pillow) otwarcie pliku za pomocą, open(fp, 'w')a następnie zapisanie pliku będzie działać. Na przykład:

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

fp jest to oczywiście ścieżka do pliku.