【チャットGPTとPython】で楽々画像処理！写真の背景削除を学ぼう

from rembg import remove
import numpy as np
import io
from PIL import Image

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

def remove_background(input_path, output_path):
    with open(input_path, 'rb') as img_file:
        img = img_file.read()

    # `remove`関数は、バイト列の入力画像から背景を削除します
    img_no_bg_bytes = remove(img)

    # バイト列の画像をPIL.Imageオブジェクトに変換します
    img_no_bg = Image.open(io.BytesIO(img_no_bg_bytes))

    # 結果をPNG形式で保存します
    img_no_bg.save(output_path, "PNG")

remove_background(input_path, output_path)

output_path = '/Users/*****/Documents/python/python_app/cut_app/output.png'

from rembg import remove
import numpy as np
import io
from PIL import Image
import os

input_dir = '/path/to/input/directory/'
output_dir = '/path/to/output/directory/'

def remove_background(input_dir, output_dir):
    # 入力ディレクトリ内のすべてのファイルを処理します
    for filename in os.listdir(input_dir):
        if filename.endswith('.png') or filename.endswith('.jpg'): # PNGとJPGファイルのみを処理します
            with open(os.path.join(input_dir, filename), 'rb') as img_file:
                img = img_file.read()

            img_no_bg_bytes = remove(img)
            img_no_bg = Image.open(io.BytesIO(img_no_bg_bytes))

            # 元のファイル名から拡張子を分割します
            base_filename, file_extension = os.path.splitext(filename)

            # 新しいファイル名を作成します
            new_filename = f"{base_filename}_透過{file_extension}"

            # 出力ディレクトリに新しいファイル名で保存します
            img_no_bg.save(os.path.join(output_dir, new_filename), "PNG")

remove_background(input_dir, output_dir)