python去除图片白边黑边

主要用于去除图片的白边和黑边，比如在截图表情包的时候，通过小米的传送门保存图片的时候，图片往往会有黑边和白边，此时使用此脚本二次处理

import os
from PIL import Image, ImageChops
def trim_white_border(image):
    bg = Image.new(image.mode, image.size, image.getpixel((0, 0)))  # 创建一张和原图相同大小像素量的单色图片，图片颜色取决于原图(0,0)坐标的像素点颜色
    # bg.show()
    diff = ImageChops.difference(image, bg)  # 将原图和底色图对比，反转色
    # diff.show()
    diff = ImageChops.add(diff, diff, 2.0, -100)  # 去除图片中的像素点，在此之后的图片已被去除白色像素
    # diff.show()
    bbox = diff.getbbox()  # 计算图像中非零区域的边界框，获取去除白边后的图形边界
    if bbox:
        return image.crop(bbox)  # 返回图像中的一个矩形区域。方框是一个 4 元组，定义了左、上、右、下像素坐标。
    else:
        print('trim white border for {} fail!'.format(file))
        return False

type_list = ['.jpg', '.png'] #处理同目录下的jpg和png图片
for file in os.listdir('.'):
    for file_type in type_list:
        if file.endswith(file_type):
            print('start trim white border for {}'.format(file))
            image = Image.open(file)
            trimmed_image = trim_white_border(image)
            trimmed_image.save(file.replace(file_type, '_trimmed'+file_type))
            print('trim white border for {} ok!'.format(file))

处理前：

处理后：