Определите угол и поверните изображение с помощью HoughLine Transform - PullRequest
Новая
       8

Определите угол и поверните изображение с помощью HoughLine Transform

0 голосов
/ 05 мая 2020

Я пытаюсь повернуть изображение, которое хорошо видно при вращении.

Я использую HoughLine с opencv.

Вот изображение с кодом ниже ( работает в google colab ):

enter image description here

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)

Это результат:

theta: 0,9773844

rho: 311.0

enter image description here enter image description here

Итак, когда я пытаюсь повернуть это изображение с помощью этой строки, а затем показать это: 

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

Это результат:

enter image description here

Этот результат не согласуется с вращением, которое должно быть для рамка должна быть горизонтальной. Какой-нибудь совет? Что я делаю не так?

1 Ответ

1 голос
/ 05 мая 2020

В ndimage. Угол поворота в градусах. 

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