AI图片去背景

AI百科6个月前更新快创云

61 0

软硬件环境

Windows 10 64位操作系统
PyTorch 1.7版本
rembg工具

简介

　　rembg是一个强大的图片背景移除工具，不仅可以在Python中轻松调用，还提供了独立的可执行程序，使用极为便捷。

安装

　　您可以选择通过安装，或者从GitHub拉取源码进行安装。安装完成后，会生成可执行文件，便于后续使用。

使用命令行

　　首先，您需要下载模型文件。这里提供两个下载地址：

Google网盘链接
百度网盘链接：点击此处下载，提取码：

　　将下载的文件拷贝到指定的文件夹中（该文件夹默认不存在，需要您自行创建）。接下来，您就可以使用本地图片进行测试了。

处理效果展示

　　以下是处理后的图片示例：

　　除了处理单张图片外，rembg还支持对文件夹的处理。例如，将上述两张图片放在一个文件夹中，然后使用相应的命令进行处理。

　　此外，rembg还提供了Web端的接口，您可以通过启动后台服务来使用它。这里使用的是FastAPI框架。在浏览器中输入对应的URL（后面跟上图片的地址）即可进行测试，如：

代码中调用rembg

　　以下是一个示例代码，展示了如何在Python中调用rembg：

from rembg import remove_background_from_img_file

from PIL import Image

import requests

import io



# 读取本地图片并移除背景

image_path = 'path_to_your_image.jpg'

bg_image = remove_background_from_img_file(image_path)

bg_image.save('result.png')  # 保存处理后的图片



# 或者通过Web接口移除背景（需要确保服务已启动）

url = 'http://localhost:8000/remove_bg'  # 假设服务在本地8000端口运行

response = requests.post(url, files={'image': open(image_path, 'rb')})

result = Image.open(io.BytesIO(response.content))

result.show()  # 显示处理后的图片（仅适用于本地测试）

　　在终端中执行上述代码，将得到与前面示例相同的结果。