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