pytorch数据增强

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

import PIL.Image as Image
import os
from torchvision import transforms as transforms
import random
import time

t1 = time.time()
inputpath = '/media/user/Elements/增强数据/imgs'
global number
# 增强的图片数目
number = 1000
filelists = os.listdir(inputpath)
for root, dirs, files in os.walk(inputpath):
    filelen = len(files)
    print("图片数量==================", len(files))
    for file in files:
        file_name = os.path.join(root, file)
        im = Image.open(file_name)
        for i in range(1, int(number / filelen)):
            t3 = time.time()
            time_id = str(t3).replace(".", "")
            im1 = transforms.RandomHorizontalFlip(p=random.random())(im)  # p表示概率
            im2 = transforms.RandomVerticalFlip(p=random.random())(im1)
            #分别代表亮度,对比度,饱和度,其中hue最大不超过0.5
            im3 = transforms.ColorJitter(brightness=0.2,contrast=0.2,hue=0.2)(im2)
            #im3 = transforms.ColorJitter(hue=0.5)(im2)
            new_im = transforms.RandomRotation(random.randint(0, 5))(im3)  # 随机旋转0-5度

            new_im.save(os.path.join(root, time_id + ".jpg"))
t2 = time.time()
print("所需时间======================", (t2 - t1))