Come posso salvare un'immagine con PIL?

Question

Ho appena elaborato alcune immagini utilizzando la libreria di immagini Python (PIL) usando un post che ho trovato prima per eseguire trasformazioni di Fourier di immagini e non riesco a far funzionare la funzione di salvataggio. L'intero codice funziona bene, ma appena wont salvare 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 l'immagine con Pythons PIL?

Answer 1

L'errore per quanto riguarda l'estensione del file è stata gestita, si sia utilizzare bmp (senza il punto), o passaggio il nome dell'output con l'estensione già. Ora per gestire l'errore è necessario modificare correttamente i dati nel dominio della frequenza per essere salvati come immagine intera, PIL ti dice che non accetta i dati mobili da salvare come bmp.

Ecco un suggerimento (con altre modifiche minori, come l'utilizzo di fftshift e numpy.array invece di numpy.asarray) per fare 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') 

Answer 2

Provare a rimuovere . prima dello .bmp (non corrisponde a BMP come previsto). Come puoi vedere dall'errore, lo save_handler ha la parte superiore del format che hai fornito e quindi cerca una corrispondenza in SAVE. Tuttavia, la chiave corrispondente in quell'oggetto è BMP (anziché .BMP).

Non so molto su PIL, ma da qualche ricerca rapida intorno sembra che sia un problema con lo mode dell'immagine. Cambiare la definizione di j a:

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

sembrava funzionare per me (comunque notare che ho pochissima conoscenza della PIL, quindi vorrei suggerire utilizzando la soluzione @ di MMGP come s/sa chiaramente quello che stanno facendo:)). Per i tipi di mode, ho usato la pagina this - si spera che una delle scelte funzionerà per voi.

Answer 3

Dovreste essere in grado di lasciare semplicemente PIL ottenere il tipo di file da estensione, vale a dire l'uso:

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

Answer 4

So che questo è vecchio, ma ho trovato che aprendo il file usando open(fp, 'w') e quindi il salvataggio del file funzionerà. E.g:

j.save(open(fp, 'w')) 

fp essendo il percorso del file, ovviamente. (Questo è con il cuscino)