# -*- 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))
pytorch数据增强
最新推荐文章于 2024-08-20 20:51:20 发布