Python?OpenCV實現(xiàn)圖形檢測示例詳解
圖形檢測在計算機視覺開發(fā)中是一項非常重要的操作,算法通過對圖像的檢測,分析出圖像中可能存在哪些形狀。除此之外,除了讓計算機識別輪廓之外,輪廓也需要讓人看到,這就需要再分別讀這些輪廓的形狀進行描繪。
1. 輪廓識別與描繪
cv2.findContours() & cv2.drawContours() 方法
在Python中OpenCV提供了findContours() 方法來判斷圖像的輪廓,drawContours()方法來繪制輪廓。
1.1 cv2.findComtours()方法
cv2.findComtours()方法的語法如下
cv2.contours, hierarchy = findContours(image, mode, method)
其中
- image 即原圖像
- mode 輪廓檢索模式,具體參數(shù)被總結在了下表中
- method 使用的方法,具體參數(shù)也被總結在了下表中
contours是一個列表,列表的每一個元素都是由某個輪廓的像素的坐標組成的數(shù)組。
hierarchy是輪廓與輪廓之間的層次關系。
mode取值表
mode值 | 描述 |
---|---|
cv2.RETR_EXTERNAL | 只檢測外輪廓 |
cv2.RETR_LIST | 檢測所有輪廓,但不建立層次關系 |
cv2.RETR_CCOMP | 檢測所有輪廓,并建立兩級層次關系 |
cv2.RETR_TREE | 建立所有輪廓,并建立樹狀結構的層次關系 |
method取值表
cv2.PATH_APPROX_ | 儲存輪廓上所有點 |
---|---|
cv2.PATH_APPROX_NONE | 只保存水平、垂直或對角線輪廓的端點 |
cv2.PATH_APPROX_SIMPLE | Ten-Chinl |
cv2.PATH_APPROX_TC89_L1 | Ten-Chinl近似算法的一種 |
cv2.PATH_APPROX_TC89_KCOS | Ten-Chinl近似算法的一種 |
1.2 cv2.drawContours() 方法
cv2.drawContours(image, contours, contourIdx, color, thickness=None, lineType=None, hierarchy=None, maxLevel=None, offset=None)
- image 目標圖像
- contours findComtours()方法得到的輪廓列表
- contourldx 輪廓中列表中,繪制輪廓的對象的索引,如果為-1則表示繪制所有
- color 繪制線條時的顏色,使用BGR格式描述
- thickness 線條粗細程度,-1表示實心
- lineType 繪制輪廓時線條的類型(可選參數(shù))
- hierarchy findComtours()方法得到的層次關系(可選參數(shù))
- maxLevel 繪制輪廓的層次深度,最深繪制在maxLevel層。(可選參數(shù))
- offset 偏移量 (可選參數(shù))
drawContours() 方法返回的是一個圖像(數(shù)組)。
1.3 代碼示例
以隊此小鳥圖操作為例(test1.jpg):
import cv2 img = cv2.imread("test1.jpg") # 彩色圖像轉為變成單通道灰度圖像 gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) # 二值化處理 t, binary = cv2.threshold(gray, 127, 255, cv2.THRESH_BINARY) # 檢測圖像中出現(xiàn)的所有輪廓,記錄輪廓的每一個點 contours, hierarchy = cv2.findContours(binary, cv2.RETR_LIST, cv2.CHAIN_APPROX_NONE) # 繪制所有輪廓,寬度為5,顏色為紅色 cv2.drawContours(img, contours, -1, (0, 0, 255), 5) cv2.imshow("img", img) cv2.waitKey() cv2.destroyAllWindows()
程序執(zhí)行結果展示如下,圖中所有能夠識別出的輪廓被描出:
2. 輪廓擬合
輪廓擬合,即,將凹凸不平的輪廓用平整的幾何圖形體現(xiàn)出來。這里展示使用矩形和圓形擬合兩種方法。
2.1 矩形包圍框擬合 - cv2.boundingRect()
在Python中OpenCV提供了cv2.boundingRect()來計算輪廓的最小矩形邊界的坐標 ,其語法如下
retval = cv2.boundingRect(array)
其中參數(shù)array為輪廓數(shù)組。即,cv2.findComtours()方法的執(zhí)行結果中的contours中的元素。
返回值retval是一個包含著四個整數(shù)值的元組,四個值依次是左上角頂點的橫坐標,左上角頂點的縱坐標,矩形的寬,矩形的高。
常常也可以寫成x,y,w,h = retval = cv2.boundingRect(array)
還以這張小鳥圖片(test1.jpg)為例,在上一個部分的示例中,我們找出并繪制出了圖中所有的輪廓,經過調試,發(fā)現(xiàn)被識別出的輪廓共有94個。
我們要從數(shù)組列表中,選擇出表示小鳥輪廓的位置的數(shù)組。
從上圖中可以看出,小鳥的輪廓是所有輪廓中最大的。即該數(shù)組的 shape[0] 最大。即我們只用找出 shape[0]最大的即可。
import cv2 img = cv2.imread("test1.jpg")? # 從彩色圖像變成單通道灰度圖像 gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) ? # 將灰度圖像進行二值化閾值處理 t, binary = cv2.threshold(gray, 127, 255, cv2.THRESH_BINARY) # 獲取二值化圖像中的輪廓極輪廓層次數(shù)據(jù) contours, hierarchy = cv2.findContours(binary, cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE) # 找出小鳥的輪廓 n1 = 1 n2 = 0 index = 0 for arr in contours: ? ? if len(arr) > n1: ? ? ? ? n1 = len(arr) ? ? ? ? index = n2 ? ? n2 += 1 print(index) x, y, w, h = cv2.boundingRect(contours[index]) # 繪制紅色矩形 cv2.rectangle(img, (x, y), (x + w, y + h), (0, 0, 255), 2) ? cv2.imshow("img", img)? cv2.waitKey() cv2.destroyAllWindows()?
繪制出矩形包圍框效果如下:
2.2圓形包圍框擬合 - cv2.minEnclosingCircle()
在Python中OpenCV提供了cv2.minEnclosingCircle()來計算輪廓的最小圓形邊界的圓心和半徑 ,其語法如下
center,radius = minEnclosingCircle(points)
其中
- points的輪廓數(shù)組
- center最小圓形包圍框的圓心的橫縱坐標。是元組類型。
- radius是最小圓形包圍款更多半徑,浮點類型。
同樣的算法,只是這次調用cv2.minEnclosingCircle()方法,其他不變:
import cv2 img = cv2.imread("test1.jpg") # 讀取原圖 gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) # 從彩色圖像變成單通道灰度圖像 # 將灰度圖像進行二值化閾值處理 t, binary = cv2.threshold(gray, 127, 255, cv2.THRESH_BINARY) # 獲取二值化圖像中的輪廓極輪廓層次數(shù)據(jù) contours, hierarchy = cv2.findContours(binary, cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE) n1 = 1 n2 = 0 index = 0 for arr in contours: if len(arr) > n1: n1 = len(arr) index = n2 n2 += 1 center, radius = cv2.minEnclosingCircle(contours[index]) # 圓心點橫坐標轉為近似整數(shù) x = int(round(center[0])) # 圓心點縱坐標轉為近似整數(shù) y = int(round(center[1])) cv2.circle(img, (x, y), int(radius), (0, 0, 255), 2) cv2.imshow("img", img) cv2.waitKey() cv2.destroyAllWindows()
繪制出圓形包圍框效果如下(因為尺寸問題,只畫出了一部分):
3. 凸包 繪制
使用矩形框和圓形框對圖形的貼合程度往往都會較差。為了提高這個貼合程度,我們可以使用“凸包”。
所謂凸包,就是最逼近輪廓的多邊形。
在Python中OpenCV提供了 cv2.bonvexHull()方法來計算凸包
cv2.bonvexHull()方法語法如下:
hull = convexHull(points, clockwise=None, returnPoints=None)
其中
- points 是輪廓數(shù)組
- clockwise 是布爾類型的參數(shù),默認為True,表示凸包中的點按順時針排序,為False時則按逆時針 排序。
- returnPoints 是布爾類型的參數(shù),默認為True時返回點坐標。如果為False則返回點索引。
返回值hull是凸包的點陣數(shù)組
依然以“test1.jpg為例”,為圖中的小鳥繪制凸包:
import cv2 img = cv2.imread("test1.jpg") # 轉為灰度圖像 gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) # 二值化閾值處理 ret, binary = cv2.threshold(gray, 127, 225, cv2.THRESH_BINARY) # 檢測圖像中出現(xiàn)的所有輪廓 contours, hierarchy = cv2.findContours(binary, cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE) n1 = 1 n2 = 0 index = 0 for arr in contours: if len(arr) > n1: n1 = len(arr) index = n2 n2 += 1 hull = cv2.convexHull(contours[index]) cv2.polylines(img, [hull], True, (0, 0, 255), 2) cv2.imshow("img", img) cv2.waitKey() cv2.destroyAllWindows()
4. Canny邊緣檢測 - cv2.Canny()
4.1 cv2.Canny() 用法簡介
Canny邊緣檢測算法是John F.Canny在1986年開發(fā)的一個多級邊緣檢測算法。
Canny邊緣檢測算法通過像素的梯度變化尋找圖像的邊緣,最終可以繪制出十分精細的二值邊緣圖像
edges = cv2.Canny(image, threshold1, threshold2, apertureSize=None, L2gradient=None)
其中
- image 即原圖像
- threshold1 第一個閾值,一般為最小閾值
- threshold2 第二個閾值,一般為最大閾值
- apertureSize Sobel算子的孔徑大小
- L2gradient 計算圖像梯度的標識。默認為False。為True時采用更精準的算法進行計算。
關于這兩個閾值怎么用,這涉及到了算法的底層邏輯,還請自行探索。這里一種可以接受的解釋是:低于閾值1的像素點,會被認為不構成邊緣,而高于閾值2的像素點,會被認為構成邊緣。
最后返回值edges是一個二值的灰度圖像。
4.2 代碼示例
下邊對test1.jpg以三組不同的閾值來做Canny邊緣檢測,根據(jù)處理結果感受算法效果:
- 當閾值為 10-50 時
import cv2 img = cv2.imread("test1.jpg") r1 = cv2.Canny(img, 10, 50) cv2.imshow("r1", r1) cv2.waitKey() cv2.destroyAllWindows()
- 當閾值為100-200時
import cv2 img = cv2.imread("test1.jpg") r2 = cv2.Canny(img, 100, 200) cv2.imshow("r2", r2) cv2.waitKey() cv2.destroyAllWindows()
- 當閾值為400-600時
import cv2 img = cv2.imread("test1.jpg") r3 = cv2.Canny(img, 400, 600) cv2.imshow("r3", r3) cv2.waitKey() cv2.destroyAllWindows()
5. 霍夫變換
5.1 概述
霍夫變換是一種特征檢測,通過霍夫變換可以檢測出圖像中存在的特殊的形狀。比如,直線,圓等。
霍夫變換檢測直線時,算法有兩個,
一個是cv2.HoughLines() 方法,用于檢測無限延長的直線;
另一個是cv2.HoughLinesP() 方法,用于檢測線段。
霍夫變換檢測圓,使用的是**cv2.HoughCircles()**方法。
使用這三個方法前,都要先對圖像進行降噪處理(使用濾波器),以去除干擾。
5.2 cv2.HoughLines() 檢測直線
cv2.HoughLines()語法如下:
lines = cv.HoughLines( image, rho, theta, threshold[,srn][,stn])
其中
- image 即原圖像
- rho 指的是搜索直線使用的半徑步長,其值為1時表示檢測所有。(即極坐標中的ρ)
- theta 指的是搜索直線的角度,值為π/180°時,表示檢測所有角度。(即極坐標中的θ)
- threshold 指的是閾值,點的數(shù)量(也稱投票數(shù))。因為直線的長度取決于直線上的點的數(shù)量。所以如果達不到這個長度,就不會被
- 判定為直線。同理,當該閾值越小,檢測出的直線也就越多。
- srn 對于多尺度霍夫變換,srn表示對rho的 距離分辨率 的除數(shù)
- stn 對于多尺度霍夫變換,stn表示對theta的 距離分辨率 的除數(shù)
- min_theat 對于標準和多尺度Hough變換,檢查線條的最小角度。必須介于0和最大θ之間。
- max_theat 對于標準和多尺度Hough變換,檢查線條的最大角度。必須介于min_theta和CV_PI之間。
返回值lines,是一個數(shù)組,shape為(n, 1, 2),n表示檢測出的所有線段數(shù)目,每個線段用極坐標(ρ, θ)表示。
以此跨海大橋圖(test2.jpg)為例,對其使用cv2.HoughLines()方法,并繪制直線:
import cv2 import numpy as np img = cv2.imread("test2.jpg") o = img.copy() # 使用中值濾波進行降噪 o = cv2.medianBlur(o, 5) # 從彩色圖像變成單通道灰度圖像 gray = cv2.cvtColor(o, cv2.COLOR_BGR2GRAY) # 繪制邊緣圖像 binary = cv2.Canny(o, 50, 150) # 檢測直線 不限步長,不限角度,至少50個點確定一條線 lines = cv2.HoughLines(binary, 1, np.pi / 180, 50) # print(lines) # print(lines.shape) for line in lines: rho, theta = line[0] a = np.cos(theta) b = np.sin(theta) x0 = a * rho y0 = b * rho x1 = int(x0 + 1000 * (-b)) y1 = int(y0 + 1000 * (a)) x2 = int(x0 - 1000 * (-b)) y2 = int(y0 - 1000 * (a)) cv2.line(img, (x1, y1), (x2, y2), (0, 0, 255), 2) cv2.imshow("img", img) cv2.waitKey() cv2.destroyAllWindows()
識別效果如下,可以自行調節(jié)參數(shù)進行改良,并在這個過程中具體感受每個參數(shù)的作用。
當選擇至少50個點確定一條線時,一共檢測出168根直線:
當選擇至少100個點確定一條線時,一共檢測出35根直線:
當選擇至少300個點確定一條直線時,符合要求的直線有3根:
當選擇至少380個點確定一條直線時,這樣的直線還剩一根:
相比檢測直線,檢測線段的cv2.HoughLinesP()相對要更常用些。
5.3 cv2.HoughLinesP() 檢測線段
lines = cv2.HoughLinesP(image, rho, theta, threshold, minLineLength=None, maxLineGap=None)
檢測線段
- image 原圖
- rho 檢測直線使用的半徑步長,值為1時表示所有可能的半徑步長
- theta 搜索直線的角度
- threshold 閾值,該值越小,檢測出的直線越多。
- minLineLength表示線段的最小長度,小于該長度的線段不會被記錄在結果中。值越大線段越少。
- maxLineGap 表示允許將同一行的點連接起來的最大距離。 值越大線段越多。
返回值lines,是一個數(shù)組,shape為(n, 1, 4),n表示檢測出的所有線段數(shù)目,4指的是每個線段的兩端端點的笛卡爾坐標(x, y) 坐標的四個點。
其中minLineLength(最小線段長度)和maxLineGap(最小線段距離)兩個參數(shù),都是越大,識別的線段越少。
import cv2 import numpy as np img = cv2.imread("test2.jpg") o = img.copy() # 使用中值濾波進行降噪 o = cv2.medianBlur(o, 5) # 從彩色圖像變成單通道灰度圖像 gray = cv2.cvtColor(o, cv2.COLOR_BGR2GRAY) # 繪制邊緣圖像 binary = cv2.Canny(o, 50, 150) # 檢測線段,不限步長,不限角度,至少100個點確定一條線。最大將距離為200的線段連城一條線。 lines = cv2.HoughLinesP(binary, 1, np.pi / 180, 50, minLineLength=100, maxLineGap=200) print(lines.shape) for line in lines: x1, y1, x2, y2 = line[0] cv2.line(img, (x1, y1), (x2, y2), (0, 0, 255), 2) cv2.imshow("canny", binary) cv2.imshow("img", img) cv2.waitKey() cv2.destroyAllWindows()
二值化邊緣圖案效果:
描繪線段效果(minLineLength=100)至少100個點確定一條線段(上邊代碼),最大將距離為200的線段連接起來
共描繪了23條線:
描繪線段效果(minLineLength=500)至少500個點確定一條線段,最大將距離為200的線段連接起來
共描繪了6條線段:
描繪線段效果(minLineLength=720)至少720個點確定一條線段,最大將距離為200的線段連接起來
共描繪了一條線:
5.4 cv2.檢測圓 - HoughCircles()
circles = HoughCircles(image, method, dp, minDist, circles=None, param1=None, param2=None, minRadius=None, maxRadius=None)
其中
- image 原圖像(降噪、灰度處理后的)
- method 檢測方法。
- dp 累加器分辨率與原圖分辨率之比的倒數(shù)。值為1時累加器與原圖像有著相同的分辨率。通常選擇1作為參數(shù)。(值為2時則累加器的分辨率是原圖像的一半)
- minDist 圓心之間的最小距離
- param1 參數(shù)1,表示Canny邊緣檢測的最大閾值。是可選參數(shù)。
- param2 參數(shù)2,表示檢測結果投票數(shù)。即至少多少個點確定一個圓。值越大,識別的圓越少,約精準。是可選參數(shù)。
- minRadius 圓環(huán)的最小半徑(可選參數(shù))
- maxRadius 圓環(huán)的最大半徑(可選參數(shù))
返回值circles是一個數(shù)組,數(shù)組內是所有檢測出的圓環(huán),shape為(n,1,3)。其中3表示圓心的x坐標,圓心的y坐標和半徑長度三個指標。
了解完語法,
接下來我們來檢測下圖(test3.jpg)中的客家土樓中的圓形。
import cv2 import numpy as np img = cv2.imread("test3.jpg") # 使用中值濾波進行降噪 o = cv2.medianBlur(img, 11) # 從彩色圖像變成單通道灰度圖像 gray = cv2.cvtColor(o, cv2.COLOR_BGR2GRAY) # 展示灰度圖像 cv2.imshow('gray', gray) # 檢測圓環(huán),圓心最小間距為50,Canny最大閾值為40,投票數(shù)超過63。最小半徑為10,最大半徑為50 circles = cv2.HoughCircles(gray, cv2.HOUGH_GRADIENT, 1, 50, param1=40, param2=63, minRadius=10, maxRadius=100) # 將數(shù)組元素四舍五入成整數(shù) circles = np.uint(np.around(circles)) # 遍歷圓環(huán)結果 for c in circles[0]: # 圓心橫坐標、縱坐標和圓半徑 x, y, r = c # 繪制圓環(huán) cv2.circle(img, (x, y), r, (0, 0, 255), 3) # 繪制圓心 cv2.circle(img, (x, y), 2, (0, 0, 255), 3) cv2.imshow("img", img) cv2.waitKey() cv2.destroyAllWindows()
降過噪的灰度圖像如下:
識別結果呈現(xiàn)如下,如圖成功識別出了圖中所有土樓的圓形:
以上就是Python OpenCV實現(xiàn)圖形檢測示例詳解的詳細內容,更多關于Python OpenCV圖形檢測的資料請關注腳本之家其它相關文章!
相關文章
python 已知一個字符,在一個list中找出近似值或相似值實現(xiàn)模糊匹配
今天小編就為大家分享一篇python 已知一個字符,在一個list中找出近似值或相似值實現(xiàn)模糊匹配,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-02-02python開發(fā)中range()函數(shù)用法實例分析
這篇文章主要介紹了python開發(fā)中range()函數(shù)用法,以實例形式較為詳細的分析了Python中range()函數(shù)遍歷列表的相關技巧,需要的朋友可以參考下2015-11-11