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

机器学习与生成对抗网络

共 845字,需浏览 2分钟

 ·

2022-05-23 14:17

来源:大邓和他的Python 编辑 Python大数据分析

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)






猜您喜欢:

 戳我,查看GAN的系列专辑~!
一顿午饭外卖,成为CV视觉前沿弄潮儿!
CVPR 2022 | 25+方向、最新50篇GAN论文
 ICCV 2021 | 35个主题GAN论文汇总
超110篇!CVPR 2021最全GAN论文梳理
超100篇!CVPR 2020最全GAN论文梳理


拆解组新的GAN:解耦表征MixNMatch

StarGAN第2版:多域多样性图像生成


附下载 | 《可解释的机器学习》中文版

附下载 |《TensorFlow 2.0 深度学习算法实战》

附下载 |《计算机视觉中的数学方法》分享


《基于深度学习的表面缺陷检测方法综述》

《零样本图像分类综述: 十年进展》

《基于深度神经网络的少样本学习综述》


      

浏览 71
点赞
评论
收藏
分享

手机扫一扫分享

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

手机扫一扫分享

分享
举报