【Python】画像100枚の背景削除も一瞬！？最強AIライブラリ「rembg」で「全自動・透過ツール」を作ってみた

#! python3.11
import os
import sys
import time

# --- 設定エリア (自分の環境に合わせて書き換えてね！) ---
INPUT_DIR = r"C:\Users\AIGURUI\pythongame\ASSET\画像\キャラ"
VALID_EXTENSIONS = (".jpg", ".jpeg", ".png", ".bmp")

def process_images():
    # ★ポイント：重い処理の前にまずメッセージを出す
    print("="*60)
    print(" 🚀 AI背景削除ツール（フォルダ指定・全自動モード）")
    print("="*60)

    # 1. 保存先のフォルダ名を入力
    folder_name = input("保存先のフォルダ名を入力してください（例: Slime）: ").strip()
    
    if not folder_name:
        folder_name = "AI_Processed"

    print("\n" + "-"*60)
    print(" ⏳ AIライブラリを読み込んでいます...")
    print("    ※ 初回はAIモデル(約170MB)のダウンロードが入るため、")
    print("       数分間ここから動かなくなる場合があります。")
    print("       エラーが出ない限り、そのままお待ちください。")
    print("-"*60)

    # ★ポイント：ここで初めて重いライブラリを読み込む
    try:
        from rembg import remove
        print("\n ✅ 読み込み完了！ 処理を開始します。\n")
    except ImportError:
        print("\n[エラー] rembgが見つかりません。")
        print("インストールコマンド: py -3.11 -m pip install rembg")
        input("Enterキーを押して終了...")
        return

    # 保存先作成
    save_dir = os.path.join(INPUT_DIR, folder_name)
    if not os.path.exists(save_dir):
        os.makedirs(save_dir)

    # 画像リストアップ
    if not os.path.exists(INPUT_DIR):
        print(f"× 入力フォルダが見つかりません: {INPUT_DIR}")
        return

    files = sorted(os.listdir(INPUT_DIR))
    image_files = [f for f in files if f.lower().endswith(VALID_EXTENSIONS) and os.path.isfile(os.path.join(INPUT_DIR, f))]

    if not image_files:
        print("× 画像が見つかりませんでした。")
        return

    print(f"📸 {len(image_files)} 枚の画像を処理します...")

    # --- 一括処理 ---
    for i, file_name in enumerate(image_files):
        input_path = os.path.join(INPUT_DIR, file_name)
        
        # 保存名は「元のファイル名.png」
        name_without_ext = os.path.splitext(file_name)[0]
        save_name = f"{name_without_ext}.png"
        output_path = os.path.join(save_dir, save_name)

        print(f"[{i+1}/{len(image_files)}] {file_name} -> {save_name} ...", end="", flush=True)

        try:
            with open(input_path, "rb") as f:
                input_data = f.read()

            # ここでAIが魔法をかける！
            output_data = remove(input_data)

            with open(output_path, "wb") as f:
                f.write(output_data)
            
            print(" 完了！✨")

        except Exception as e:
            print(f" エラー: {e}")

    print("=" * 60)
    print("🎉 すべて完了しました！")
    print(f"保存場所: {save_dir}")
    input("Enterキーを押して終了...")

if __name__ == "__main__":
    process_images()