import os
import numpy as np
from PIL import Image

def lista_immagini():
    filelist=os.listdir(r"C:UsersfdivitoDesktopimage")
    for fichier in filelist[:]: # filelist[:] makes a copy of filelist.
        if not(fichier.endswith(".png")):
            filelist.remove(fichier)
    print(filelist)
    return filelist
brand = {'Ferrari': (0, 47, 128), 'Lamborghini': (196, 18, 48)}

for key, value in brand.iteritems():
    brand_model = key
    color_model_r = value[0]
    color_model_g = value[1]
    color_model_b = value[2]

    for i in lista_immagini():
        im = Image.open(r'path'+ i)
        data = np.array(im)

        r1, g1, b1 = 54, 115, 51 # Original value
        r2, g2, b2 = color_model_r, color_model_g, color_model_b # 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(r"path" + brand_model + "_" + i)
print("Done " + str(i))