OpenCV обнаруживает края прихожей и прямые линии в режиме реального времени / неподвижные изображения

Question

У меня есть картинка, которая выглядит следующим образом.

Я знаю, что в simplecv вы можете использовать:

img = Image('hallway.jpg')
img.show()
img.edges.show()
lines = img.findLines()
lines = lines.filter(lines.length() > 50) 
lines.show()

Мне интересно, знает ли кто-нибудь какую-либо библиотеку / документ или может указать мне в любом направлении, который способен определять края углов, дверей и c в режиме реального времени или в неподвижных изображениях с OpenCV ?

Answer 1

Opencv python имеет реализации строк Хофа, которые могут помочь. Хотя al go тяжелый, есть вероятностная c версия, которая работает в реальном времени. Вы можете даже настроить параметры, чтобы ускорить его за счет точности.

import cv2
import numpy as np

img = cv2.imread('hallway.jpg')
gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
edges = cv2.Canny(gray,50,150,apertureSize = 3)
minLineLength = 100
maxLineGap = 10
lines = cv2.HoughLinesP(edges,1,np.pi/180,100,minLineLength,maxLineGap)
for x1,y1,x2,y2 in lines[0]:
    cv2.line(img,(x1,y1),(x2,y2),(0,255,0),2)

cv2.imshow("preview", img)
cv2.waitkey(0)

Обратите внимание, что вам может потребоваться настроить пороговые значения в canny и других параметрах в соответствии с вашими требованиями.

Альтернатива это использовать контуры. Это может помочь https://opencv-python-tutroals.readthedocs.io/en/latest/py_tutorials/py_imgproc/py_contours/py_contour_features/py_contour_features.html