Script to replace png image color

change_png_image_color.py

#!/usr/bin/env python
import sys
import numpy as np
from PIL import Image

if len(sys.argv) != 2:
  print 'Invalid number of arguments'
  sys.exit()

NEW_PATH = 'images/fig.png'

# R_OLD, G_OLD, B_OLD = (255, 255, 255)
R_NEW, G_NEW, B_NEW = (255, 255, 255)


im = Image.open(sys.argv[1])
pixels = im.load()

width, height = im.size
for x in range(width):
    for y in range(height):
        r, g, b, a = pixels[x, y]
        # if (r, g, b) == (R_OLD, G_OLD, B_OLD):
        pixels[x, y] = (R_NEW, G_NEW, B_NEW, a)

im.save(NEW_PATH)