rembg 是使用 Python 实现的用于移除背景图片的工具,是一个多达7.5K star 的项目,要求 Python 3.8 或更高版本(目前暂时不支持python3.11),支持批量操作,使用方式比较灵活,可以直接使用命令行、作为服务运行、还可以作为库调用 。
当然配合PIL库还可以实现图片背景修改
原理
原理是基于图像分割和Matting算法,支持u2net的一些列模型
安装方式(该工具分为CPU 和GPU两个版本,可以根据自己的实际情况进行安装)
【不会ps,也可以一键抠图,修改图片背景颜色】CPU版本pip install rembgGPU版本pip install rembg[gpu]
安装所需要的依赖库比较多,由于网络问题,建议大家使用国内的pip 镜像
pip install rembg -i ***/simple/
训练模型下载安装,由于网络的原因我将训练模型已经放入网盘,供大家下载
链接:***/s/11aXuMBJaJUgkqlhyVEo8lA 提取码:9ut1
将下载好的训练模型移动到到用户根目录的.u2net文件夹下
windows
C:\Users\(用户路径)\.u2net文件夹下
linux (比如root用户)
/root/.u2net
使用方式(命令行使用)
安装成功后,可以在命令行中调动rembg 。如果只对单个图片进行处理
rembg i path/to/input.png path/to/output.png
对多个图片文件处理(批处理),
rembg p path/to/input path/to/output
在Python中使用
#!/bin/env python# -*- coding: utf-8 -*-#本测试程序仅仅实现了去除背景的功能from rembg import removefrom PIL import Imageinput_path = '123.jpg'output_path = 'output1.png'input = Image.open(input_path)output = remove(input)output.save(output_path)
实现效果如下(抠出后的人物图像是一个带
推荐阅读
- 一键卸载安卓全机型预装应用
- 王者荣耀扫码登录会不会被盗号 王者荣耀扫码登录后怎么删除设备
- 吃什么零食不会发胖
- 最伤感的说说
- 这5类人吃大蒜不会防癌会伤身
- 鸡蛋煮几分钟能熟 这样煮口感才不会老
- ins的一键翻译怎么设置
- 忘年恋的婚姻不管什么时候都不会被看好
- 仓鼠兄弟姐妹之间会不会打架
- 春节网商贷会不会停止放款 关闭淘宝店铺以后网商贷还有么