基于 PaddleHub 人像摳圖進行證件照背景色修改，即使背景凌亂，也能摳出人像，更換背景色，在家也能拍證件照。核心代碼僅數行。

先安裝 PaddleHub

pip install paddlehub==1.6.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

實現代碼

# 導入需要的庫 import paddlehub as hub import matplotlib.image as mpimg from PIL import Image import numpy as np import matplotlib.pyplot as plt# 基礎設置 img_path = "photo.jpg" # 需要換背景色的圖片 background_color = 'white' # 背景顏色：紅 red/藍 blue/白 white/自定義 (r,g,b) sava_path = "./IDphoto.jpg" # 結果圖片路徑# 背景色設置 if isinstance(background_color, str):if background_color == 'red':color = [255, 0, 0, 1]elif background_color == 'blue':color = [67, 142, 219, 1]elif background_color == 'white':color = [255, 255, 255, 1]else:raise Exception('背景色設置有誤') elif isinstance(background_color, list) or isinstance(background_color, tuple):color = [background_color[0], background_color[1], background_color[2], 1] else:raise Exception('背景色設置有誤')# 進行摳圖 test_img_path = ['./' + img_path] module = hub.Module(name="deeplabv3p_xception65_humanseg") input_dict = {"image": test_img_path} results = module.segmentation(data=input_dict)# 顏色修改 human_img = './humanseg_output/' + img_path[:-4] + '.png' base_image = Image.open(human_img) img = np.array(base_image)for i in range(0, img.shape[0]):for j in range(0, img.shape[1]):if img[i][j][3] < 1:img[i][j] = color# 生成證件照 im = Image.fromarray(img) im = im.convert('RGB') im.save(sava_path)

搞定！

總結

以上是生活随笔為你收集整理的AI证件照背景色修改，几行代码搞定的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。