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))
处理前:
处理后: