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 工具:包含命令列介面,無需編寫程式碼即可快速去除背景。

相關

  • 專案
  • 專案
  • 專案
  • 專案
  • 專案