Como posso salvar uma imagem com PIL?

Acabei de fazer algum processamento de imagem usando a biblioteca de imagens Python (PIL) usando um post que encontrei anteriormente para realizar transformações de Fourier de imagens e não consigo fazer a função salvar funcionar. Todo o código funciona bem, mas não vai salvar a imagem 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")

O erro que recebo é o seguinte:

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

Como posso salvar uma imagem com Pythons PIL?

O erro referente à extensão do arquivo foi resolvido, ou você usa BMP(sem o ponto) ou passa o nome de saída já com a extensão. Agora, para lidar com o erro, você precisa modificar adequadamente seus dados no domínio da frequência para serem salvos como uma imagem inteira,PIL está lhe dizendo que não aceita dados flutuantes para salvar como BMP.

Aqui está uma sugestão (com outras pequenas modificações, como usar fftshifte em numpy.arrayvez de numpy.asarray) para fazer a conversão para visualização adequada:

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

Você deve ser capaz de simplesmente deixar o PIL obter o tipo de arquivo da extensão, ou seja, usar:

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

Tente remover o .antes de .bmp(não corresponde BMPao esperado). Como você pode ver pelo erro, save_handlerestá colocando em maiúsculas o que formatvocê forneceu e procurando uma correspondência SAVE. No entanto, a chave correspondente nesse objeto é BMP(em vez de .BMP).

Não sei muito sobre PIL , mas de uma busca rápida ao redor parece que é um problema com o modeda imagem. Alterando a definição de jpara:

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

Pareceu funcionar para mim (no entanto, note que tenho muito pouco conhecimento sobre PILisso, então sugiro usar a solução de @mmgp, pois ele / ela sabe claramente o que está fazendo :)). Para os tipos de mode, usei esta página - espero que uma das opções funcione para você.

Eu sei que isso é antigo, mas descobri que (ao usar o Pillow) abrir o arquivo usando open(fp, 'w')e salvá-lo funcionará. Por exemplo:

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

fp sendo o caminho do arquivo, é claro.