用Python去除图片背景:​Rembg库

Python大数据分析

共 610字,需浏览 2分钟

 ·

2022-05-16 14:56

Python 的 Rembg 库可以去掉图片中的背景,效果如下


安装

CPU版

pip install rembg

GPU版

pip install rembg[gpu]


快速上手

命令行调用

安装成功后,可以在命令行中调动Rembg。如果只对单个图片进行处理

rembg i path/to/input.png path/to/output.png

对多个图片文件处理(批处理),

rembg p path/to/input path/to/output


在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)


来源:大邓和他的Python



加入知识星球【我们谈论数据科学】

500+小伙伴一起学习!







· 推荐阅读 ·

pandas中鲜为人知的隐藏排序技巧

在jupyter中实现实时协同是一种什么体验

新一代Python包管理工具来了




浏览 39
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报