python去除图片背景_Python 图片去除背景

安装

pip install removebg

使用

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

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

找到 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