Changer la couleur de tous les pixels avec une autre couleur

Je voudrais changer une couleur unique avec Python.

Si une solution rapide avec PIL existe, je préfère cette solution.

Pour le moment, j'utilise 

convert -background black -opaque '#939393' MyImage.png MyImage.png

OriginalL'auteur Martin Thoma | 2011-06-26

  1. 18

    Si numpy est disponible sur votre ordinateur, essayez de faire quelque chose comme:

    import numpy as np
from PIL import Image

im = Image.open('fig1.png')
data = np.array(im)

r1, g1, b1 = 0, 0, 0 # Original value
r2, g2, b2 = 255, 255, 255 # Value that we want to replace it with

red, green, blue = data[:,:,0], data[:,:,1], data[:,:,2]
mask = (red == r1) & (green == g1) & (blue == b1)
data[:,:,:3][mask] = [r2, g2, b2]

im = Image.fromarray(data)
im.save('fig1_modified.png')

    Il va utiliser un peu (3x) plus de mémoire, mais il devrait être considérablement (~5x, mais plus pour les images de taille plus importante) plus rapide.

    Noter aussi que le code ci-dessus est un peu plus compliqué que cela doit l'être si vous n'avez RVB (et pas RGBA) des images. Cependant, cet exemple va quitter la bande alpha seul, alors qu'une version plus simple de ne pas avoir.

    OriginalL'auteur Joe Kington

  2. 4

    J'ai juste trouvé cette solution:

    import Image
im = Image.open("MyImage.png")
width, height = im.size
colortuples = im.getcolors()
mycolor1 = min(colortuples)[1]
mycolor2 = max(colortuples)[1]
pix = im.load()
for x in range(0, width):
    for y in range(0, height):
        if pix[x,y] == mycolor1:
            im.putpixel((x, y), mycolor2)
im.save('MyImage.png')

    Bien que putpixel n'est pas rapide, il semble être assez rapide pour moi.

    N'a probablement pas faire une différence de vitesse sage, mais vous pouvez le faire pix[x,y] = mycolor2 au lieu d'utiliser im.putpixel().
    J'ai essayé ceci, mais j'obtiens une erreur à cause de la mode ('r') dans laquelle j'ai ouvert l'Image. J'ai essayé Image.open("MyImage.png", "w") , mais il ne fonctionne pas. Je n'ai aucune idée de comment je pourrais utiliser pix[x,y] = mycolor2
    je ne sais pas pourquoi ça ne marcherait pas, peut-être un bug?

    OriginalL'auteur Martin Thoma

  3. 2

    C'est une modification de Joe kingston portant la réponse ci-dessus. Voici comment procéder si votre image contient un canal alpha.

    import numpy as np
import Image

im = Image.open('fig1.png')
im = im.convert('RGBA')
data = np.array(im)

r1, g1, b1 = 0, 0, 0 # Original value
r2, g2, b2, a2 = 255, 255, 255, 255 # Value that we want to replace it with

red, green, blue, alpha = data[:,:,0], data[:,:,1], data[:,:,2], data[:,:,3]
mask = (red == r1) & (green == g1) & (blue == b1)
data[:,:,:4][mask] = [r2, g2, b2, a2]

im = Image.fromarray(data)
im.save('fig1_modified.png')

    Il m'a fallu du temps pour comprendre comment le faire fonctionner. J'espère que cela aide quelqu'un d'autre.

    OriginalL'auteur hazzey

  4. 0

    Cette solution utilise glob à modifier toutes les png dans un dossier, la suppression d'une couleur et d'échanger avec l'autre, mais utilise RGBA. 

    import glob
from PIL import Image

old_color = 255, 0, 255, 255
new_color = 0, 0, 0, 0

for path in glob.glob("*.png"):
    if "__out" in path:
        print "skipping on", path
        continue

    print "working on", path

    im = Image.open(path)
    im = im.convert("RGBA")
    width, height = im.size
    colortuples = im.getcolors()

    pix = im.load()
    for x in xrange(0, width):
        for y in xrange(0, height):
            if pix[x,y] == old_color:
                im.putpixel((x, y), new_color)

    im.save(path+"__out.png")

    C'est une modification de https://stackoverflow.com/a/6483549/541208

    OriginalL'auteur TankorSmash