Pythonで人物画像などの背景を簡単に抜く:rembgライブラリ

AI

人物画像の背景を抜くには一昔前までは、Photoshopなどで職人技が必要だったのですが、AIを使うと簡単に同様のことができます。
今回はPythonのrembgライブラリを使用してみます。

サンプルコード GoogleColabファイル

上が元画像で下が加工後の画像

スポンサーリンク

参考コード

rembgのinstall

pip install rembg

背景を抜くためのコード
my_img.jpgが元画像
output_pathが出来上がり画像(完成画像の拡張子を.pngにしておくのがポイント
PILのinstallが必要です。

from rembg import remove
from PIL import Image

input_path = 'my_img.jpg'
output_path = 'new_img.png'
inp  = Image.open(input_path)
output = remove(inp)
output.save(output_path)
タイトルとURLをコピーしました