Python将证件背景更改为白色底

import cv2
from PIL import Image

# 读取原始图片
img = cv2.imread('original_id_card.jpg')

# 转换到HSV色彩空间，便于区分前景和背景
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)

# 定义背景颜色范围（这里假设背景不是白色，需根据实际证件照调整）
lower_white = (0, 0, 200)
upper_white = (255, 255, 255)

# 进行阈值处理得到二值图像，其中白色区域为0
mask = cv2.inRange(hsv, lower_white, upper_white)

# 反转mask，使得人物部分为白色，背景为黑色
mask_inv = cv2.bitwise_not(mask)

# 将原图与反向掩模做位运算，提取出人像部分
foreground = cv2.bitwise_and(img, img, mask=mask_inv)

# 创建一个纯白色背景图片
width, height = img.shape[:2]
white_bg = np.full((height, width, 3), (255, 255, 255), dtype=np.uint8)

# 将人像部分粘贴到白色背景上
result = cv2.bitwise_or(white_bg, foreground)

# 使用PIL保存最终图片
pil_img = Image.fromarray(cv2.cvtColor(result, cv2.COLOR_BGR2RGB))
pil_img.save('new_id_card_with_white_bg.jpg')