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功能上新:截图→背景填充一气呵成,再也不怕没拍全了
167 0
|
机器学习/深度学习 存储 人工智能
带你读《企业级云原生白皮书项目实战》——6.1.1背景
带你读《企业级云原生白皮书项目实战》——6.1.1背景
151 0
|
机器学习/深度学习 人工智能 编解码
Meta开发新虚拟背景处理AI,让元宇宙中人像不再虚糊
Meta开发新虚拟背景处理AI,让元宇宙中人像不再虚糊
207 0
|
机器学习/深度学习 人工智能 自然语言处理
50岁文科背景,日本副社长一年自学AI 1k小时,独自完成编程测试,项目被雀巢采纳
50岁文科背景,日本副社长一年自学AI 1k小时,独自完成编程测试,项目被雀巢采纳
252 0
|
机器学习/深度学习 人工智能 算法
一周AI最火论文 | 世界就是你的绿屏 ,任何背景一键抠像!
一周AI最火论文 | 世界就是你的绿屏 ,任何背景一键抠像!
176 0
|
1天前
|
数据采集 人工智能 安全
阿里云携手DeepSeek,AI应用落地五折起!
近年来,人工智能技术飞速发展,越来越多的企业希望借助AI的力量实现数字化转型,提升效率和竞争力。然而,AI应用的开发和落地并非易事,企业往往面临着技术门槛高、成本投入大、落地效果难以保障等挑战。
24 1
|
6天前
|
存储 人工智能 NoSQL
Airweave:快速集成应用数据打造AI知识库的开源平台,支持多源整合和自动同步数据
Airweave 是一个开源工具,能够将应用程序的数据同步到图数据库和向量数据库中,实现智能代理检索。它支持无代码集成、多租户支持和自动同步等功能。
66 14
|
7天前
|
人工智能 开发框架 数据可视化
Eino:字节跳动开源基于Golang的AI应用开发框架,组件化设计助力构建AI应用
Eino 是字节跳动开源的大模型应用开发框架,帮助开发者高效构建基于大模型的 AI 应用。支持组件化设计、流式处理和可视化开发工具。
134 27
|
8天前
|
人工智能 自然语言处理 搜索推荐
现在最火的AI是怎么应用到体育行业的
AI在体育行业的应用日益广泛,涵盖数据分析、伤病预防、观众体验、裁判辅助等多个领域。通过传感器和可穿戴设备,AI分析运动员表现,提供个性化训练建议;预测伤病风险,制定康复方案;优化比赛预测和博彩指数;提升观众的个性化内容推荐和沉浸式观赛体验;辅助裁判判罚,提高准确性;发掘青训人才,优化训练计划;智能管理场馆运营和票务;自动生成媒体内容,提供实时翻译;支持电竞分析和虚拟体育赛事;并为运动员提供个性化营养和健康管理方案。未来,随着技术进步,AI的应用将更加深入和多样化。

热门文章

最新文章