第一步:首先安裝RemoveBg庫
pip install removebg
Removebg是利用人工智能算法實現的自動識別出前景主體與背景圖,處理速度極快。需要註冊獲取API接口,一個接口一個月只能處理50張圖片。
第二步:導入RemoveBg,輸入API接口
#處理單張圖片
from removebg import RemoveBg
rmbg = RemoveBg("RKxxxxx","error.log") #輸入獲取得API
rmbg.remove_background_from_img_file(r"C:\Jupyter\baby.n.png") #圖片地址
#處理多張圖片
from removebg import RemoveBg
import os
rmbg = RemoveBg("RKxxxxxxx","error.log") #輸入獲取得API
path = '%s/photoss'%os.getcwd() #圖片放到同級文件夾photos中
for pic in os.listdir(path):
rmbg.remove_background_from_img_file("%s\%s"%(path,pic))
第三步:複製已經去掉背景的圖片路徑,進行更換底色
#changebg:調用PIL添加背景色
from PIL import Image
im = Image.open(r"C:\Jupyter\baby.n.png_no_bg.png")
x,y = im.size
try:
p = Image.new('RGBA', im.size, (255,0,0))
p.paste(im, (0, 0, x, y), im)
p.save(r"C:\Jupyter\baby.n.png_no_bg.png")
except:
print('err')
好了,大功告成,用哥哥的照片試了一下,還請他的粉絲不要罵我