withoutbg/withoutbg-python

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

What it solves

Python だけで画像の背景を簡単に除去でき、無料のプライベートローカルモデルまたは高品質なクラウドベース API のどちらかを選択できます。

How it works

このライブラリは統合 ONNX グラフ（WBGNet）を使用し、深度推定、セグメンテーション、マット、リファインを 1 回の推論で実行します。ユーザーは単一 API で 2 つのモードを選択できます：open_weights() でローカル実行（Hugging Face から約 495 MB のモデルをダウンロード）または api() で API キーを用いたクラウド処理。

Who it’s for

  • オフラインまたはプライベートな画像処理が必要な開発者。
  • 大量の画像を処理し、画像ごとのコストを回避したいユーザー。
  • 高品質（特に髪や毛皮）でインフラ管理が不要なマネージドクラウドサービスを好むプロダクト開発者。

Highlights

  • Dual-mode API：1 行のコードでローカルとクラウド処理を切り替え。
  • Local Execution：ONNX を使用し CPU だけで実行、GPU 不要。
  • Batch Processing：モデルを一度ロードすれば複数画像を効率的に処理可能。
  • CLI Tool：コマンドラインインターフェースを備え、素早い背景除去とバッチジョブが可能。
  • Flexible Output：RGBA モードの PIL Image を返し、PNG と WebP の透過をサポート。