Pythonで画像の背景を削除する

Python を使用して画像の背景を簡単に削除する方法を学びます。 Python で画像の背景を簡単に削除する方法を学びます。画像処理の技術を習得して、見事なビジュアルと洗練されたデザインを実現します。

このために、オープン ソース ツール rembg を使用します。必要なコードは数行だけです。

rembgをインストールする

CPU で使用するには、次のコマンドを使用します。

pip install rembg

GPU がある場合は、次のものを使用できます。

pip install rembg[gpu]

正しい Pillow バージョンがインストールされていることを確認してください。 >= 9.3.0 である必要があります

import PIL
PIL.__version__

そうでない場合は、次のコマンドを使用してみてください (Google Colab を使用している場合は、この回避策を使用する必要がありました)。

pip install --ignore-installed Pillow==9.3.0

背景を削除する

これで、次のようにして背景を簡単に削除できます。

from rembg import remove

input_path = 'input.png'
output_path = 'output.png'

with open(input_path, 'rb') as i:
    with open(output_path, 'wb') as o:
        input = i.read()
        output = remove(input)
        o.write(output)

これにより、画像がロードされ、背景の削除が適用され、結果が新しい出力画像に書き込まれます。

CLIとして使用する

コマンド ラインから rembg を使用することもできます。

rembg i path/to/input.png path/to/output.png