Как определить угол изображения, чтобы проверить, является ли изображение вертикальным или горизонтальным?

Question

Я пытаюсь выровнять все изображения по вертикали / горизонтали. Мне нужно повернуть изображение, используя следующий код:

import cv2
import imutils
import numpy as np
def rotate_bound(image, angle):
    # grab the dimensions of the image and then determine the
    # center
    (h, w) = image.shape[:2]
    (cX, cY) = (w // 2, h // 2)
    # grab the rotation matrix (applying the negative of the
    # angle to rotate clockwise), then grab the sine and cosine
    # (i.e., the rotation components of the matrix)
    M = cv2.getRotationMatrix2D((cX, cY), -angle, 1.0)
    cos = np.abs(M[0, 0])
    sin = np.abs(M[0, 1])
    # compute the new bounding dimensions of the image
    nW = int((h * sin) + (w * cos))
    nH = int((h * cos) + (w * sin))
    # adjust the rotation matrix to take into account translation
    M[0, 2] += (nW / 2) - cX
    M[1, 2] += (nH / 2) - cY
    # perform the actual rotation and return the image
    return cv2.warpAffine(image, M, (nW, nH))
image = cv2.imread("10247.png")
# loop over the rotation angles again, this time ensuring
# no part of the image is cut off
dim=(600,400)
im=rotate_bound(image,30)
im = cv2.resize(im, dim, interpolation = cv2.INTER_AREA)
cv2.imshow("Rotated (Correct)", im)

cv2.waitKey(0)
cv2.destroyWindow("Rotated (Correct)")
for angle in np.arange(0, 360, 15):
    rotated = imutils.rotate_bound(image, angle)
    dim=(800,600)
    resized = cv2.resize(rotated, dim, interpolation = cv2.INTER_AREA)
    cv2.imshow("Rotated (Correct)", resized)
    cv2.waitKey(0)
    cv2.destroyWindow("Rotated (Correct)")

Я хочу остановить вращение, когда изображение вертикальное / горизонтальное. Как мне это остановить? Как определить текущий угол?

Answer 1

Вы можете ввести немного AI, используя MediaPipe . Оттуда вы получаете ориентиры для пальцев и поворачиваете, например, указательный палец вверх.

Аналогичной операцией может быть преобразование изображения в черно-белое и использование морфологической операции эрозия для сокращения руки до простых линий.

Затем используйте Преобразование линии Хафа , чтобы получить пальцы, отсортировать их (слева направо), получить угол к вертикали указательного пальца и поверните.