证件照改底色（背景色）

黑麋鹿收录于类别图像处理

2023-10-06 2023-10-06 约 491 字预计阅读 1 分钟

作用

将证件照杂乱背景改为指定纯色背景，同时修改照片尺寸。所用到的库是rembg
详细代码如下：

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30


from PIL import Image
import rembg
def process_image(input_path):
    # 打开源图像
    input_img = Image.open(input_path)
    # 移除背景
    output = rembg.remove(input_img)

    # 以 4:3 的比例计算宽度.
    new_width = int(output.height * 3 / 4)

    # 计算起始X坐标的裁剪，以确保受试者保持居中。
    x_start = (output.width - new_width) // 2

    # 裁剪图像。
    cropped_output = output.crop((x_start, 0, x_start + new_width, output.height))

    # 创建一个带有裁剪尺寸的新图像，并用蓝色填充。
    blue_background = Image.new("RGB", cropped_output.size, "#438EDB") # #438EDB为颜色代码

    # 将裁剪的图像粘贴到蓝色背景上。
    blue_background.paste(cropped_output, (0, 0), cropped_output)

    # 保存结果图像。
    inputname = input_path.split(".")[0]
    blue_background.save(f'out_{inputname}.jpg')

# 调用函数
input_path = 'tsett.JPG'
process_image(input_path)