圖像旋轉后背景填白（python代碼）

from PIL import Image
import os
import cv2
import numpy as np

def rotation(root_path, img_name):
    pilim = Image.open(os.path.join(root_path, img_name))
    im2 = pilim.convert('RGBA')
    rot = im2.rotate(-18.0, expand=1) #旋轉角度
    fff = Image.new('RGBA', rot.size, (255,)*4)
    # 使用alpha層的rot作為掩碼創建一個復合圖像
    out = Image.composite(rot, fff, rot)
    out.convert(pilim.mode)
    return out

imageDir="C:/Users/16003/Desktop/data"
saveDir="C:/Users/16003/Desktop/data1" 
i=0
for name in os.listdir(imageDir):
    i=i+1
    saveName="b"+str(i)+".png"
    saveImage=rotation(imageDir,name)
    saveImage.save(os.path.join(saveDir,saveName))

對於圖像旋轉之后，背景色會自動填充0，也就是黑色。

使用以上的方法，可以將背景色自動填充為255，簡單實用。

當然其中內容比較重要的包括圖像的旋轉操作。