web-dev-qa-db-fra.com

Conversion d'un tableau NumPy en une image PIL

Je veux créer une image PIL à partir d'un tableau NumPy. Voici ma tentative:

# Create a NumPy array, which has four elements. The top-left should be pure red, the top-right should be pure blue, the bottom-left should be pure green, and the bottom-right should be yellow
pixels = np.array([[[255, 0, 0], [0, 255, 0]], [[0, 0, 255], [255, 255, 0]]])

# Create a PIL image from the NumPy array
image = Image.fromarray(pixels, 'RGB')

# Print out the pixel values
print image.getpixel((0, 0))
print image.getpixel((0, 1))
print image.getpixel((1, 0))
print image.getpixel((1, 1))

# Save the image
image.save('image.png')

Cependant, l’impression donne:

(255, 0, 0)
(0, 0, 0)
(0, 0, 0)
(0, 0, 0)

Et l'image enregistrée a un rouge pur en haut à gauche, mais tous les autres pixels sont en noir. Pourquoi ces autres pixels ne conservent-ils pas la couleur que je leur ai assignée dans le tableau NumPy?

Merci!

pythonimagenumpypython-imaging-library
22
Karnivaurus

Le mode RGB attend des valeurs sur 8 bits, il suffit donc de transtyper votre tableau pour résoudre le problème:

In [25]: image = Image.fromarray(pixels.astype('uint8'), 'RGB')
    ...:
    ...: # Print out the pixel values
    ...: print image.getpixel((0, 0))
    ...: print image.getpixel((0, 1))
    ...: print image.getpixel((1, 0))
    ...: print image.getpixel((1, 1))
    ...:
(255, 0, 0)
(0, 0, 255)
(0, 255, 0)
(255, 255, 0)
39
Randy