python数据增强代码_深度学习中的数据增强（平移和旋转）python代码实现

import os

import numpy as np

import cv2

Angles = [10, 20, 100]

Trans_Select_Imgs = 10

def read_imgs(imgs_path):

imgs_name = os.listdir(imgs_path)

imgs = []

for img_name in imgs_name:

img_path = os.path.join(imgs_path, img_name)

img = cv2.imread(img_path)

imgs.append(img)

return imgs

# 旋转

def rotate_img(img, angle):

(height, width) = img.shape[:2]

center = (height//2, width//2)

matrix = cv2.getRotationMatrix2D(center, angle, 1)

# 旋转图像

rotate_img = cv2.warpAffine(img, matrix, (width,height))

return rotate_img

def get_rotate_imgs(imgs, rotate_img_path):

for idx_img in range(len(imgs)):

for angle in Angles:

r_img = rotate_img(imgs[idx_img], angle)

rotate_img_name = str(idx_img) + str