干翻全岛蛙蛙 发表于 2024-5-20 22:27:52

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))处理前:
https://img2024.cnblogs.com/blog/3375219/202405/3375219-20240520112035326-338738164.jpg
处理后:
https://img2024.cnblogs.com/blog/3375219/202405/3375219-20240520112045578-766283974.jpg

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
页: [1]
查看完整版本: python去除图片白边黑边