还在用PS抠图?这个GitHub开源AI工具3秒自动去背景,头发丝都能抠干净!

哈喽!大家好呀!我是一直在研究和使用AI工具的小明~
今天要来给大家分享的是一个抠图工具!我也不卖什么关子了!它就是Github上获得了2.2W+Star的Rembg。
可能有些人已经用过了,也有些人可能没用过,今天我就来聊聊~这款工具有多好用!
照例~开始之前大家如果对我的小程序有兴趣的话!可以点击下方体验一下!
告别”抠图抠到手抽筋”的日子
做过图的朋友都知道,抠图是设计工作中最枯燥的环节。用魔棒工具?背景太杂选不干净。用钢笔工具?一根根发丝能抠到你怀疑人生。
网上那些在线抠图工具要么收费,要么有水印,要么一上传图片就提示”服务器繁忙”。
今天介绍的 rembg,就像是给图片配了一个”智能剪刀手”。
它是一个开源的 Python 库,基于深度学习模型(主要是 U²-Net),专门做一件事:自动识别并移除图片背景。
想象一下,你拍了一张产品照片,背景乱糟糟的。以前你需要打开 PS,小心翼翼地把产品”描”出来。
现在呢?只需要一行命令,rembg 就能像有经验的修图师一样,自动把主体抠出来,连半透明婚纱、毛茸茸的小猫、飘逸的头发都能处理得干净利落。
这个”智能剪刀手”有多厉害?
rembg 的核心能力可以用三个词概括:快、准、全。
“快” 体现在速度上。基于 ONNX Runtime 优化,普通 CPU 上处理一张图也就几秒钟,如果有 NVIDIA 显卡,速度还能再起飞。更厉害的是它支持批量处理——你可以把整个文件夹扔给它,去喝口咖啡回来,几百张图全处理好了。
“准” 体现在模型选择上。它内置了 15+ 种针对不同场景的”专项技能”:
-
• u2net:通用场景,什么图都能抠 -
• u2net_human_seg:专攻人像,适合证件照、写真修图 -
• isnet-anime:专门针对动漫角色,二次元创作者的福音 -
• sam:Meta 的”分割一切”模型,你可以用手指点哪里抠哪里
“全” 体现在使用方式上。不管你是喜欢敲命令行的程序员,还是用 Python 做自动化的数据分析师,或是想搭建在线服务的开发者,它都能满足——支持 CLI、Python API、HTTP 服务器,甚至能打包成 Docker 镜像。
三步上手,真的就这么简单
方式一:命令行(适合快速处理)
安装超简单,CPU 版本直接 pip:
pip install rembg[cpu,cli]
然后抠图只要一行:
# 单张图片rembg i input.jpg output.png# 批量处理整个文件夹rembg p input_folder/ output_folder/# 启动 HTTP 服务,相当于自建抠图 APIrembg s --host 0.0.0.0 --port 7000
方式二:Python 代码(适合集成到项目)
from rembg import removefrom PIL import Image# 打开图片input_img = Image.open('product.jpg')# 一键去背景!就是这么简单output_img = remove(input_img)# 保存为透明背景 PNGoutput_img.save('product_clean.png')
如果你要处理大量图片,建议复用 session,就像把剪刀磨快了一样,效率翻倍:
from rembg import new_session# 创建专用"剪刀"(选择人像模型)session = new_session("u2net_human_seg")# 批量处理for img_path in image_list:withopen(img_path, 'rb') as f: result = remove(f.read(), session=session)
为什么选它,而不是某图秀秀?
可能有朋友会问:现在这么多在线抠图工具,为什么要用 rembg?
首先,它是完全开源免费的。很多在线工具免费版有水印或分辨率限制,rembg 没有这些套路,你甚至可以部署在自己的服务器上,数据不用上传第三方,隐私有保障。
其次,它是”白盒”的。你可以看到它用了什么模型,可以换模型,可以调参数。比如电商场景需要保留阴影?可以用 Alpha Matting 功能精细化边缘。视频需要逐帧处理?它可以和 FFmpeg 配合,直接处理视频流。
最后,它是可编程的。想象一下,你可以写一个脚本,监控文件夹里的新图片,自动抠图后上传到电商平台。这种自动化工作流,是在线工具给不了的。
它值得放进你的工具箱
rembg 就像一个24小时待命的数字修图师,它不挑环境(支持 CPU/GPU/AMD),不挑场景(15+专用模型),还不收一分钱。对于电商运营需要批量处理商品图、自媒体作者需要制作封面、程序员需要图像预处理来说,都是实实在在提效的神器。
如果你也被抠图困扰过,不妨试试这个开源项目。毕竟,能把重复劳动交给机器,咱们才有时间做更有创意的事,对吧?
项目地址:https://github.com/danielgatis/rembg


–End –

夜雨聆风