美工福利！10行代码实现批量抠图

前言

继老张片修复，黑白照片上色之后，我又发现一个图像处理神器！

rembg模块

官方文档只用了一句话介绍了rembg模块:Rembg is a tool to remove images background.

然后简单粗暴地展示了这个模块能做的事 – 抠图（去除背景）：

官方还提供基于该模块实现的在线抠图工具：

如何正确使用rembg模块

如果你的Python版本小于3.7或者大于3.12，将无法使用该模块。

此外该模块又分了两个版本，一个是基于CPU实现计算，另一个则是基于GPU实现计算。

模块安装

CPU版

pip install rembg

GPU版

pip install rembg[gpu]

使用GPU版你还确认下硬件和系统，通过https://onnxruntime.ai查看，你的系统、硬件是否支持onnxruntime-gpu。

我是win10系统+N卡，所以是这样的选的：

然后我还要使用pip命令安装onnxruntime-gpu模块。

pip install onnxruntime-gpu

使用

基于命令行实现抠图

• 抠网络资源图

curl -s http://input.png | rembg i > output.png

• 抠本地图

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)

我又使用代码方式，完成了一次抠图：

抠图后：

最后

这个模块解决了一些简单的抠图需求，但我在实际测试中也翻了不少车。

一些物体之间颜色界限不是特别明显的图片，大概率都很会变成这样“缺胳膊少腿”的情况。

---

如果你想30天学会一门技能！那学习Python准没错

以下是我整理的一些提升程序员自身能力的资料，都已经整理并打包好了。

Python入门全套学习资料附带源码：

学习编程前准备

全套软件安装包

附带完整的安装包的安装视频教程资源（新手大礼包已备好）

整套零基础入门视频+课件笔记

👉实战案例👈

光学理论是没用的，要学会跟着一起敲，要动手实操，才能将自己的所学运用到实际当中去，这时候可以搞点实战案例来学习。

👉100道Python练习题👈

检查学习结果。

👉面试刷题👈

总结

好啦，这就是今天的内容，入门知识点资料免费发送的哈，想要的小伙伴儿不要错过，带你直

接弯道超车，少走一大波弯路，准备好了嘛？！我们要开始学习一项编程技术啦！

资料领取

这份完整版的Python全套学习资料已经上传网盘，朋友们如果需要可以点击下方微信卡片免费领取 ↓↓↓【保证100%免费】
或者
【点此链接】领取

好文推荐

了解python的前景：https://blog.csdn.net/weixin_49895216/article/details/127186741

了解python的兼职：https://blog.csdn.net/weixin_49895216/article/details/127124870