使用移除图片背景(提取人像)的开源项目rembg

这个软件可以把图片中的背景删除,

原图片

 

去除背景后
 

项目官网:https://github.com/danielgatis/rembg

安装步骤:

1、安装Python环境。根据官网说明,Python必须是3.9版本

2、安装rembg,用CPU的话 

pip install rembg

用GPU的话

pip install rembg[gpu]

3、如果运行rembg提示命令不存在,则把rembg所在的路径加入PATH环境变量

4、rembg运行的时候,要从google文档网站下载模型文件,如果是在Wall内,会下载失败,会提示如下报错信息

requests.exceptions.ConnectTimeout: HTTPSConnectionPool(host='docs.google.com', port=443): Max retries exceeded with url: /uc?export=download&id=1ao1ovG1Qtx4b7EoskHXmi2E9rp5CHLcZ (Caused by ConnectTimeoutError(<urllib3.connection.HTTPSConnection object at 0x000002586D5D8D00>, 'Connection to docs.google.com timed out. (connect timeout=None)'))

 

这种情况下,我们就要把报错信息中的文件路径,比如https://docs.google.com/uc?export=download&id=1ao1ovG1Qtx4b7EoskHXmi2E9rp5CHLcZ 下载到C:\Users\(用户路径)\.u2net文件夹下

5、命令行执行如下命令,就可以完成背景去除从而提取人像

rembg 源文件名 -o 目标文件名

6、rembg提供了多种模型,模型的说明和下载地址在github页面中的“Models”这个区域

  把这些文件下载到C:\Users\(用户路径)\.u2net文件夹下,然后通过rembg的-m参数来指定使用的模型。如果运行的时候又提示“'Connection to docs.google.com timed out”,那么就继续把缺少的文件按照提示下载放到C:\Users\(用户路径)\.u2net文件夹下即可。

 

 

 

 

 

 

 
posted @   杨中科  阅读(3494)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· 阿里巴巴 QwQ-32B真的超越了 DeepSeek R-1吗?
· 【译】Visual Studio 中新的强大生产力特性
· 10年+ .NET Coder 心语 ── 封装的思维:从隐藏、稳定开始理解其本质意义
· 【设计模式】告别冗长if-else语句:使用策略模式优化代码结构
点击右上角即可分享
微信分享提示