python - 在使用 PIL 保持透明度的同时为图像着色？

Question

好的，情况如下：

我想使用 Python 图像库来“主题化”这样的图像：

主题颜色： "#33B5E5"

输入： 输出：

我在 ImageMagick 中使用这个命令得到了结果：

convert image.png -colorspace gray image.png
mogrify -fill "#33b5e5" -tint 100 image.png

解释：

图像首先转换为黑白，然后进行主题化。

我想用 Python Image Library 得到同样的结果。但似乎我在使用它时遇到了一些问题，因为：

1. 无法处理透明度
2. 背景（主图像中的透明度）也成为主题。

我正在尝试使用此脚本：

import Image
import ImageEnhance

def image_overlay(src, color="#FFFFFF", alpha=0.5):
    overlay = Image.new(src.mode, src.size, color)
    bw_src = ImageEnhance.Color(src).enhance(0.0)
    return Image.blend(bw_src, overlay, alpha)

img = Image.open("image.png")
image_overlay(img, "#33b5e5", 0.5)

您可以看到我没有先将其转换为灰度，因为这也不适用于透明度。

很抱歉在一个问题中发布了这么多问题，但我无能为力：$

希望大家理解。

Answer 1

注意：这里有这个答案的 PIL 版本的 Python 3/pillow 叉。

更新 4：猜猜我的答案的上一个更新毕竟不是最后一个。尽管将其转换为PIL专用是一项重大改进，但有几件事似乎应该有更好的、不那么尴尬的方法，只要PIL有能力。

好吧，在仔细阅读了文档以及一些源代码之后，我意识到我想做的事情实际上是可能的。权衡是现在它必须手动构建查找表，因此整体代码略长。然而结果是它只需要调用一个相对慢的Image.point()方法，而不是三个。

from PIL import Image
from PIL.ImageColor import getcolor, getrgb
from PIL.ImageOps import grayscale

def image_tint(src, tint='#ffffff'):
    if Image.isStringType(src):  # file path?
        src = Image.open(src)
    if src.mode not in ['RGB', 'RGBA']:
        raise TypeError('Unsupported source image mode: {}'.format(src.mode))
    src.load()

    tr, tg, tb = getrgb(tint)
    tl = getcolor(tint, "L")  # tint color's overall luminosity
    if not tl: tl = 1  # avoid division by zero
    tl = float(tl)  # compute luminosity preserving tint factors
    sr, sg, sb = map(lambda tv: tv/tl, (tr, tg, tb))  # per component adjustments

    # create look-up tables to map luminosity to adjusted tint
    # (using floating-point math only to compute table)
    luts = (map(lambda lr: int(lr*sr + 0.5), range(256)) +
            map(lambda lg: int(lg*sg + 0.5), range(256)) +
            map(lambda lb: int(lb*sb + 0.5), range(256)))
    l = grayscale(src)  # 8-bit luminosity version of whole image
    if Image.getmodebands(src.mode) < 4:
        merge_args = (src.mode, (l, l, l))  # for RGB verion of grayscale
    else:  # include copy of src image's alpha layer
        a = Image.new("L", src.size)
        a.putdata(src.getdata(3))
        merge_args = (src.mode, (l, l, l, a))  # for RGBA verion of grayscale
        luts += range(256)  # for 1:1 mapping of copied alpha values

    return Image.merge(*merge_args).point(luts)

if __name__ == '__main__':
    import os

    input_image_path = 'image1.png'
    print 'tinting "{}"'.format(input_image_path)

    root, ext = os.path.splitext(input_image_path)
    result_image_path = root+'_result'+ext

    print 'creating "{}"'.format(result_image_path)
    result = image_tint(input_image_path, '#33b5e5')
    if os.path.exists(result_image_path):  # delete any previous result file
        os.remove(result_image_path)
    result.save(result_image_path)  # file name's extension determines format

    print 'done'

这是一个屏幕截图，左侧显示输入图像，右侧显示相应输出。上面一行是带有 alpha 层的，下面是类似的，没有一层。

Answer 2

您需要先转换为灰度。我做了什么：

1. 使用获取原始 alpha 层Image.split()
2. 转换为灰度
3. 着色使用ImageOps.colorize
4. 放回原来的alpha层

结果代码：

import Image
import ImageOps

def tint_image(src, color="#FFFFFF"):
    src.load()
    r, g, b, alpha = src.split()
    gray = ImageOps.grayscale(src)
    result = ImageOps.colorize(gray, (0, 0, 0, 0), color) 
    result.putalpha(alpha)
    return result

img = Image.open("image.png")
tinted = tint_image(img, "#33b5e5")