AI 一键去背景

简介: 本文介绍如何使用 rembg 库去除图片背景,并解决其使用过程中的一些问题。rembg 库:[https://pypi.org/project/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 是被完整的保留了下来。

pexels 人像

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

pexels 动植物

可能出现的问题

问题一: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

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

相关文章
|
人工智能 编解码
Photoshop的AI功能上新:截图→背景填充一气呵成,再也不怕没拍全了
Photoshop的AI功能上新:截图→背景填充一气呵成,再也不怕没拍全了
162 0
|
机器学习/深度学习 存储 人工智能
带你读《企业级云原生白皮书项目实战》——6.1.1背景
带你读《企业级云原生白皮书项目实战》——6.1.1背景
149 0
|
机器学习/深度学习 人工智能 编解码
Meta开发新虚拟背景处理AI,让元宇宙中人像不再虚糊
Meta开发新虚拟背景处理AI,让元宇宙中人像不再虚糊
205 0
|
机器学习/深度学习 人工智能 自然语言处理
50岁文科背景,日本副社长一年自学AI 1k小时,独自完成编程测试,项目被雀巢采纳
50岁文科背景,日本副社长一年自学AI 1k小时,独自完成编程测试,项目被雀巢采纳
250 0
|
机器学习/深度学习 人工智能 算法
一周AI最火论文 | 世界就是你的绿屏 ,任何背景一键抠像!
一周AI最火论文 | 世界就是你的绿屏 ,任何背景一键抠像!
174 0
|
19天前
|
人工智能 运维 物联网
云大使 X 函数计算 FC 专属活动上线!享返佣,一键打造 AI 应用
如今,AI 技术已经成为推动业务创新和增长的重要力量。但对于许多企业和开发者来说,如何高效、便捷地部署和管理 AI 应用仍然是一个挑战。阿里云函数计算 FC 以其免运维的特点,大大降低了 AI 应用部署的复杂性。用户无需担心底层资源的管理和运维问题,可以专注于应用的创新和开发,并且用户可以通过一键部署功能,迅速将 AI 大模型部署到云端,实现快速上线和迭代。函数计算目前推出了多种规格的云资源优惠套餐,用户可以根据实际需求灵活选择。
|
15天前
|
人工智能 算法 前端开发
OmAgent:轻松构建在终端设备上运行的 AI 应用,赋能手机、穿戴设备、摄像头等多种设备
OmAgent 是 Om AI 与浙江大学联合开源的多模态语言代理框架,支持多设备连接、高效模型集成,助力开发者快速构建复杂的多模态代理应用。
151 72
OmAgent:轻松构建在终端设备上运行的 AI 应用,赋能手机、穿戴设备、摄像头等多种设备
|
1天前
|
人工智能 自然语言处理 搜索推荐
【上篇】-分两篇步骤介绍-如何用topview生成和自定义数字人-关于AI的使用和应用-如何生成数字人-优雅草卓伊凡-如何生成AI数字人
【上篇】-分两篇步骤介绍-如何用topview生成和自定义数字人-关于AI的使用和应用-如何生成数字人-优雅草卓伊凡-如何生成AI数字人
46 17
【上篇】-分两篇步骤介绍-如何用topview生成和自定义数字人-关于AI的使用和应用-如何生成数字人-优雅草卓伊凡-如何生成AI数字人
|
9天前
|
人工智能 自然语言处理 JavaScript
微软开源课程!21节课程教你开发生成式 AI 应用所需了解的一切
微软推出的生成式 AI 入门课程,涵盖 21 节课程,帮助开发者快速掌握生成式 AI 应用开发,支持 Python 和 TypeScript 代码示例。
163 14

热门文章

最新文章