AI 一键去背景

AI百科2周前发布 快创云
11 0

  本文详细介绍如何使用 rembg 库轻松去除图片背景,并解析使用过程中可能遇到的一些常见问题及其解决方案。rembg 库:https://pypi.org/project/rembg/

  前文已提及如何利用 Paddle 库处理人物背景移除,本文将聚焦于通用图像背景移除,旨在保留图像主体。显然,rembg 的应用范围更为广泛,能够处理多种类型的图像背景。

  首先,通过 pip 安装 rembg 库:

  接下来,通过编写代码进行测试:

  测试中,我们继续使用之前的人像图片,以展示背景移除的效果。从结果来看,任务的手提包被完整地保留下来。

pexels 人像
pexels 人像

  总体而言,rembg 库的性能表现相当出色,尽管在某些情况下可能需要轻微调整,比如示例中的两只老虎,其中一只缺少耳朵,引发一阵欢笑……

pexels 动植物
pexels 动植物

常见问题及解决方案

问题一:cv2 版本问题

  在运行时可能会遇到以下错误:

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

  针对此问题,建议尝试更换 opencv 的版本。具体解决方案如下:

问题二:模型下载

  有时会遇到模型下载错误,例如:

下载报错
下载报错

  检查源码后发现,模型文件需通过 drive 网盘下载。提供百度网盘链接如下:

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

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

  下载并解压模型文件后,将 onnx 模型文件复制到用户目录下的指定文件夹即可。

© 版权声明

相关文章