Python实现PNG图片黑色边框自动去除工具

import os
import cv2
import numpy as np

def remove_black_borders(image):
    """
    改进版的去除黑色边框函数
    """
    # 检查是否有alpha通道
    if image.shape[2] == 4:
        # 使用alpha通道作为掩码
        gray = image[:,:,3]
        _, thresh = cv2.threshold(gray, 1, 255, cv2.THRESH_BINARY)
    else:
        # 没有alpha通道，使用常规方法
        gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
        _, thresh = cv2.threshold(gray, 10, 255, cv2.THRESH_BINARY)  # 提高阈值
    
    # 找到所有非黑色像素的坐标
    coords = np.column_stack(np.where(thresh > 0))
    
    if len(coords) == 0:
        return image
    
    # 获取边界坐标
    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"无法读取图片: {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"  # 修改为你想要的输出文件夹路径