Python 图片去除背景

安装

pip install removebg 

使用

1. 官网注册账号：https://www.remove.bg/users/sign_up

2. 阅读 API 文档: https://www.remove.bg/api

3. 找到 python 相关的 API 示例如下：

    # Requires "requests" to be installed (see python-requests.org)
    import requests
    
    response = requests.post(
        'https://api.remove.bg/v1.0/removebg',
        files={'image_file': open('/path/to/file.jpg', 'rb')},
        data={'size': 'auto'},
        headers={'X-Api-Key': 'INSERT_YOUR_API_KEY_HERE'},
    )
    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)
   

4.拿到一个免费的 api key: https://www.remove.bg/profile#api-key
5.找到一张合适的实例图片：

然后我们就可以开始写调用程序了。

代码示例

有关 python 的调用示例可以参考： https://github.com/brilam/remove-bg

from removebg import RemoveBg
import os

# 参数填入 api key 以及批处理的日志文件位置
rmbg = RemoveBg("u3NcF5HuRAbKow7x9v7BJbfS", "./error.log")

# 批处理图片的存放位置
path = os.path.join(os.getcwd(), "pic")

for pic in os.listdir(path):
    rmbg.remove_background_from_img_file(os.path.join(path, pic))

文件的位置如下，处理之后会在图片的路径保留一份去除背景的图片：

将去除背景的图像换上其他的背景色

# 给去除了背景的图像添加各色背景

from PIL import Image

# 输入已经去除背景的图像
im = Image.open('/Users/furuiyang/mygit/pydailynotes/image/pic/lixian.jpeg_no_bg.png')
x, y = im.size

try:
    # 使用白色来填充背景
    # (alpha band as paste mask).
    p = Image.new('RGBA', im.size, (255, 0, 255))
    p.paste(im, (0, 0, x, y), im)
    # 保存转换后的退图像 
    p.save('/Users/furuiyang/mygit/pydailynotes/image/pic/lixian.jpeg_new_bg.png')
except:
    pass

实现的效果：

打包（TODO）

实现一个 Tk 图形用户界面 （TODO)

源代码文件

https://github.com/furuiyang0715/pydailynotes/tree/master/image