Comment puis-je enregistrer une image avec PIL?

Je viens de faire un traitement d'image en utilisant la bibliothèque d'images Python (PIL) en utilisant un article que j'ai trouvé plus tôt pour effectuer des transformations de Fourier des images et je ne parviens pas à faire fonctionner la fonction de sauvegarde. L'ensemble du code fonctionne bien mais il ne sauvera pas l'image résultante:

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'erreur que j'obtiens est la suivante:

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

Comment puis-je enregistrer une image avec Pythons PIL?

L'erreur concernant l'extension de fichier a été gérée, soit vous utilisez BMP(sans le point), soit vous passez le nom de sortie avec l'extension déjà. Maintenant, pour gérer l'erreur, vous devez modifier correctement vos données dans le domaine de fréquence à enregistrer en tant qu'image entière, PILcela vous indique qu'il n'accepte pas les données flottantes à enregistrer en BMP.

Voici une suggestion (avec d'autres modifications mineures, comme utiliser fftshiftet numpy.arrayau lieu de numpy.asarray) pour effectuer la conversion pour une visualisation correcte:

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

Vous devriez pouvoir simplement laisser PIL obtenir le type de fichier de l'extension, c'est-à-dire utiliser:

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

Essayez de supprimer le .avant le .bmp(il ne correspond pas BMPcomme prévu). Comme vous pouvez le voir à partir de l'erreur, la save_handlercasse supérieure est celle que formatvous avez fournie, puis recherche une correspondance dans SAVE. Cependant, la clé correspondante dans cet objet est BMP(au lieu de .BMP).

Je ne sais pas grand-chose PIL, mais après quelques recherches rapides, il semble que ce soit un problème avec model'image. Modification de la définition de jen:

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

Cela a semblé fonctionner pour moi (notez cependant que je n'ai que très peu de connaissances PIL, donc je suggérerais d'utiliser la solution de @ mmgp car il sait clairement ce qu'ils font :)). Pour les types de mode, j'ai utilisé cette page - j'espère que l'un des choix fonctionnera pour vous.

Je sais que c'est vieux, mais j'ai trouvé que (en utilisant Pillow) ouvrir le fichier en utilisant open(fp, 'w')puis en enregistrant le fichier fonctionnera. Par exemple:

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

fp étant le chemin du fichier, bien sûr.