Как определить линии на изображении с помощью метода houghlines в opencv python? - PullRequest
Новая
       110

Как определить линии на изображении с помощью метода houghlines в opencv python?

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

У меня есть изображение плана этажа, на котором есть только стены. С помощью метода opencv houghlines я хочу идентифицировать эти строки. Я пробовал этот метод и вероятностный c метод. Вероятностный метод c houghlines не идентифицирует все строки. Когда я использую метод houghlines, он не определяет все строки должным образом. Прикрепляю изображение на стене и полученный результат. Я новичок в opencv, поэтому не понимаю, какой параметр нужно изменить. Кто-нибудь может мне помочь? This is the image of the wall

This is the result I got

вот мой код: 


import cv2
import numpy as np

img = cv2.imread("C:\\Users\\User\\Desktop\\tkinter_codes\\floorplans\\ROBIN\\Dataset_3roomsmall\\wallimage.jpg")
gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
edges = cv2.Canny(gray,10,150,apertureSize = 3)

lines = cv2.HoughLines(edges,1,np.pi/180,100,100,10)
for x in range(len(lines)):
    for rho,theta in lines[x]:
        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(img,(x1,y1),(x2,y2),(0,255,0),2)

cv2.imwrite("C:\\Users\\User\\Desktop\\tkinter_codes\\floorplans\\ROBIN\\Dataset_3roomsmall\\stackimage.jpg",img)
cv2.imshow('image',img)
k = cv2.waitKey(0)
cv2.destroyAllWindows()
...