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 工具:包含命令行界面,无需编写代码即可快速去除背景。

相关

  • 项目
  • 项目
  • 项目
  • 项目
  • 项目