Experiments/finalPixel.py at main · Drarrryl/Experiments · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
from PIL import Image, ImageColor
from webcolors import rgb_to_name
from functions import returnGreat, closest_colour, inColBounds, rgb2name
from newPixel import zoom, coordBound

im = Image.open(r"/workspace/Experiments/Pictures/ocean3.png")
newImg = zoom(im)
px = newImg.load()

# Max and Min for Pixels in Image
xBound, yBound = newImg.size

img = Image.new('RGB', (xBound, yBound))

# Available Color Names: red, yellow, green, blue, plastic
colorname = "plastic"

needColor = True

for x in range(xBound):
    for y in range(yBound):
        r = px[x, y][0]
        g = px[x, y][1]
        b = px[x, y][2]
        pxColor = rgb2name((r, g, b))
        if inColBounds(pxColor, colorname):
            print("Found", colorname, "at coords:", x, ",", y)
            img.putpixel((x, y), (r, g, b))
        else:
            contrast = 50
            f = (returnGreat(r, g, b)-contrast)
            img.putpixel((x, y), (f, f, f))
img.save("newImg.jpg")
print("Image Saved!")