图像数据增强

图像数据增强

最近因为要用到的数据很少，所以想到用数据增强的方法，大家可以参考博客 link，增强的方法没变，主要是在后面图片生成后放到新的文件夹做了一些改进

遇到的第一个问题是，在其他文件下生成新的相同文件夹，并实现文件名不重复

for filename in os.listdir(imageDir):
len+=1
if os.path.exists(os.path.join(saveDir, filename)):
# 若该文件存在，则跳过，否则继续创建；
# break 是跳出整个循环，continu是跳出本次循环
continue
filename_Aug = os.makedirs(os.path.join(saveDir, filename)) # 在增强后的数据集文件中创建空的各类文件夹

下面贴出最终增强后的文件

for x,y in map(lambda *row: list(row),os.listdir(imageDir),os.listdir(saveDir)):
k=0
origin_filename, save_filename = x,y
ori_path = imageDir+origin_filename
#for i,j in zip(os.listdir(ori_path),os.listdir(save_path)): 错误教训，save_path里面还没有文件
for i in os.listdir(ori_path):
origin_filename_1 = i
print(origin_filename_1)
k = k + 1
saveName = origin_filename + str(k) + “.jpg”
saveImage = flip(os.path.join(imageDir, origin_filename), origin_filename_1)
print(type(saveImage))
saveImage.save(os.path.join(os.path.join(saveDir, save_filename), saveName))
print(origin_filename)

这里面有涉及到如何同时遍历两个list，主要有map，和zip的方法，大家可以去查一下，第一次写博客，虽简陋但还是迈出了第一步，希望自己可以坚持。加油！