如何获取旋转的边界框的坐标

时间:2016-09-26 14:10:42

标签: python opencv

使用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()

0 个答案:

没有答案