python opencv實(shí)現(xiàn)直線檢測(cè)并測(cè)出傾斜角度(附源碼+注釋)
由于學(xué)習(xí)需要,我想要檢測(cè)出圖片中的直線,并且得到這些直線的角度。于是我在網(wǎng)上搜了好多直線檢測(cè)的代碼,但是沒有搜到附有計(jì)算直線傾斜角度的代碼,所以我花了一點(diǎn)時(shí)間,自己寫了一份直線檢測(cè)并測(cè)出傾斜角度的代碼,希望能夠幫助到大家!
注:這份代碼只能夠檢測(cè)簡(jiǎn)單結(jié)構(gòu)圖片的直線,復(fù)雜結(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) # 對(duì)得到的圖像進(jìn)行形態(tài)學(xué)操作(閉運(yùn)算和開運(yùn)算) mask = cv2.morphologyEx(mask1, cv2.MORPH_CLOSE, kernel) #閉運(yùn)算:表示先進(jìn)行膨脹操作,再進(jìn)行腐蝕操作 mask = cv2.morphologyEx(mask, cv2.MORPH_OPEN, kernel) #開運(yùn)算:表示的是先進(jìn)行腐蝕,再進(jìn)行膨脹操作 # 繪制輪廓 edges = cv2.Canny(mask, 50, 150, apertureSize=3) # 顯示圖片 cv2.imshow("edges", edges) # 檢測(cè)白線 這里是設(shè)置檢測(cè)直線的條件,可以去讀一讀HoughLinesP()函數(shù),然后根據(jù)自己的要求設(shè)置檢測(cè)條件 lines = cv2.HoughLinesP(edges, 1, np.pi / 180, 40,minLineLength=10,maxLineGap=10) print "lines=",lines print "========================================================" i=1 # 對(duì)通過霍夫變換得到的數(shù)據(jù)進(jìn)行遍歷 for line in lines: # newlines1 = lines[:, 0, :] print "line["+str(i-1)+"]=",line x1,y1,x2,y2 = line[0] #兩點(diǎn)確定一條直線,這里就是通過遍歷得到的兩個(gè)點(diǎn)的數(shù)據(jù) (x1,y1)(x2,y2) cv2.line(image,(x1,y1),(x2,y2),(0,0,255),2) #在原圖上畫線 # 轉(zhuǎn)換為浮點(diǎn)數(shù),計(jì)算斜率 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: # 計(jì)算斜率 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)
測(cè)試圖片:
效果圖:
圖像處理比較粗糙,由于時(shí)間問題,也就沒有深入研究了。
到此這篇關(guān)于python opencv實(shí)現(xiàn)直線檢測(cè)并測(cè)出傾斜角度(附源碼+注釋)的文章就介紹到這了,更多相關(guān)opencv 直線檢測(cè)內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
- 利用Opencv中Houghline方法實(shí)現(xiàn)直線檢測(cè)
- OpenCV霍夫變換(Hough Transform)直線檢測(cè)詳解
- Python+OpenCV圖像處理——實(shí)現(xiàn)直線檢測(cè)
- OpenCV實(shí)現(xiàn)圖像的直線檢測(cè)
- Java+opencv3.2.0實(shí)現(xiàn)hough直線檢測(cè)
- opencv3/C++實(shí)現(xiàn)霍夫圓/直線檢測(cè)
- OpenCV利用霍夫變換進(jìn)行直線檢測(cè)
- Opencv Hough算法實(shí)現(xiàn)圖片中直線檢測(cè)
- 詳解在Python中使用OpenCV進(jìn)行直線檢測(cè)
- OpenCV實(shí)現(xiàn)直線檢測(cè)
相關(guān)文章
Ubuntu 下 vim 搭建python 環(huán)境 配置
這篇文章主要介紹了Ubuntu 下 vim 搭建python環(huán)境配置,需要的朋友可以參考下2017-06-06初學(xué)python數(shù)學(xué)建模之?dāng)?shù)據(jù)導(dǎo)入(小白篇)
本篇文章是小白篇初學(xué)python的同學(xué)可以來共同學(xué)習(xí)了,本篇文章主要講解了python數(shù)學(xué)建模過程中的第一步數(shù)據(jù)導(dǎo)入,數(shù)據(jù)導(dǎo)入是所有數(shù)模編程的第一步,比你想象的更重要2021-08-08Python自動(dòng)化運(yùn)維之Ansible定義主機(jī)與組規(guī)則操作詳解
這篇文章主要介紹了Python自動(dòng)化運(yùn)維之Ansible定義主機(jī)與組規(guī)則操作,結(jié)合實(shí)例形式分析了自動(dòng)化運(yùn)維工具Ansible定義主機(jī)與組規(guī)則相關(guān)配置操作與注意事項(xiàng),需要的朋友可以參考下2019-06-06matplotlib bar()實(shí)現(xiàn)多組數(shù)據(jù)并列柱狀圖通用簡(jiǎn)便創(chuàng)建方法
這篇文章主要介紹了matplotlib bar()實(shí)現(xiàn)多組數(shù)據(jù)并列柱狀圖通用簡(jiǎn)便創(chuàng)建方法,文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-02-02