Ubuntu22.04DesktopでRembgのインストール記録

Rembgをインストールしたのでその作業ログを残しておきます。

Rembgは画像から背景を除去してくれる効果があります。
上手く行けば写真から人物を一発で切り出してくれる点が素晴らしいです。

pythonの仮想環境を作成

cd ~/git
python3 -m venv rembgvenv

環境名をrembgvenvにしました。

仮想環境の開始

source ~/git/rembgvenv/bin/activate

仮想環境の終了要確認

deactivate

必要なパッケージをインストール

python -m pip install --upgrade pip
python -m pip install -U --ignore-installed pip
python -m pip install -U torch torchvision torchaudio numpy --index-url https://download.pytorch.org/whl/cu115
python -m pip install -U rembg

pytrochが必要かは不明

実行スクリプト

#!/home/ユーザー名/git/rembgvenv/bin/python3
import os
import glob
from rembg import remove
from PIL import Image

input_dir = '/mnt/workspace/inputs'
output_dir = '/mnt/workspace/outputs'

for f in glob.glob((input_dir+'/*.*'), recursive=False):
    basename = os.path.splitext(os.path.basename(f))[0]
    output_path = os.path.join(output_dir, (basename+'.png'))
    print(output_path)
    input = Image.open(f)
    output = remove(input)
    output.save(output_path)