withoutbg/withoutbg-python

Python SDK for local and cloud background removal (pip install withoutbg)

解决的问题

它提供了一种使用 Python 移除图像背景的简单方法,并提供了灵活的选择:既可以在本地运行模型以兼顾隐私与成本,也可以使用高质量的云端 API 以获得处理头发和毛发等复杂细节的更高精度。

运作方式

该库允许用户在两种模式之间切换:

  • 本地模式 (Local Mode):使用基于 DINOv3、Depth Anything V2 和 BiRefNet 的 ONNX 套件,可在 CPU 或 GPU 上运行。路由器会决定使用专门的抠图模型来处理精细细节,还是使用分割模型来处理硬边缘物体。
  • 云端模式 (Cloud Mode):连接至 withoutBG API,无需本地基础设施即可获得专业级的效果。

适用对象

需要自动化图像背景移除功能的开发者,适用于构建脚本、笔记本、后端服务或批处理作业。

特色亮点

  • 混合方法:单一 API 同时支持本地开源权重推理与云端处理。
  • 批处理:通过一次加载模型来优化多张图像的处理效率。
  • 隐私优先:本地模式在初始权重下载后可完全离线运行。
  • CLI 工具:包含命令行界面,无需编写代码即可快速移除背景。
  • PIL 集成:以 RGBA 模式的 PIL Image 对象返回图像,方便进行合成。

相关

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