1 换取抠图API

到RemoveBg官网注册账号免费获得API，每月可用50次。
RemoveBg官网：https://www.remove.bg/r/a9Dy3RhvgnKDphmsMbarJYrG
image.png

2 安装现需要的库

  1. pip install removebg
  2. pip install pillow（python2安装：pip install PIL
  3. pip install shutil

3 编写代码

  1. #!/usr/bin/env python
  2. # _*_ coding: utf-8 _*_
  3. # @Time : 2021/10/12 15:07
  4. # @Author : cfxin
  5. # @File : photo-background-color-change.py
  6. # @Software: PyCharm
  7. # -*- 功能说明 -*-
  8. # 图片抠图和更换背景颜色
  9. # -*- 功能说明 -*-
  10. import os
  11. import shutil
  12. from PIL import Image
  13. from removebg import RemoveBg
  15. # 证件照背景色
  16. Background_Color = {
  17.     'Red': (255, 0, 0, 255),
  18.     'Blue': (67, 142, 219, 255),
  19.     'White': (255, 255, 255, 255)
  20. }
  21. pic = input("Tip：请将图片移至当前所在目录下！\n请输入图片文件名(含文件后缀)：")  # 图片文件名
  22. print("当前所在目录：" + os.getcwd())
  24. if os.path.exists('img'):
  25.     print("\n当前目录已存在img文件夹！！！")
  26. else:
  27.     print("\n正在新建输出目录img ...")
  28.     new_folder = '%s' % os.mkdir(os.getcwd() + '\\img')  # 新建文件夹
  29.     print("输出目录img创建成功！！！\n")
  31. os.chdir('./img')  # 切换至img目录下
  33. if not os.path.exists(pic):
  34.     print("\n正在复制图片至img文件夹......")
  35.     shutil.copy('../' + pic, pic)
  36.     print("目标图片已复制到img文件夹！\n")
  38. path = '%s' % os.getcwd()  # 当前所在路径
  39. print("当前目标图片路径：\t" + '%s\%s' % (path, pic))  # 输出图片所在路径
  41. # Api每月可用50次，降低同一张图片重复抠图频率
  42. if os.path.exists(pic + '_no_bg.png'):
  43.     print("\n当前目录已存在透明背景图片" + pic + '_no_bg.png\n')
  44. else:
  45.     print("\n正在生成错误日志(可不用理会)......")
  46.     rmbg = RemoveBg("Your-RemoveBg-Api-Key", "error.log")  # 密钥，错误日志
  47.     print("正在进行抠图......")
  48.     rmbg.remove_background_from_img_file('%s\%s' % (path, pic))  # 抠图
  49.     print("已生成透明背景图片：" + pic + "_no_bg.png\n")
  51. select_bgColor = input("Tip：更换红色背景按 r，更换蓝色背景按 b，更换白色背景按 w \n请选择需要更换的证件照底色：")  # 选择所需更换的底色
  53. while True:
  54.     if select_bgColor == 'r':
  55.         color = 'Red'
  56.         break
  57.     elif select_bgColor == 'b':
  58.         color = 'Blue'
  59.         break
  60.     elif select_bgColor == 'w':
  61.         color = 'White'
  62.         break
  63.     else:
  64.         select_bgColor = input("\n输入有误！请重新输入：")
  66. foreground = Image.open(pic + '_no_bg.png')  # 前景图
  67. background = Image.new('RGBA', foreground.size, Background_Color[color])  # 背景图，大小同前景图
  68. background.paste(foreground, mask=foreground)
  70. print("\n正在展示图片......")
  71. background.show()  # 显示图片
  73. print("正在保存图片......\n")
  74. new_pic = pic + '_' + color + '.png'  # 输出文件名
  75. background.save(os.getcwd() + '\\' + new_pic)  # 保存图片
  77. print("输出路径为：" + os.getcwd())
  78. os.chdir('../')  # 切换回原工作目录
  79. print("\n保存成功！输出文件为：" + new_pic + "\n\n按任意键退出\n")  # 显示保存图片名
  80. os.system('pause')  # 按任意键继续

到此处已经可以实现抠图并更换背景颜色的功能。为了方便使用可以按下面的步骤打包成可执行程序。

4 打包成exe可执行文件

  1. 安装PyInstaller包：pip install PyInstaller
  2. 切换至该刚才编写的 .py 文件所在目录下，命令行窗口输入pyinstaller -F Your_File_Name.py

image.png

  1. 打包成功后会出现3个文件夹和一个文件，打包成功的.exe 文件在 dist 目录下

image.png