¿Cómo puedo guardar una imagen con PIL?

Acabo de hacer un procesamiento de imágenes usando la biblioteca de imágenes de Python (PIL) usando una publicación que encontré anteriormente para realizar transformaciones de Fourier de imágenes y no puedo hacer que la función de guardar funcione. Todo el código funciona bien, pero no guarda la imagen resultante:

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

El error que obtengo es el siguiente:

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

¿Cómo puedo guardar una imagen con Pythons PIL?

Se ha manejado el error con respecto a la extensión del archivo, puede usar BMP(sin el punto) o pasar el nombre de salida con la extensión ya. Ahora, para manejar el error, necesita modificar correctamente sus datos en el dominio de frecuencia para guardarlos como una imagen entera, PILle dice que no acepta datos flotantes para guardarlos como BMP.

Aquí hay una sugerencia (con otras modificaciones menores, como usar fftshifty en numpy.arraylugar de numpy.asarray) para realizar la conversión para una visualización adecuada:

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

Debería poder simplemente dejar que PIL obtenga el tipo de archivo de la extensión, es decir, use:

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

Intente eliminar el .antes del .bmp(no coincide BMPcomo se esperaba). Como puede ver en el error, save_handlerestá en mayúsculas lo formatque proporcionó y luego está buscando una coincidencia en SAVE. Sin embargo, la clave correspondiente en ese objeto es BMP(en lugar de .BMP).

No sé mucho sobre PIL, pero por una búsqueda rápida, parece que es un problema con modela imagen. Cambiar la definición de ja:

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

Pareció funcionar para mí (sin embargo, tenga en cuenta que tengo muy poco conocimiento PIL, por lo que sugeriría usar la solución de @ mmgp ya que él / ella sabe claramente lo que están haciendo :)). Para los tipos de mode, utilicé esta página; con suerte, una de las opciones allí funcionará para usted.

Sé que esto es antiguo, pero descubrí que (mientras usaba Pillow) abrir el archivo usando open(fp, 'w')y luego guardar el archivo funcionará. P.ej:

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

fp siendo la ruta del archivo, por supuesto.