首页 ›编程› Python › 查看内容

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

2022-4-24 09:06 3447 0

摘要: 非常简洁干净的一个背景移除工具，可以命令行方式运行、也可以web方式运行，也可以python代码方式运行，安装运行都非常简单。原理是基于图像分割和Matting算法，支持u2net的一些列模型。开源：https://github.com/d ...

关键词： output 运行 input rembg 方式 path 可以代码安装 python

非常简洁干净的一个背景移除工具，可以命令行方式运行、也可以web方式运行，也可以python代码方式运行，安装运行都非常简单。原理是基于图像分割和Matting算法，支持u2net的一些列模型。

开源：https://github.com/danielgatis/rembg

安装

# CPU版本
pip install rembg 
# 如果有GPU，也可以安装GPU版
pip install rembg[gpu]

运行

# 命令行方式
rembg i path/to/input.webp path/to/output.webp

# web方式
rembg s

# python代码

Python代码例子

from rembg import remove

input_path = 'input.webp'
output_path = 'output.webp'

with open(input_path, 'rb') as i:
    with open(output_path, 'wb') as o:
        input = i.read()
        output = remove(input)
        o.write(output)

本文出处： https://www.toutiao.com/article/7089747658608624162/

声明：文章版权归原作者所有部分文章转自互联网如有侵权请联系 [邮箱地址] 删除

路过

雷人

握手

鲜花

鸡蛋

收藏分享邀请

上一篇：Python自动化工具Pywinauto、Pyautogui

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

最新评论

相关分类