rembg Python 库:自动抠图的简单指南和代码示例
rembg 是一个用于自动化图像背景去除的 Python 库。它使用深度学习模型,可以实现高质量的图像背景去除。以下是一个简单的代码示例,演示如何使用 rembg 进行图像背景去除:
import rembg
from PIL import Image
# 加载图像
img = Image.open('input_image.png')
# 运行rembg模型,去除背景
output = rembg.remove(img)
# 将去除背景后的图像保存为新文件
output.save('output_image.png')
上述代码示例中,首先加载了一个 PNG 格式的输入图像,然后使用 rembg 库中的 remove() 函数对该图像进行背景去除。最后,将去除背景后的图像保存为 PNG 格式的输出图像。
需要注意的是,rembg 库需要依赖于 PyTorch 深度学习框架,因此在使用之前需要先安装 PyTorch。另外,由于 rembg 是一个基于深度学习的库,因此在处理大型图像时可能会需要比较长的处理时间。
原文地址: https://www.cveoy.top/t/topic/m6zJ 著作权归作者所有。请勿转载和采集!