深度学习中常用的数据增强方法，旋转，翻转等

最新推荐文章于 2024-07-21 07:50:37 发布

Free-self

最新推荐文章于 2024-07-21 07:50:37 发布

阅读量193

点赞数

文章标签：深度学习人工智能 python

本文链接：https://blog.csdn.net/m0_49491470/article/details/133748163

版权

数据增强方法，旋转，翻转

import os
from PIL import Image
import argparse
import shutil

parser = argparse.ArgumentParser(description="数据增强")
parser.add_argument("--input-path", type=str, default='增强前的地址', help="地址")
parser.add_argument("--out-path", type=str, default='增强后的保存地址', help="地址")

opt = parser.parse_args()

pathlist = os.listdir(opt.input_path)

if not os.path.exists(opt.out_path):
    os.mkdir(opt.out_path)

##尺度缩放
scale = [1]
##旋转角度
angle = [0]
flip = [0, 1, 2, 3, 4, 5, 6]
count = 0

for image_name in pathlist:
    image = Image.open(os.path.join(opt.input_path, image_name)).convert('RGB')

    for i in range(len(scale)):
        for j in range(len(angle)):
            for c in range(len(flip)):

                image = image.rotate(angle[j])
                image = image.resize((int(image.size[0] * scale[i]), int(image.size[1] * scale[i])), Image.BICUBIC)
                if flip[c] == 0:  ##水平翻转
                    image = image  ##不翻转
                elif flip[c] == 1:  ##水平翻转
                    image = image.transpose(Image.FLIP_LEFT_RIGHT)
                elif flip[c] == 2:  ###垂直翻转
                    image = image.transpose(Image.FLIP_TOP_BOTTOM)
                elif flip[c] == 3:  ###垂直翻转
                    image = image.transpose(Image.ROTATE_90)
                elif flip[c] == 4:  ###垂直翻转
                    image = image.transpose(Image.ROTATE_180)
                elif flip[c] == 5:  ###垂直翻转
                    image = image.transpose(Image.ROTATE_270)
                else:
                    image = image.transpose(Image.TRANSPOSE)

                count += 1
                aa = image_name.split('.')[0] + '{}_{}_{}.jpg'.format(i, j, c)
                image.save(os.path.join(opt.out_path, aa))
                image = Image.open(os.path.join(opt.input_path, image_name)).convert('RGB')
                # image.show()