import os
import cv2
import numpy as np

def remove_black_borders(image):
    """
    改進(jìn)版的去除黑色邊框函數(shù)
    """
    # 檢查是否有alpha通道
    if image.shape[2] == 4:
        # 使用alpha通道作為掩碼
        gray = image[:,:,3]
        _, thresh = cv2.threshold(gray, 1, 255, cv2.THRESH_BINARY)
    else:
        # 沒(méi)有alpha通道，使用常規(guī)方法
        gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
        _, thresh = cv2.threshold(gray, 10, 255, cv2.THRESH_BINARY)  # 提高閾值
    
    # 找到所有非黑色像素的坐標(biāo)
    coords = np.column_stack(np.where(thresh > 0))
    
    if len(coords) == 0:
        return image
    
    # 獲取邊界坐標(biāo)
    y_min, y_max = coords[:, 0].min(), coords[:, 0].max()
    x_min, x_max = coords[:, 1].min(), coords[:, 1].max()
    
    # 裁剪圖片
    cropped = image[y_min:y_max+1, x_min:x_max+1]
    
    return cropped

def process_png_files(input_folder, output_folder):
    """
    處理文件夾中的所有PNG文件
    """
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)
    
    for filename in os.listdir(input_folder):
        if filename.lower().endswith('.png'):
            input_path = os.path.join(input_folder, filename)
            image = cv2.imread(input_path, cv2.IMREAD_UNCHANGED)  # 保留alpha通道
            
            if image is None:
                print(f"無(wú)法讀取圖片: {filename}")
                continue
            
            # 去除邊框
            processed_image = remove_black_borders(image)
            
            # 保存圖片
            output_path = os.path.join(output_folder, filename)
            cv2.imwrite(output_path, processed_image)
            print(f"已處理: {filename} (原始尺寸: {image.shape}, 處理后尺寸: {processed_image.shape})")


if __name__ == "__main__":
    # 輸入文件夾路徑（包含PNG圖片的文件夾）
    input_folder = "C:\\Users\\zxc8756\\Desktop\\input_images"  # 修改為你的輸入文件夾路徑
    
    # 輸出文件夾路徑（處理后的圖片將保存到這里）
    output_folder = "C:\\Users\\zxc8756\\Desktop\\output_images"  # 修改為你想要的輸出文件夾路徑
    
    # 處理圖片
    process_png_files(input_folder, output_folder)
    
    print("所有圖片處理完成！")

pip install opencv-python numpy

# 輸入文件夾路徑（包含PNG圖片的文件夾）
input_folder = "C:\\Users\\zxc8756\\Desktop\\input_images"  # 修改為你的輸入文件夾路徑

# 輸出文件夾路徑（處理后的圖片將保存到這里）
output_folder = "C:\\Users\\zxc8756\\Desktop\\output_images"  # 修改為你想要的輸出文件夾路徑