数据集扩增--水平和垂直翻转

这几天在做一个图像分割的比赛，涉及到一点数据增强的内容，为了使数据集的容量变得大一点，所以使原始数据集进行水平和垂直翻转。代码如下：

水平翻转：

'''
Python批量实现图像水平、垂直翻转
函数功能：扩大数据量
'''
import PIL.Image as img
import os
path_old = r"./data/masks"
path_new = r"./data/mask_1"
filelist = os.listdir(path_old)
total_num = len(filelist)

for subdir in filelist:
    sub_dir = path_old + '/' + subdir
    im = img.open(sub_dir)
    ng = im.transpose(img.FLIP_LEFT_RIGHT) #左右对换。
    total_num+=1
    ng.save(path_new + '/' + str(total_num)+'.png')
    print(f'第{total_num}张图片已做好')
print('done')

垂直翻转：

'''
Python批量实现图像水平、垂直翻转
函数功能：扩大数据量
'''
import PIL.Image as img
import os

path_old = r"./data/imgs"
path_new = r"./data/imgs_2"

filelist = os.listdir(path_old)
total_num = 2*len(filelist)

for subdir in filelist:
    sub_dir = path_old + '/' + subdir
    im = img.open(sub_dir)
    ng = im.transpose(img.FLIP_TOP_BOTTOM)  # 上下对换。
    total_num+=1
    ng.save(path_new + '/' + str(total_num)+'.png')
    print(f'第{total_num}张图片已做好')

print('done')

使用这个代码的时候要注文件意路径问题和数据集中每一张图片命名问题