【機械学習】データ水増しプログラム

from PIL import Image, ImageOps
import glob, os
import cv2

in_path = "/Users/SuperMan/Desktop/test/test_before" # 入力ファイルを指定する
out_path = "/Users/SuperMan/Desktop/test/test_after" # 出力ファイルを指定する
files = glob.glob("{0}/*.jpg".format(in_path)) #jpgだけの写真ををとってくる

i = 1
for name in files:
    print(name)

    img = Image.open("{0}".format(name)) # 画像読み込み
    
    zero_i = "{0:03d}".format(i) # 3桁の0埋め連番作成
    new_name =  zero_i
    
    img_fl_tb = img.transpose(Image.FLIP_TOP_BOTTOM) # 上下反転
    img_fl_lr = img.transpose(Image.FLIP_LEFT_RIGHT) # 左右反転

    new_name_tb = new_name + "tb" + ".jpg"
    new_name_lr = new_name + "lr" + ".jpg"
    new_name    = new_name + ".jpg"
    
    print(new_name)
    
    img.save(out_path + "/" + "{0}".format(new_name))
    img_fl_tb.save(out_path + "/" + "{0}".format(new_name_tb)) # 画像保存
    img_fl_lr.save(out_path + "/" + "{0}".format(new_name_lr)) # 画像保存
    
    i = i + 1