You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
17 lines
706 B
17 lines
706 B
2 years ago
|
from cv2 import getRotationMatrix2D,warpAffine
|
||
|
|
from math import fabs,sin,radians,cos
|
||
|
|
|
||
|
|
def ratate(img,degree=0):
|
||
|
|
|
||
|
|
height, width = img.shape[:2]
|
||
|
|
# 旋转后的尺寸
|
||
|
|
heightNew = int(width * fabs(sin(radians(degree))) + height * fabs(cos(radians(degree))))
|
||
|
|
widthNew = int(height * fabs(sin(radians(degree))) + width * fabs(cos(radians(degree))))
|
||
|
|
# 获得仿射变换矩阵
|
||
|
|
matRotation = getRotationMatrix2D((width / 2, height / 2), degree, 1)
|
||
|
|
matRotation[0, 2] += (widthNew - width) / 2
|
||
|
|
matRotation[1, 2] += (heightNew - height) / 2
|
||
|
|
# 进行仿射变换
|
||
|
|
imgRotation = warpAffine(img, matRotation, (widthNew, heightNew), borderValue=(68, 68, 68))
|
||
|
|
return imgRotation
|