在我的项目中,我需要检测房屋平面图的屋顶。为此,我必须使用蓝图的正面立面图。我计划使用open-cv HoughLines方法。使用该方法,我需要检测房屋的屋顶。下面我附上了房屋的图像和我的代码。
我使用的房屋平面图。
import cv2
import numpy as np
import matplotlib.pyplot as plt
img = cv2.imread('lol.png')
gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
edges = cv2.Canny(gray,50,150,apertureSize = 3)
lines = cv2.HoughLines(edges,1,np.pi/180,100)
count =0
for rho,theta in lines[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))
if( 20 < 180*theta/np.pi < 88):
cv2.line(img, (x1, y1), (x2, y2), (255, 0, 255), 2)
if (160 > 180 * theta / np.pi > 93):
cv2.line(img, (x1, y1), (x2, y2), (0, 255, 0), 2)
cv2.imwrite('detectP.jpg',img)
plt.imshow(img)
我使用的房屋平面图。
执行代码后得到的结果,
如您所见,检测程序会遍历整个图像。我需要它仅检测线条并在角度线的顶部绘制线条,而不是整个图像。
for rho,theta in lines[0]:
中的theta
是线条的角度。你需要什么帮助? - Dan Mašek