python opencv批量旋转图像

import os
import random

import cv2
import glob
import pathlib
import numpy as np

# 批量旋转图片

data_path = r'E:\datasets\gen_text_cls\train'

save_path = r'E:\datasets\gen_text_cls\train4'
if not os.path.exists(save_path):
    os.makedirs(save_path)

cnt = 0

#旋转文本图像
def rotate_text_images(img_file,rotate_type):  # np.rot90(img, -1) 顺时针旋转90度   旋转程度：1,2,-1,-2
    img = cv2.imread(img_file)
    rotate_image = np.rot90(img, rotate_type)
    cv2.imshow("rotate", rotate_image)
    cv2.waitKey(3)
    return rotate_image

for img_path in glob.glob(data_path + '/*.jpg', recursive=True):
    r_img=rotate_text_images(img_path,-1)
    save_img_path = save_path + '/3_' + str(random.randint(100000, 999999))+str(cnt) + '.jpg'

    cnt = cnt + 1
    print(cnt)
    cv2.imwrite(save_img_path, r_img)