python - 如何使用pillow的python lib实现图像 channel 下降

Question

我需要随机将枕头图像的 0、1 或 2 个 channel 归零。这意味着我需要将图像的 0、1 或 2 个 channel 随机设置为 0。

我怎样才能用 pil 做到这一点？

Answer 1

这是一种简单的原生 PIL 方法，通过乘以变换来实现此目的。我设置了默认变换，所以数学看起来像这样:

newRed   = 1*oldRed  +  0*oldGreen  +  0*oldBlue  + constant
newGreen = 0*oldRed  +  1*OldGreen  +  0*OldBlue  + constant
newBlue  = 0*oldRed  +  0*OldGreen  +  1*OldBlue  + constant

然后我只需将 1 更改为 0，我希望 channel 归零。

#!/usr/bin/env python3

from PIL import Image

# Open image
im = Image.open('input.png').convert("RGB")

# Pre-set R, G and B multipliers to 1
Rmult, Gmult, Bmult = 1, 1, 1

# Select one (or more) channels to zero out, I choose B channel here
Bmult=0

# Make transform matrix
Matrix = ( Rmult, 0, 0, 0,
           0, Gmult, 0, 0,
           0, 0, Bmult, 0)

# Apply transform and save
im = im.convert("RGB", Matrix)
im.save('result.png')

所以，如果你从这个开始:

并将蓝色乘数 (Bmult) 设置为零，您将得到:

如果将红色和蓝色归零:

Rmult = Bmult = 0 

你会得到: