AI 一键去背景

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

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

使用 pip 安装：

编写代码进行测试：

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

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

问题一：cv2 版本问题

运行报错：

AttributeError module ‘cv2’ has no attribute ‘gapi_wip_gst_GStreamerPipeline’

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

问题二：模型下载

模型下载错误：

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

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

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

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

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