昨晚我用python幫隔壁小姐姐P證件照然后發(fā)現(xiàn)

#安裝python模塊
pip install pillow
pip install removebg

from PIL import Image
 
old_img = 'C:/Users/lex/desktop/img/詩(shī)詩(shī).png'
new_img = 'C:/Users/lex/desktop/img/詩(shī)詩(shī)-new.png'
img = Image.open(old_img)
#讀取照片尺寸
(x,y) = img.size
#重新設(shè)置照片尺寸
x_s = 295 #寬
y_s = 413 #高
out = img.resize((x_s,y_s),Image.ANTIALIAS) #resize image with high-quality
out.save(new_img)
 
print ('原始照片尺寸(寬x高): ',x,"x",y)
print ('調(diào)整后照片尺寸:(寬x高) ',x_s,"x",y_s)

BACKGROUND_COLOR = {
    'RED': (255, 0, 0, 255),
    'BLUE': (67, 142, 219, 255),
    'WHITE': (255, 255, 255, 255)
}

#老照片路徑、新照片路徑、無(wú)背景照片路徑、顏色
def get_img_bg(old_img_path,new_img_path,no_bg_img_path,color):
    #去掉背景圖,提取照片
    rmbg.remove_background_from_img_file(old_img_path)
    foreground = Image.open(no_bg_img_path)
    background = Image.new('RGBA', foreground.size, BACKGROUND_COLOR[color])  # 背景圖，大小同前景圖
    background.paste(foreground, mask=foreground)
    background.save(new_img_path)
 
if __name__ == '__main__':
    get_img_bg('C:/Users/pacer/Desktop/img/詩(shī)詩(shī).png','C:/Users/pacer/desktop/img/詩(shī)詩(shī)_red.png','C:/Users/pacer/desktop/img/詩(shī)詩(shī).png_no_bg.png','RED')
    get_img_bg('C:/Users/pacer/Desktop/img/詩(shī)詩(shī).png','C:/Users/pacer/desktop/img/詩(shī)詩(shī)_blue.png','C:/Users/pacer/desktop/img/詩(shī)詩(shī).png_no_bg.png','BLUE')