python - PIL 中是否有一个 Image.point() 方法可以让您同时对所有三个 channel 进行操作？

Question

我想为每个像素编写一个基于红色、绿色和蓝色 channel 的点过滤器，但看起来这可能达不到 point() 的能力——它似乎它一次在单个 channel 中的单个像素上运行。我想做这样的事情:

def colorswap(pixel):
    """Shifts the channels of the image."""
    return (pixel[1], pixel[2], pixel[0])
image.point(colorswap)

有没有等效的方法可以让我使用一个接受 RGB 值的三元组并输出一个新的三元组的过滤器？

Answer 1

您可以使用load 方法来快速访问所有像素。

def colorswap(pixel):
    """Shifts the channels of the image."""
    return (pixel[1], pixel[2], pixel[0])

def applyfilter(image, func):
    """ Applies a function to each pixel of an image."""
    width,height = im.size
    pixel = image.load()
    for y in range(0, height):
        for x in range(0, width):
            pixel[x,y] = func(pixel[x,y])

applyfilter(image, colorswap)