OpenCV中幾何形狀識別與測量

經(jīng)常看到有學(xué)習(xí)OpenCV不久的人提問，如何識別一些簡單的幾何形狀與它們的顏色，其實通過OpenCV的輪廓發(fā)現(xiàn)與幾何分析相關(guān)的函數(shù)，只需不到100行的代碼就可以很好的實現(xiàn)這些簡單幾何形狀識別與對象測量相關(guān)操作。本文就會演示給大家如何通過OpenCV 輪廓發(fā)現(xiàn)與幾何分析相關(guān)函數(shù)實現(xiàn)如下功能：

幾何形狀識別（識別三角形、四邊形/矩形、多邊形、圓）

計算幾何形狀面積與周長、中心位置

提取幾何形狀的顏色

在具體代碼實現(xiàn)與程序演示之前，我們先要搞清楚一些概念。

一：基本概念與函數(shù)介紹

1. 輪廓(contours)

什么是輪廓，簡單說輪廓就是一些列點相連組成形狀、它們擁有同樣的顏色、輪廓發(fā)現(xiàn)在圖像的對象分析、對象檢測等方面是非常有用的工具，在OpenCV中使用輪廓發(fā)現(xiàn)相關(guān)函數(shù)時候要求輸入圖像是二值圖像，這樣便于輪廓提取、邊緣提取等操作。輪廓發(fā)現(xiàn)的函數(shù)與參數(shù)解釋如下：

findContours(image, mode, method, contours=None, hierarchy=None, offset=None)

- image輸入/輸出的二值圖像

- mode 返回輪廓的結(jié)構(gòu)、可以是List、Tree、External

- method 輪廓點的編碼方式，基本是基于鏈?zhǔn)骄幋a

- contours 返回的輪廓集合

- hieracrchy 返回的輪廓層次關(guān)系

- offset 點是否有位移

2. 多邊形逼近

多邊形逼近，是通過對輪廓外形無限逼近，刪除非關(guān)鍵點、得到輪廓的關(guān)鍵點，不斷逼近輪廓真實形狀的方法，OpenCV中多邊形逼近的函數(shù)與參數(shù)解釋如下：

approxPolyDP(curve, epsilon, closed, approxCurve=None)

- curve 表示輸入的輪廓點集合

- epsilon 表示逼近曲率，越小表示相似逼近越厲害

- close 是否閉合

3. 幾何距計算

圖像幾何距是圖像的幾何特征，高階幾何距中心化之后具有特征不變性，可以產(chǎn)生Hu距輸出，用于形狀匹配等操作，這里我們通過計算一階幾何距得到指定輪廓的中心位置，計算幾何距的函數(shù)與參數(shù)解釋如下：

moments(array, binaryImage=None)

- array表示指定輸入輪廓

- binaryImage默認(rèn)為None

二：代碼實現(xiàn)與演示

基于輪廓發(fā)現(xiàn)與多邊形逼近、幾何距實現(xiàn)幾何形狀識別與對象測量，測量時候還用到另外兩個相關(guān)API分別是計算輪廓的周長與面積。具體用法在代碼中體現(xiàn)。整個代碼實現(xiàn)分為如下幾步完成：1.圖像二值化

# 二值化圖像

print("start to detect lines... ")

gray = cv.cvtColor(frame, cv.COLOR_BGR2GRAY)

ret, binary = cv.threshold(gray, 0, 255, cv.THRESH_BINARY_INV | cv.THRESH_OTSU)

cv.imshow("input image", frame)

2.輪廓發(fā)現(xiàn)

out_binary, contours, hierarchy = cv.findContours(binary, cv.RETR_EXTERNAL, cv.CHAIN_APPROX_SIMPLE)

for cnt in range(len(contours)):

# 提取與繪制輪廓

cv.drawContours(result, contours, cnt, (0, 255, 0), 2)

3.幾何形狀識別

# 輪廓逼近

epsilon = 0.01 * cv.arcLength(contours[cnt], True)

approx = cv.approxPolyDP(contours[cnt], epsilon, True)

# 分析幾何形狀

corners = len(approx)

shape_type = ""

if corners == 3:

count = self.shapes['triangle']

count = count+1

self.shapes['triangle'] = count

shape_type = "三角形"

if corners == 4:

count = self.shapes['rectangle']

count = count + 1

self.shapes['rectangle'] = count

shape_type = "矩形"

if corners >= 10:

count = self.shapes['circles']

count = count + 1

self.shapes['circles'] = count

shape_type = "圓形"

if 4 < corners < 10:

count = self.shapes['polygons']

count = count + 1

self.shapes['polygons'] = count

shape_type = "多邊形"

4.測量周長、面積、計算中心

# 求解中心位置

mm = cv.moments(contours[cnt])

cx = int(mm['m10'] / mm['m00'])

cy = int(mm['m01'] / mm['m00'])

cv.circle(result, (cx, cy), 3, (0, 0, 255), -1)

# 計算面積與周長

p = cv.arcLength(contours[cnt], True)

area = cv.contourArea(contours[cnt])

5.顏色提取

# 顏色分析

color = frame[cy][cx]

color_str = "(" + str(color[0]) + ", " + str(color[1]) + ", " + str(color[2]) + ")"

運行顯示原圖如下：

分析結(jié)果：

控制臺輸出：

審核編輯 ：李倩