So konvertieren Sie ein NumPy-Array mithilfe der Matplotlib-Farbkarte in ein PIL-Bild

132

Ich habe ein einfaches Problem, aber ich kann keine gute Lösung dafür finden.

Ich möchte ein NumPy 2D-Array, das ein Graustufenbild darstellt, in ein RGB-PIL-Bild konvertieren, während ich einige der Matplotlib-Farbkarten anwende.

Ich kann eine vernünftige PNG-Ausgabe erhalten, indem ich den folgenden pyplot.figure.figimageBefehl verwende:

dpi = 100.0
w, h = myarray.shape[1]/dpi, myarray.shape[0]/dpi
fig = plt.figure(figsize=(w,h), dpi=dpi)
fig.figimage(sub, cmap=cm.gist_earth)
plt.savefig('out.png')

Obwohl ich dies anpassen könnte, um das zu bekommen, was ich will (wahrscheinlich bekomme ich mit StringIO das PIL-Bild), frage ich mich, ob es keinen einfacheren Weg gibt, dies zu tun, da es ein sehr natürliches Problem der Bildvisualisierung zu sein scheint. Sagen wir so etwas:

colored_PIL_image = magic_function(array, cmap)

— Heltonbiker
quelle

Für voll funktionsfähigen Code können Sie Folgendes beachten: Gibt es eine gute Farbkarte, um Graustufenbilder mithilfe von Pythons PIL in bunte Bilder umzuwandeln?

— Adam

Antworten:

220

Ein ziemlich beschäftigter Einzeiler, aber hier ist es:

Stellen Sie zunächst sicher, dass Ihr NumPy-Array myarraymit dem Maximalwert bei normalisiert ist 1.0.
Wenden Sie die Farbkarte direkt auf an myarray.
In den 0-255Bereich skalieren .
Konvertieren Sie mit in ganze Zahlen np.uint8().
Verwenden Sie Image.fromarray().

Und du bist fertig:

from PIL import Image
from matplotlib import cm
im = Image.fromarray(np.uint8(cm.gist_earth(myarray)*255))

mit plt.savefig():

Geben Sie hier die Bildbeschreibung ein

mit im.save():

Geben Sie hier die Bildbeschreibung ein

— fraxel
quelle

Der Teil "Farbkarte direkt anwenden auf myarray" schneidet direkt ins Herz! Ich wusste nicht, dass es möglich ist, danke!

— Heltonbiker

Beim Studium der Dokumente zu LinearSegmentedColormap (von denen cm.gist_earth eine Instanz ist) stellte ich fest, dass es möglich ist, es mit einem "Bytes" -Argument aufzurufen, das es bereits in uint8 konvertiert. Dann wird der im = Image.fromarray(cm.gist_earth(myarray, bytes=True))

— Einzeiler

@CiprianTomoiaga, die Form des Arrays sollte den gewünschten Bildabmessungen entsprechen. Beispielsweise würde ein VGA-Bild aus einem Array mit Form (1024.768) erzeugt. Sie sollten beachten, dass dies für monochrome Bilder gilt. Dies ist wichtig, da normalerweise beim Konvertieren eines RGB-Bilds in ein Array die Form beispielsweise (1024.768.3) beträgt, da es drei Kanäle hat.

— Heltonbiker

Ich NameError: name 'cm' is not defined

— erhalte

@msofrom matplotlib import cm

— Quantum7

input = numpy_image
np.unit8 -> konvertiert in ganze Zahlen
konvertieren ('RGB') -> konvertiert in RGB

Image.fromarray -> gibt ein Bildobjekt zurück

from PIL import Image
import numpy as np

PIL_image = Image.fromarray(np.uint8(numpy_image)).convert('RGB')

PIL_image = Image.fromarray(numpy_image.astype('uint8'), 'RGB')

— Aravinda_gn
quelle

Hoffe, es wird das Problem lösen, aber bitte fügen Sie eine Erklärung Ihres Codes hinzu, damit der Benutzer ein perfektes Verständnis dafür erhält, was er wirklich will.

— Jaimil Patel

Gute, aktualisierte Antwort. Die vorherigen stammen von vor einigen Jahren.

— Catalina Chircu

Die in der akzeptierten Antwort beschriebene Methode hat bei mir auch nach Anwendung der in den Kommentaren genannten Änderungen nicht funktioniert. Aber der folgende einfache Code hat funktioniert:

import matplotlib.pyplot as plt
plt.imsave(filename, np_array, cmap='Greys')

np_array kann entweder ein 2D-Array mit Werten von 0..1 Floats o2 0..255 uint8 sein, und in diesem Fall benötigt es cmap. Bei 3D-Arrays wird cmap ignoriert.

— Shital Shah
quelle