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