如何将所有图像旋转到相同的方向？

Question

我正在尝试执行图像分类任务并希望确保我的输入数据都具有相同的方向。 下面的代码没有将所有图像匹配到相同的方向，并且有些翻转错误。 如果有人能帮助我解决这个问题，我将不胜感激，谢谢 original image 1
original image 2

import cv2
import numpy as np
import matplotlib.pyplot as plt
def getSubImage(rect, image):
    center, size, theta = rect
    center, size = tuple(map(int, center)), tuple(map(int, size))
    M = cv2.getRotationMatrix2D( center, theta, 1)
    dst = cv2.warpAffine(image, M, src.shape[:2])
    out = cv2.getRectSubPix(dst, size, center)
    return out

image = cv2.imread('orginal1.png')
im_bw = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)
blur = cv2.GaussianBlur(im_bw, (5,5), 0)
im_bw = cv2.Canny(blur, 10, 90)
contours, hierarchy = cv2.findContours(im_bw, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
rect = cv2.minAreaRect(contours[0])

out = getSubImage(rect, image)
cv2.imwrite('rotedorginal1.jpg', out)
plt.imshow(out)
plt.show()

Answer 1

您只需要使用cv2.getRotationMatrix2D 构建将图像旋转直角的矩阵，并使用cv2.warpAffine 对矩阵进行运算：

(x, y), (w, h), angle = cv2.minAreaRect(contours[0])
result = cv2.warpAffine(image, cv2.getRotationMatrix2D((image.shape[1]//2, image.shape[0]//2), angle-90, 1), (image.shape))

上面的代码转换了这张图片：

进入这个：