withoutbg/withoutbg-python
Python SDK for local and cloud background removal (pip install withoutbg)
解决的问题
它提供了一种在 Python 中去除图像背景的简单方法,允许用户选择运行免费、私密的本地模型,或者使用高质量的云端 API 以获得头发和皮毛等复杂细节的更好效果。
工作原理
该库为两种不同的执行路径提供统一的 API:
- 本地模式:使用从 Hugging Face 下载的开源权重 ONNX 模型(约 455 MB),在 CPU 上运行。它使用 DINOv3 骨干网络在单次传递中结合了深度、分割、抠图和精修。
- 云端模式:连接到 withoutBG 云端 API,无需本地硬件资源即可获得专业级的质量。
适用对象
- 开发者:正在构建需要自动背景去除的脚本、笔记本或后端。
- 注重隐私的用户:希望在自己的机器上进行离线、无限制处理的用户。
- 产品构建者:需要通过 API 为商业应用获取高精度抠图的用户。
亮点
- 双路径 API:只需一行代码即可在本地和云端处理之间切换。
- 对 CPU 友好:本地推理通过 ONNX 在 CPU 上运行,这意味着不需要 GPU。
- 批量处理:包含
remove_background_batch方法,通过一次加载模型来高效处理多张图像。 - PIL 集成:以 RGBA 模式返回 PIL
Image对象,方便图像合成或保存为 PNG/WebP。 - CLI 工具:包含命令行界面,无需编写代码即可快速去除背景。
相关
- 项目
- 项目
- 项目
- 项目
- 项目