Python ライブラリ Rembg で画像の背景透過を一括でおこなう

$ python --version
Python 3.10.6

pip install rembg

from rembg import remove
from PIL import Image
import os
import glob


print('start...')

# 変換前の画像を格納したフォルダ名
ORIGINAL_IMG_DIR = 'original/'
# 変換後の画像を格納したフォルダ名
REMOVED_IMG_DIR = 'removed/'

# 格納先フォルダがなければ作成する
os.makedirs(REMOVED_IMG_DIR, exist_ok=True)

# 画像を取得する
files = glob.glob(ORIGINAL_IMG_DIR + '*.jpg')

for file in files:
    # 拡張子を png に変更する
    file_name = os.path.splitext(os.path.basename(file))[0] + '.png'

    input = Image.open(file)

    # 画像の背景を透過させて保存する
    output = remove(input)
    output.save(REMOVED_IMG_DIR + file_name)

print('finish!')

pyhthon rembg-app.py