本文共 1565 字,大约阅读时间需要 5 分钟。
python随机生成图像旋转任意角度后的图像,需要得到一张图像旋转任意角度之后的结果,随机生成10000张图像,并将输出的图像命名为角度值。
# -*- coding: utf-8 -*-import cv2from math import *import numpy as npimport random# 旋转angle角度,缺失背景白色(255, 255, 255)填充def rotate_bound_white_bg(image, angle): # grab the dimensions of the image and then determine the # center (h, w) = image.shape[:2] (cX, cY) = (w // 2, h // 2) # grab the rotation matrix (applying the negative of the # angle to rotate clockwise), then grab the sine and cosine # (i.e., the rotation components of the matrix) # -angle位置参数为角度参数负值表示顺时针旋转; 1.0位置参数scale是调整尺寸比例(图像缩放参数),建议0.75 M = cv2.getRotationMatrix2D((cX, cY), -angle, 1.0) cos = np.abs(M[0, 0]) sin = np.abs(M[0, 1]) # compute the new bounding dimensions of the image nW = int((h * sin) + (w * cos)) nH = int((h * cos) + (w * sin)) # adjust the rotation matrix to take into account translation M[0, 2] += (nW / 2) - cX M[1, 2] += (nH / 2) - cY # perform the actual rotation and return the image # borderValue 缺失背景填充色彩,此处为白色,可自定义 return cv2.warpAffine(image, M, (nW, nH),borderValue=(255,255,255)) # borderValue 缺省,默认是黑色(0, 0 , 0) # return cv2.warpAffine(image, M, (nW, nH))img = cv2.imread("/Users/Cheney/Downloads/testrotation/1.jpg")l = [random.randint(0, 10000) for a in range(10000)]print(l)for i in l: imgRotation = rotate_bound_white_bg(img, i) # cv2.imshow("img",img) #cv2.imshow(str(i),imgRotation) print(i) cv2.imwrite("/Users/Cheney/Downloads/testrotation/" + str(i) + '.jpg', imgRotation) #cv2.waitKey(0)