1. 程式人生 > 程式設計 >python opencv實現直線檢測並測出傾斜角度(附原始碼+註釋)

python opencv實現直線檢測並測出傾斜角度(附原始碼+註釋)

由於學習需要,我想要檢測出圖片中的直線,並且得到這些直線的角度。於是我在網上搜了好多直線檢測的程式碼,但是沒有搜到附有計算直線傾斜角度的程式碼,所以我花了一點時間,自己寫了一份直線檢測並測出傾斜角度的程式碼,希望能夠幫助到大家!

注:這份程式碼只能夠檢測簡單結構圖片的直線,複雜結構的圖片還需要設定合理的引數

下面展示 原始碼

import cv2
import numpy as np

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

  # 對得到的影象進行形態學操作(閉運算和開運算)
  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)
  # 檢測白線  這裡是設定檢測直線的條件,可以去讀一讀HoughLinesP()函式,然後根據自己的要求設定檢測條件
  lines = cv2.HoughLinesP(edges,1,np.pi / 180,40,minLineLength=10,maxLineGap=10)
  print "lines=",lines
  print "========================================================"
  i=1
  # 對通過霍夫變換得到的資料進行遍歷
  for line in lines:
    # newlines1 = lines[:,:]
    print "line["+str(i-1)+"]=",line
    x1,y1,x2,y2 = line[0]  #兩點確定一條直線,這裡就是通過遍歷得到的兩個點的資料 (x1,y1)(x2,y2)
    cv2.line(image,(x1,y1),(x2,y2),(0,255),2)   #在原圖上畫線
    # 轉換為浮點數,計算斜率
    x1 = float(x1)
    x2 = float(x2)
    y1 = float(y1)
    y2 = float(y2)
    print "x1=%s,x2=%s,y1=%s,y2=%s" % (x1,y2)
    if x2 - x1 == 0:
      print "直線是豎直的"
      result=90
    elif y2 - y1 == 0 :
      print "直線是水平的"
      result=0
    else:
      # 計算斜率
      k = -(y2 - y1) / (x2 - x1)
      # 求反正切,再將得到的弧度轉換為度
      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")
  # 設定視窗大小
  cv2.namedWindow("input image",cv2.WINDOW_AUTOSIZE)
  # 顯示原始圖片
  cv2.imshow("input image",src)
  # 呼叫函式
  line_detect(src)
  cv2.waitKey(0)

測試圖片:

在這裡插入圖片描述

效果圖:

在這裡插入圖片描述

影象處理比較粗糙,由於時間問題,也就沒有深入研究了。

到此這篇關於python opencv實現直線檢測並測出傾斜角度(附原始碼+註釋)的文章就介紹到這了,更多相關opencv 直線檢測內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們!