使用 HoughLine 变换检测角度和旋转图像

Question

我正在尝试旋转通过旋转清晰可见的图像。

我正在将 HoughLine 与 opencv 结合使用。

这是带有以下代码的图像 (working in google colab):

import numpy as np
import cv2
from scipy import ndimage
from google.colab.patches import cv2_imshow

image1 = cv2.imread('/content/rotate.png')
gray=cv2.cvtColor(image1,cv2.COLOR_BGR2GRAY)

edges = cv2.Canny(gray,50,150,apertureSize = 3)

canimg = cv2.Canny(gray, 50, 200)
lines= cv2.HoughLines(canimg, 1, np.pi/180.0, 250, np.array([]))
#lines= cv2.HoughLines(edges, 1, np.pi/180, 80, np.array([]))
for line in lines:
    rho, theta = line[0]
    a = np.cos(theta)
    b = np.sin(theta)
    x0 = a*rho
    y0 = b*rho
    x1 = int(x0 + 1000*(-b))
    y1 = int(y0 + 1000*(a))
    x2 = int(x0 - 1000*(-b))
    y2 = int(y0 - 1000*(a))

    cv2.line(image1,(x1,y1),(x2,y2),(0,0,255),2)
print(theta)
print(rho)
cv2_imshow(image1)
cv2_imshow(edges)

这是输出：

西塔：0.9773844

rho: 311.0

因此，当我尝试用这条线旋转此图像然后显示它时：

img_rotated = ndimage.rotate(image1, theta)
cv2_imshow(img_rotated)

这是输出：

此结果与框架水平应有的旋转不一致。有什么建议吗？我做错了什么？

Answer 1

在 ndimage.rotate 中以度为单位的角度。

img_rotated = ndimage.rotate(image1, 180*theta/3.1415926)

使用 HoughLine 变换检测角度和旋转图像

Detect angle and rotate image using HoughLine Transform

python

opencv

image

image-processing

hough-transform