不会ps,也可以一键抠图,修改图片背景颜色

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)实现效果如下(抠出后的人物图像是一个带

    推荐阅读