快捷導(dǎo)航

python hough變換檢測直線的實(shí)現(xiàn)方法

更新時間：2019年07月12日 11:38:05 作者：Jackilina_Stone

這篇文章主要介紹了python hough變換檢測直線的實(shí)現(xiàn)方法，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

1 原理

2 檢測步驟

將參數(shù)空間(ρ,θ) 量化成m*n(m為ρ的等份數(shù)，n為θ的等份數(shù))個單元，并設(shè)置累加器矩陣，初始值為0；

對圖像邊界上的每一個點(diǎn)（x,y）帶入ρ=xcosθ+ysinθ，求得每個θ對應(yīng)的ρ值，并在ρ和θ所對應(yīng)的單元，將累加器加1，即：Q(i,j)=Q(i,j)+1；

檢驗(yàn)參數(shù)空間中每個累加器的值，累加器最大的單元所對應(yīng)的ρ和θ即為直角坐標(biāo)系中直線方程的參數(shù)。

3 接口

image：二值圖像，canny邊緣檢測輸出。這里是result。
rho: 以像素為單位的距離精度，這里為1像素。如果想要檢測的線段更多，可以設(shè)為0.1。
theta：以弧度為單位的角度精度，這里為numpy.pi/180。如果想要檢測的線段更多，可以設(shè)為0.01 * numpy.pi/180。
threshod: 閾值參數(shù)，int類型，超過設(shè)定閾值才被檢測出線段，這里為10。
minLineLength：線段以像素為單位的最小長度。
maxLineGap：同一方向上兩條線段判定為一條線段的最大允許間隔。

4 代碼及結(jié)果

import os
import numpy as np
import cv2
from PIL import Image, ImageEnhance
import math
 
 
def img_processing(img):
  # 灰度化
  gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
  ret, binary = cv2.threshold(gray, 0, 255, cv2.THRESH_OTSU)
  # canny邊緣檢測
  edges = cv2.Canny(binary, 50, 150, apertureSize=3)
  return edges
 
 
def line_detect(img):
  img = Image.open(img)
  img = ImageEnhance.Contrast(img).enhance(3)
  # img.show()
  img = np.array(img)
  result = img_processing(img)
  # 霍夫線檢測
  lines = cv2.HoughLinesP(result, 1, 1 * np.pi/180, 10, minLineLength=10, maxLineGap=5)
  # print(lines)
  print("Line Num : ", len(lines))
 
  # 畫出檢測的線段
  for line in lines:
    for x1, y1, x2, y2 in line:
      cv2.line(img, (x1, y1), (x2, y2), (255, 0, 0), 1)
    pass
  img = Image.fromarray(img, 'RGB')
  img.show()
 
 
if __name__ == "__main__":
  line_detect("1.jpg")
  pass

原圖如下：