python去除圖片白邊黑邊

粉红毛猫猫發表於2024-05-20
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))

處理前：

處理後：

相關文章