opencv 检测图像线条 霍夫线检测

# Writer : [e-mail]

# Data  : 2020.3.21

import cv2 as cv

import numpy as np

img = cv.imread('../paojie.jpg')

gray = cv.cvtColor(img,cv.COLOR_BGR2GRAY)

# 50,150 为二值化时的阈值 apertureSize为Sobel滤波器的大小

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

cv.imshow('Canny Result',edges)

cv.imwrite('Canny_Result.jpg',edges)

# 高效的霍夫线检测算法

# edges ： 二值图像

# 1    ： ρ

# pi/180： θ

# 100  ： Accumulator threshold parameter. Only those lines are returned that get enough votes ( >threshold ).

# minLineLength ： Minimum length of line. Line segments shorter than this are rejected.

# maxLineGap    ： Maximum allowed gap between line segments to treat them as a single line.

lines = cv.HoughLinesP(edges,1,np.pi/180,100,minLineLength=100,maxLineGap=10)

for line in lines:

    x1,y1,x2,y2 = line[0]

    cv.line(img,(x1,y1),(x2,y2),(0,255,0),2)

cv.imshow('HoughLines Result',img)

cv.imwrite('HoughLines_Result.jpg',img)

cv.waitKey(0)

cv.destroyAllWindows()

 

---

 

原图 ↑

 

Canny_Result.jpg ↑

 

HoughLines_Result.jpg ↑