图片进行翻转并重命名保存（Python代码）

图片进行翻转并重命名保存代码

import os

from torchvision import transforms
from PIL import Image


def data_augmentation(data_augmentation_path):
    images_name = os.listdir(data_augmentation_path)
    for image_name in images_name:
        image_new_name = '0' + image_name
        image_path = os.path.join(data_augmentation_path, image_name)
        img = Image.open('%s' % image_path)
        transform = transforms.Compose([
           transforms.RandomHorizontalFlip(1)
        ])
        image = transform(img)
        image.save("%s" % (os.path.join(data_augmentation_path, image_new_name)))
        print(os.path.join(data_augmentation_path, image_name) + '\n  to  ' +
              os.path.join(data_augmentation_path, image_new_name) + '\n' +
              '------------------------------------------------------------------------------')

if __name__ == '__main__':
    Data_Augmentation_Path = r'images_path'
    data_augmentation(Data_Augmentation_Path)