<返回更多

一个删除图像背景的工具Rembg

2022-04-24    AI工具箱
加入收藏

非常简洁干净的一个背景移除工具,可以命令行方式运行、也可以web方式运行,也可以Python/ target=_blank class=infotextkey>Python代码方式运行,安装运行都非常简单。原理是基于图像分割和Matting算法,支持u.NET的一些列模型。

开源:https://github.com/danielgatis/rembg
  • 安装
# CPU版本
pip install rembg 
# 如果有GPU,也可以安装GPU版
pip install rembg[gpu]
  • 运行
# 命令行方式
rembg i path/to/input.png path/to/output.png

# web方式
rembg s

# python代码
  • Python代码例子
from rembg import remove

input_path = 'input.png'
output_path = 'output.png'

with open(input_path, 'rb') as i:
    with open(output_path, 'wb') as o:
        input = i.read()
        output = remove(input)
        o.write(output)
一个删除图像背景的工具Rembg

 

声明:本站部分内容来自互联网,如有版权侵犯或其他问题请与我们联系,我们将立即删除或处理。
▍相关推荐
更多资讯 >>>