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 文件夹即可。

相关文章
|
11月前
|
人工智能 编解码
Photoshop的AI功能上新:截图→背景填充一气呵成,再也不怕没拍全了
Photoshop的AI功能上新:截图→背景填充一气呵成,再也不怕没拍全了
123 0
|
机器学习/深度学习 存储 人工智能
带你读《企业级云原生白皮书项目实战》——6.1.1背景
带你读《企业级云原生白皮书项目实战》——6.1.1背景
118 0
|
机器学习/深度学习 人工智能 编解码
Meta开发新虚拟背景处理AI,让元宇宙中人像不再虚糊
Meta开发新虚拟背景处理AI,让元宇宙中人像不再虚糊
150 0
|
机器学习/深度学习 人工智能 自然语言处理
50岁文科背景,日本副社长一年自学AI 1k小时,独自完成编程测试,项目被雀巢采纳
50岁文科背景,日本副社长一年自学AI 1k小时,独自完成编程测试,项目被雀巢采纳
221 0
|
机器学习/深度学习 人工智能 算法
一周AI最火论文 | 世界就是你的绿屏 ,任何背景一键抠像!
一周AI最火论文 | 世界就是你的绿屏 ,任何背景一键抠像!
157 0
|
7天前
|
机器学习/深度学习 人工智能 自动驾驶
「AIGC」Agent AI智能体的未来:技术、伦理与经济的交汇点
Agent AI智能体融合机器学习与深度学习,推动社会效率与创新,但也引发伦理、法律及就业挑战。技术上,它们能自我优化、积累知识,如自动驾驶汽车通过学习改善驾驶。伦理上,需建立AI准则,确保透明度和责任归属,如医疗AI遵循道德原则。经济上,AI改变就业市场结构,创造新职业,如AI顾问,同时要求教育体系更新。未来,平衡技术进步与社会影响至关重要。
29 0
|
7天前
|
存储 机器学习/深度学习 人工智能
AI Agent技术的最新进展与改变世界的典型项目巡礼
【7月更文挑战第3天】 AI Agent技术的最新进展与改变世界的典型项目巡礼
 AI Agent技术的最新进展与改变世界的典型项目巡礼
|
1天前
|
机器学习/深度学习 人工智能 自然语言处理
智能引领未来:精准选择与高效利用AI技术的实战指南
【7月更文第14天】在当今科技飞速发展的时代,人工智能(AI)已不再局限于科幻电影的想象,而是成为了推动各行各业革新升级的核心驱动力。从自动驾驶汽车到个性化推荐系统,从医疗诊断辅助到智能制造,AI正深刻改变着我们的生活和工作方式。然而,面对琳琅满目的AI技术和工具,如何精准选择并高效利用AI,成为企业和开发者亟需解决的关键问题。本文将为您揭示这一过程中的核心策略与实践技巧,并通过一个简单的代码示例,让您直观感受AI技术的魅力。
23 3
|
2天前
|
机器学习/深度学习 人工智能 算法
未来AI技术在医疗领域的应用与挑战
随着人工智能技术的迅猛发展,其在医疗领域的应用前景广阔,但同时也面临着诸多挑战。本文探讨了AI技术在医疗中的多个关键应用,分析了其带来的潜在益处和可能的风险,以及如何在未来克服技术和伦理上的挑战。 【7月更文挑战第13天】

热门文章

最新文章