1 获得图片中某一块旋转矩形区域
1.1 将原始图片旋转到水平矩形位置进行裁剪
M = cv2.getRotationMatrix2D((x_center, y_center), float(theta)*180/math.pi, 1)
cols,rows,_ = img.shape
dst = cv2.warpAffine(img, M, (rows,cols))
crop_img = dst[y0:y0 + int(hight),x0:x0 + int(width)]
1.2 采用透视变换
可以用来变换任意四边形到一个长方形
height_left = int(np.sqrt(np.power(x1-x4,2) + np.power(y1-y4,2)))
height_right = int(np.sqrt(np.power(x3-x2,2) + np.power(y3-y2,2)))
width_top = int(np.sqrt(np.power(x1-x2,2)+np.power(y1-y2,2)))
width_bottom = int(np.sqrt(np.power(x3-x4,2)+np.power(y3-y4,2)))
max_height = max(height_left, height_right)
max_width = max(width_top, width_bottom)
orig_coord = np.float32([[x1,y1], [x2,y2], [x3,y3], [x4,y4]])
res_coord = np.float32([[0, 0], [max_width, 0],\
[max_width, max_height], [0, max_height]])
bound = cv2.getPerspectiveTransform(orig_coord, res_coord)
box = cv2.warpPerspective(img, bound,(max_width, max_height))
cv2.imwrite(os.path.join(cropDir, boxName), box)