如何使用Python去除图片绿幕，将背景调整为透明？

问题原因 #

• 你读取的JPG图片默认是RGB三通道模式，没有存储透明度的alpha通道，即便保存为PNG也无法呈现透明效果，原本的绿幕区域被你把G通道清零后就会显示为黑色。
• 当前代码直接将所有像素的G通道设为0，既没有判断绿幕区域，也没有设置透明度参数，还会导致图片中原本带绿色的正常内容出现偏色。

修正方案 #

将图片转为带alpha通道的RGBA模式，通过阈值判断识别绿幕区域，将对应像素的alpha值设为0（完全透明）即可，参考代码如下：

from PIL import Image

# 读取原图
image = Image.open('pic.jpg')
# 转换为带透明通道的RGBA模式
image = image.convert('RGBA')
pixel_data = image.load()
height, width = image.size

# 绿幕颜色阈值，可根据实际绿幕深浅调整数值
lower_green = (0, 170, 0)
upper_green = (90, 255, 90)

for x in range(height):
    for y in range(width):
        r, g, b, a = pixel_data[x, y]
        # 判断像素是否属于绿幕范围
        if lower_green[0] <= r <= upper_green[0] and lower_green[1] <= g <= upper_green[1] and lower_green[2] <= b <= upper_green[2]:
            # 绿幕区域设置为完全透明
            pixel_data[x, y] = (r, g, b, 0)

# 保存为支持透明的PNG格式
image.save('changed.png')

使用提示 #

• 若抠图边缘有残留或者正常内容被误判为透明，调整lower_green和upper_green的RGB阈值即可，所有数值取值范围为0-255。
• 处理大尺寸图片时，可以用numpy矩阵运算替代嵌套循环，大幅提升处理速度。

内容的提问来源于stack exchange，提问作者CatalinBranc