python:画像処理tips その4 フォルダ内の画像の大きさを揃える

その３をループで回しただけ。

拡大縮小後の保存ファイル名は、リストから位置と拡張子を取得してformatdeぶち込んでる。

これで簡単にデータセット作れそう。

# -*- coding:utf-8 -*-

from PIL import Image
import os

images = os.listdir("path") #ディレクトリのパスをここに書く

for i in images:

    if i.find('.png') > 0 or i.find('.jpg') > 0 or i.find('.jpeg') > 0: #ここを増やせば他の拡張子もいける。
        
        img = Image.open(i)
        img = img.resize((50,50))
        img.save("resize{}.{}".format(images.index(i),i.split(".")[1]))
        #img.save(i) #上書き保存したい時はこっち
    
    else: continue