图片进行翻转并重命名保存代码
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)