'''
Author: price32768 bnjicd@163.com
Date: 2024-10-16 08:32:27
LastEditors: price32768 bnjicd@163.com
LastEditTime: 2024-10-16 08:45:17
FilePath: \script\cleanBackground.py
Description: 这是默认设置,请设置`customMade`, 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
'''
import cv2
import numpy as np

def change_white_background_to_transparent(image_path, output_path):
    # 读取图片
    image = cv2.imread(image_path, cv2.IMREAD_UNCHANGED)

    # 将图片转换为 RGBA 格式（如果原图是 RGB 或 BGR，需要添加 alpha 通道）
    if image.shape[2] == 3:  # 如果是 RGB/BGR 图片
        alpha_channel = np.ones((image.shape[0], image.shape[1]), dtype=image.dtype) * 255
        image = cv2.merge((image, alpha_channel))

    # 定义白色的范围
    lower_white = np.array([200, 200, 200, 255])
    upper_white = np.array([255, 255, 255, 255])

    # 创建掩码，找到白色区域
    mask = cv2.inRange(image, lower_white, upper_white)

    # 使白色区域变为透明
    image[mask != 0] = [0, 0, 0, 0]  # 将白色区域的 RGBA 值设为 0

    # 保存结果
    cv2.imwrite(output_path, image)

# 使用示例
for filename in ["chan_0","chan_1","qiaojia_0","qiaojia_1","piaochong_0","piaochong_1"]:
# filename = "chan_0"
    change_white_background_to_transparent(f"./img/{filename}.png", f"{filename}.png")