使用python OpenCV我旋转已经有边界框的图像。我使用getRotationMatrix2D
函数执行此操作。如何使用它来计算边界框的新坐标?
这是我的源代码:
import cv2
import numpy as np
import matplotlib.pyplot as plt
from PIL import Image
def rotateImage(image, angle, bboxes):
center=tuple(np.array(image.shape[0:2])/2)
rot_mat = cv2.getRotationMatrix2D(center,angle,1.0)
return cv2.warpAffine(image, rot_mat, image.shape[0:2],flags=cv2.INTER_LINEAR)
img = cv2.imread('input.jpg')
img = img[:,:,::-1]
img_new = rotateImage(img, 5.0, [(0,0,50,50)])
plt.subplot(121),plt.imshow(img),plt.title('Input')
plt.subplot(122),plt.imshow(img_new),plt.title('Output')
plt.show()