本文介绍如何使用 rembg 库去除图片背景，并解决其使用过程中的一些问题。rembg 库：https://pypi.org/project/rembg/
上一篇文章介绍了如何通过 Paddle 库去除人物背景，这一篇介绍如何去除通用图像背景，只保留图像主体。显而易见，一个只可以处理人像，一个则可以去除多种图像的背景。
安装使用
使用 pip 安装：
pip install rembg
编写代码进行测试：
from rembg import remove
from PIL import Image
input =Image.open("pexels-5872320.jpg")
output=remove(input)
output.save("output.png")
效果展示
这里我们一样使用之前的人像测试，因为只是去背景，我们可以看到任务的手提包相较于 deeplabv3p_xception65_humanseg 是被完整的保留了下来。
rembg 库的效果总体来说还是非常棒的，虽然有时候可能会有一些个小问题需要稍微修一下图，比如两只老虎，一只没有耳朵，哈哈哈……
可能出现的问题
问题一：cv2 版本问题
运行报错：
AttributeError: module ‘cv2’ has no attribute ‘gapi_wip_gst_GStreamerPipeline’
尝试更改 opencv 版本，解决方案：
pip install opencv-python==4.6.0.66
问题二：模型下载
模型下载错误：
这里检查了下源码，发现模型文件需要通过 drive 网盘下载。
想办法搞到模型文件，放到其指定的用户目录下即可。这里分享在百度网盘：
链接: https://pan.baidu.com/s/172OeWM9ogHzXmB3PnSrf0Q?pwd=e28e 提取码: e28e
或前往 csdn 下载 https://download.csdn.net/download/marin1993/86845047
下载解压模型文件，将这些 onnx 模型文件复制到用户目录下的 .u2net 文件夹即可。