python给证件照换底色

import cv2
import numpy as np
from tkinter import *
from tkinter import ttk
import tkinter as tk
import tkinter.filedialog
import os

root = Tk()
root.title('证件照背景色更换')
root.geometry('400x300')
SelectNum = tk.IntVar()  # 用来储存下面勾选项1中返回的0或１

FileName = ""
UsingColour = ""
DesColour = ""
DesSize1 = ""
DesSize2 = ""
BlueRange = [[90, 70, 70], [110, 255, 255]]
RedRange = [[0,135,135], [180,245,230]]
WhiteRange = [[0,135,135], [255 255 255]]
Blue = [219, 142, 67]
Red = [0, 0, 255]
White = [255, 255, 255]

def Button1Anwser():
    global FileName
    Path = Label(root, text='')
    FileName = tkinter.filedialog.askopenfilename()
    Path.config(text = FileName)
    Path.grid(row=2, column=0, sticky=W)

def Button2Anwser():
    global DesColour
    global DesSize1
    global DesSize2

    ColNum = str(Descmb.get())
    if ColNum == "蓝色":
        DesColour = Blue
    if ColNum == "红色":
        DesColour = Red
    if ColNum == "白色":
        DesColour = White

    img = cv2.imread(r'D:\id_photo\000.jpg')
    img = cv2.resize(img,None,fx=0.5,fy=0.5)

    Text = Label(root, text='')
    Text.config(text='正在转换')
    Text.grid(row=9, column=1)

# 缩放
    rows, cols, channels = img.shape  # 长 宽 通道数
    rows, cols, channels = img.shape
    cv2.imshow('img', img)

    # 转换hsv
    hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)  # rgb转hsv
    lower_colour = np.array([WhiteRange[0]])
    upper_colour = np.array([WhiteRange[1]])
    # 去除背景，低于lower_red和高于upper_red的部分分别变成0，lower_red～upper_red之间的值变成255
    mask = cv2.inRange(hsv, lower_colour, upper_colour)

    # 腐蚀膨胀
    erode = cv2.erode(mask, None, iterations=1)  # 腐蚀
    dilate = cv2.dilate(erode, None, iterations=1)  # 膨胀

    # 遍历替换
    for i in range(rows):
        for j in range(cols):
            if dilate[i, j] == 255:
                img[i, j] = (DesColour)  # 此处替换颜色，为BGR通道
    cv2.imshow('res', img)

    Text = Label(root, text='')
    Text.config(text='转换完成')
    Text.grid(row=9, column=1)
    if SelectNum.get() == 1:
        DesSize1 = int(Size1.get())
        DesSize2 = int(Size2.get())
        img = cv2.resize(img, (DesSize1, DesSize2), interpolation=cv2.INTER_CUBIC)
    cv2.imwrite(r'D:\id_photo\Result1.jpg', img)

    cv2.waitKey(0)
    cv2.destroyAllWindows()

# -------------------------------------界面---------------------------------
SelectFile = Button(root, text="待处理照片", command=Button1Anwser,width=10, height=1, bg="DarkGray", fg="white")
SelectFile.grid(row=1, column=0, sticky=W)

UsingCol = Label(root, text='')
UsingCol.config(text='当前背景颜色')
UsingCol.grid(row=3, column=0, sticky=W)

Usingcmb = ttk.Combobox(root)
Usingcmb.grid(row=4, column=0, sticky=W)
Usingcmb['value'] = ('白色')

DesCol = Label(root, text='')
DesCol.config(text='目标背景颜色')
DesCol.grid(row=5, column=0, sticky=W)

Descmb = ttk.Combobox(root)
Descmb.grid(row=6, column=0, sticky=W)
Descmb['value'] = ('蓝色', '红色', '白色')

PicSize1 = Label(root, text='目标像素：')
PicSize1.grid(row=7, sticky=W)

Select = tk.Checkbutton(root, text='调整像素', variable=SelectNum, onvalue=1, offvalue=0)
Select.grid(row=7, column=1, sticky=W)

Size1 = Entry(root)
Size1.place(width=5, height=1)
Size1.grid(row=8, column=0, sticky=W)
MultiText = Label(root, text='* ')
MultiText.grid(row=8, column=1)
Size2 = Entry(root)
Size2.place(width=5, height=1)
Size2.grid(row=8, column=2, sticky=W)

SelectFile = Button(root, text="开始转换", command=Button2Anwser, width=10, height=1, bg="DarkGray", fg="white")
SelectFile.grid(row=9, sticky=W)

 ps. 原图为有色底色实现更改

2. 一键换底色

import requests
import os
from PIL import Image
#第一阶段，利用api去除背景色
os.chdir(r'D:\id_photo')
response = requests.post(
    'https://api.remove.bg/v1.0/removebg',
    files={'image_file': open('000.jpg', 'rb')},
    data={'size': 'auto'},
    headers={'X-Api-Key': 'JSL7bAjXKGuSK9M9Gi7LKvip'},
)
if response.status_code == requests.codes.ok:
    with open('no-bg.png', 'wb') as out:
        out.write(response.content)
else:
    print("Error:", response.status_code, response.text)
print("想换成的颜色： blue for 蓝色，red for 红色 white for 白色 ,回车 for no_change\n:")
#第二阶段，利用PIL库填充背景色
color=input()
im = Image.open('no-bg.png')
x,y = im.size
try:
  # (alpha band as paste mask).
  if color=='':
      print('no change')
      pass
  elif color=='blue':
      p = Image.new('RGBA', im.size, (0, 0, 255))
      p.paste(im, (0, 0, x, y), im)
      p.save('bg_blue.png')
  elif color=='red':
      p = Image.new('RGBA', im.size, (255, 0, 0))
      p.paste(im, (0, 0, x, y), im)
      p.save('bg_red.png')
  elif color=='white':
      p = Image.new('RGBA', im.size, (255, 255, 255))
      p.paste(im, (0, 0, x, y), im)
      p.save('bg_white.png')
  else:
      print('no match!')
except:
  pass