如何通过Python代码指定rembg模型进行图像抠图

在Python中，你可以通过rembg库的new_session()方法指定目标模型，替代默认的u2net模型，实现和命令行rembg -m参数相同的效果。

步骤与示例代码 #

1. 确保已安装rembg库（未安装的话执行以下命令）：

pip install rembg

2. 编写代码指定模型处理图像：

from rembg import remove, new_session
from PIL import Image

# 替换为你需要使用的模型名称，可选值包括u2netp、u2net_human_seg、u2net_cloth_seg、isnet-general-use等
target_model = "u2net_human_seg"
# 创建指定模型的会话
model_session = new_session(target_model)

# 读取输入图像
input_image = Image.open("你的输入图片路径.jpg")
# 使用指定模型去除背景
output_image = remove(input_image, session=model_session)
# 保存处理后的图像
output_image.save("处理后图片路径.png")

说明 #

• 支持的模型名称和命令行rembg -m参数完全一致，可根据场景选择（比如u2net_human_seg适合人物抠图，u2net_cloth_seg适合衣物类抠图）
• 首次使用某模型时，库会自动下载对应模型文件并缓存，后续调用无需重复下载

内容的提问来源于stack exchange，提问作者风说起就起