patchy/keys
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Switch to rawpy/PIL and simplify

Browse Source
This commit is contained in:
Ian Gulliver
2024-08-11 14:28:00 -07:00
parent 81a9fd5fb8
commit 35d78bf37e
2 changed files with 57 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
chromakey.py
View File

@@ -6,18 +6,11 @@ from PIL import Image, ImageOps
path = 'data/RAW/SC1/BR1'
filename = os.path.join(path, '00570/back.orf')
raw = rawpy.imread(filename).postprocess()
rgba = Image.fromarray(raw).convert('RGBA')
raw = rawpy.imread(filename)
rgb_image = raw.postprocess()
h, _, _ = rgba.convert('HSV').split()
mask = ImageOps.invert(Image.eval(h, lambda x: 255 if 50 < x < 90 else 0))
image = Image.fromarray(rgb_image).convert('RGBA')
hsv_image = image.convert('HSV')
h, s, v = hsv_image.split()
green_mask = Image.eval(h, lambda x: 255 if 50 < x < 90 else 0)
green_mask = ImageOps.invert(green_mask)
rgba_image = image.copy()
rgba_image.putalpha(green_mask)
rgba_image.save('output.png', 'PNG')
rgba.putalpha(mask)
rgba.save('output.png', 'PNG')