AI 一键去背景

本文介绍如何使用 rembg 库去除图片背景，并解决其使用过程中的一些问题。rembg 库：https://pypi.org/project/rembg/

上一篇文章介绍了如何通过 Paddle 库去除人物背景，这一篇介绍如何去除通用图像背景，只保留图像主体。显而易见，一个只可以处理人像，一个则可以去除多种图像的背景。

安装使用

使用 pip 安装：

pip install rembg
1

编写代码进行测试：

from rembg import remove
from PIL import Image

input =Image.open("pexels-5872320.jpg")
output=remove(input)
output.save("output.png")
1
2
3
4
5
6

效果展示

这里我们一样使用之前的人像测试，因为只是去背景，我们可以看到任务的手提包相较于 deeplabv3p_xception65_humanseg 是被完整的保留了下来。

rembg 库的效果总体来说还是非常棒的，虽然有时候可能会有一些个小问题需要稍微修一下图，比如两只老虎，一只没有耳朵，哈哈哈……

可能出现的问题

问题一：cv2 版本问题

运行报错：

AttributeError: module ‘cv2’ has no attribute ‘gapi_wip_gst_GStreamerPipeline’

尝试更改 opencv 版本，解决方案：

pip install opencv-python==4.6.0.66
1

问题二：模型下载

模型下载错误：

这里检查了下源码，发现模型文件需要通过 drive 网盘下载。

想办法搞到模型文件，放到其指定的用户目录下即可。这里分享在百度网盘：

链接: https://pan.baidu.com/s/172OeWM9ogHzXmB3PnSrf0Q?pwd=e28e 提取码: e28e

或前往 csdn 下载 https://download.csdn.net/download/marin1993/86845047

下载解压模型文件，将这些 onnx 模型文件复制到用户目录下的 .u2net 文件夹即可。