快捷導航

python opencv實現(xiàn)直線檢測并測出傾斜角度(附源碼+注釋)

更新時間：2020年12月31日 11:41:36 作者：一只新手猿

這篇文章主要介紹了python opencv實現(xiàn)直線檢測并測出傾斜角度(附源碼+注釋)，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧

由于學習需要，我想要檢測出圖片中的直線，并且得到這些直線的角度。于是我在網(wǎng)上搜了好多直線檢測的代碼，但是沒有搜到附有計算直線傾斜角度的代碼，所以我花了一點時間，自己寫了一份直線檢測并測出傾斜角度的代碼，希望能夠幫助到大家！

注：這份代碼只能夠檢測簡單結(jié)構(gòu)圖片的直線，復雜結(jié)構(gòu)的圖片還需要設(shè)置合理的參數(shù)

下面展示 源碼。

import cv2
import numpy as np

def line_detect(image):
  # 將圖片轉(zhuǎn)換為HSV
  hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
  # 設(shè)置閾值
  lowera = np.array([0, 0, 221])
  uppera = np.array([180, 30, 255])
  mask1 = cv2.inRange(hsv, lowera, uppera)
  kernel = np.ones((3, 3), np.uint8)

  # 對得到的圖像進行形態(tài)學操作（閉運算和開運算）
  mask = cv2.morphologyEx(mask1, cv2.MORPH_CLOSE, kernel) #閉運算：表示先進行膨脹操作，再進行腐蝕操作
  mask = cv2.morphologyEx(mask, cv2.MORPH_OPEN, kernel)  #開運算：表示的是先進行腐蝕，再進行膨脹操作

  # 繪制輪廓
  edges = cv2.Canny(mask, 50, 150, apertureSize=3)
  # 顯示圖片
  cv2.imshow("edges", edges)
  # 檢測白線  這里是設(shè)置檢測直線的條件，可以去讀一讀HoughLinesP()函數(shù)，然后根據(jù)自己的要求設(shè)置檢測條件
  lines = cv2.HoughLinesP(edges, 1, np.pi / 180, 40,minLineLength=10,maxLineGap=10)
  print "lines=",lines
  print "========================================================"
  i=1
  # 對通過霍夫變換得到的數(shù)據(jù)進行遍歷
  for line in lines:
    # newlines1 = lines[:, 0, :]
    print "line["+str(i-1)+"]=",line
    x1,y1,x2,y2 = line[0]  #兩點確定一條直線，這里就是通過遍歷得到的兩個點的數(shù)據(jù) （x1,y1）(x2,y2)
    cv2.line(image,(x1,y1),(x2,y2),(0,0,255),2)   #在原圖上畫線
    # 轉(zhuǎn)換為浮點數(shù)，計算斜率
    x1 = float(x1)
    x2 = float(x2)
    y1 = float(y1)
    y2 = float(y2)
    print "x1=%s,x2=%s,y1=%s,y2=%s" % (x1, x2, y1, y2)
    if x2 - x1 == 0:
      print "直線是豎直的"
      result=90
    elif y2 - y1 == 0 :
      print "直線是水平的"
      result=0
    else:
      # 計算斜率
      k = -(y2 - y1) / (x2 - x1)
      # 求反正切，再將得到的弧度轉(zhuǎn)換為度
      result = np.arctan(k) * 57.29577
      print "直線傾斜角度為：" + str(result) + "度"
    i = i+1
  #   顯示最后的成果圖
  cv2.imshow("line_detect",image)
  return result

if __name__ == '__main__':
  # 讀入圖片
  src = cv2.imread("lines/line6.jpg")
  # 設(shè)置窗口大小
  cv2.namedWindow("input image", cv2.WINDOW_AUTOSIZE)
  # 顯示原始圖片
  cv2.imshow("input image", src)
  # 調(diào)用函數(shù)
  line_detect(src)
  cv2.waitKey(0)

測試圖片：

在這里插入圖片描述