Come posso salvare un'immagine con PIL?

Ho appena eseguito l'elaborazione delle immagini utilizzando la libreria di immagini Python (PIL) utilizzando un post che ho trovato in precedenza per eseguire trasformazioni di Fourier delle immagini e non riesco a far funzionare la funzione di salvataggio. L'intero codice funziona bene ma non salverà l'immagine risultante:

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

L'errore che ottengo è il seguente:

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

Come posso salvare un'immagine con Pythons PIL?

L'errore relativo all'estensione del file è stato gestito, si utilizza BMP(senza il punto) o si passa il nome dell'output già con l'estensione. Ora per gestire l'errore è necessario modificare correttamente i dati nel dominio della frequenza per essere salvati come immagine intera, ti PILsta dicendo che non accetta dati float da salvare come BMP.

Ecco un suggerimento (con altre modifiche minori, come usare fftshifte numpy.arrayinvece di numpy.asarray) per eseguire la conversione per una corretta visualizzazione:

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

Dovresti essere in grado di consentire semplicemente a PIL di ottenere il tipo di file dall'estensione, ovvero utilizzare:

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

Prova a rimuovere .prima di .bmp(non corrisponde BMPcome previsto). Come puoi vedere dall'errore, save_handlerè in maiuscolo quello che formathai fornito e quindi cerca una corrispondenza in SAVE. Tuttavia la chiave corrispondente in quell'oggetto è BMP(invece di .BMP).

Non ne so molto PIL, ma da una rapida ricerca in giro sembra che sia un problema con model'immagine. Modifica della definizione di jin:

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

Sembrava funzionare per me (tuttavia nota che ne ho pochissima conoscenza PIL, quindi suggerirei di usare la soluzione di @ mmgp poiché sa chiaramente cosa stanno facendo :)). Per i tipi di mode, ho usato questa pagina - spero che una delle scelte funzionerà per te.

So che questo è vecchio, ma ho scoperto che (durante l'utilizzo di Pillow) l'apertura del file utilizzando open(fp, 'w')e quindi il salvataggio del file funzionerà. Per esempio:

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

fp essendo il percorso del file, ovviamente.