У меня есть два изображения, оба с альфа-каналами. Я хочу поместить одно изображение поверх другого, в результате чего получится новое изображение с альфа-каналом, как если бы они были визуализированы в слоях. Я хотел бы сделать это с Python Imaging Library, но рекомендации в других системах были бы фантастическими, даже грубая математика была бы благом; Я мог бы использовать NumPy.
С помощью библиотеки изображений Python (PIL), как создать изображение с альфа-каналом над другим изображением?
Ответ 1
Я не мог найти функцию alpha composite в PIL, так что вот моя попытка реализовать ее с numpy:
import numpy as np
import Image
def alpha_composite(src, dst):
'''
Return the alpha composite of src and dst.
Parameters:
src -- PIL RGBA Image object
dst -- PIL RGBA Image object
The algorithm comes from http://en.wikipedia.org/wiki/Alpha_compositing
'''
# http://stackoverflow.com/a/3375291/190597
# http://stackoverflow.com/a/9166671/190597
src = np.asarray(src)
dst = np.asarray(dst)
out = np.empty(src.shape, dtype = 'float')
alpha = np.index_exp[:, :, 3:]
rgb = np.index_exp[:, :, :3]
src_a = src[alpha]/255.0
dst_a = dst[alpha]/255.0
out[alpha] = src_a+dst_a*(1-src_a)
old_setting = np.seterr(invalid = 'ignore')
out[rgb] = (src[rgb]*src_a + dst[rgb]*dst_a*(1-src_a))/out[alpha]
np.seterr(**old_setting)
out[alpha] *= 255
np.clip(out,0,255)
# astype('uint8') maps np.nan (and np.inf) to 0
out = out.astype('uint8')
out = Image.fromarray(out, 'RGBA')
return out
Например, учитывая эти два изображения,
img1 = Image.new('RGBA', size = (100, 100), color = (255, 0, 0, 255))
draw = ImageDraw.Draw(img1)
draw.rectangle((33, 0, 66, 100), fill = (255, 0, 0, 128))
draw.rectangle((67, 0, 100, 100), fill = (255, 0, 0, 0))
img1.save('/tmp/img1.png')
img2 = Image.new('RGBA', size = (100, 100), color = (0, 255, 0, 255))
draw = ImageDraw.Draw(img2)
draw.rectangle((0, 33, 100, 66), fill = (0, 255, 0, 128))
draw.rectangle((0, 67, 100, 100), fill = (0, 255, 0, 0))
img2.save('/tmp/img2.png')
alpha_composite
производит:
img3 = alpha_composite(img1, img2)
img3.save('/tmp/img3.png')
Ответ 2
Это похоже на трюк:
from PIL import Image
bottom = Image.open("a.png")
top = Image.open("b.png")
r, g, b, a = top.split()
top = Image.merge("RGB", (r, g, b))
mask = Image.merge("L", (a,))
bottom.paste(top, (0, 0), mask)
bottom.save("over.png")
Ответ 3
Pillow 2.0 теперь содержит функцию alpha_composite
, которая делает это.
img3 = Image.alpha_composite(img1, img2)