經常看到有學習OpenCV不久的人提問,如何識別一些簡單的幾何形狀與它們的顏色,其實通過OpenCV的輪廓發現與幾何分析相關的函數,只需不到100行的代碼就可以很好的實現這些簡單幾何形狀識別與對象測量相關操作。本文就會演示給大家如何通過OpenCV 輪廓發現與幾何分析相關函數實現如下功能:
- 幾何形狀識別(識別三角形、四邊形/矩形、多邊形、圓)
- 計算幾何形狀面積與周長、中心位置
- 提取幾何形狀的顏色
在具體代碼實現與程序演示之前,我們先要搞清楚一些概念。
一:基本概念與函數介紹
1. 輪廓(contours)
什麼是輪廓,簡單說輪廓就是一些列點相連組成形狀、它們擁有同樣的顏色、輪廓發現在圖像的對象分析、對象檢測等方面是非常有用的工具,在OpenCV中使用輪廓發現相關函數時候要求輸入圖像是二值圖像,這樣便於輪廓提取、邊緣提取等操作。輪廓發現的函數與參數解釋如下:
findContours(image, mode, method, contours=None, hierarchy=None, offset=None) - image輸入/輸出的二值圖像 - mode 返回輪廓的結構、可以是List、Tree、External - method 輪廓點的編碼方式,基本是基於鏈式編碼 - contours 返回的輪廓集合 - hieracrchy 返回的輪廓層次關係 - offset 點是否有位移
2. 多邊形逼近
多邊形逼近,是通過對輪廓外形無限逼近,刪除非關鍵點、得到輪廓的關鍵點,不斷逼近輪廓真實形狀的方法,OpenCV中多邊形逼近的函數與參數解釋如下:
approxPolyDP(curve, epsilon, closed, approxCurve=None) - curve 表示輸入的輪廓點集合 - epsilon 表示逼近曲率,越小表示相似逼近越厲害 - close 是否閉合
3. 幾何距計算
圖像幾何距是圖像的幾何特徵,高階幾何距中心化之後具有特徵不變性,可以產生Hu距輸出,用於形狀匹配等操作,這裏我們通過計算一階幾何距得到指定輪廓的中心位置,計算幾何距的函數與參數解釋如下:
moments(array, binaryImage=None) - array表示指定輸入輪廓 - binaryImage默認爲None
二:代碼實現與演示
基於輪廓發現與多邊形逼近、幾何距實現幾何形狀識別與對象測量,測量時候還用到另外兩個相關API分別是計算輪廓的周長與面積。具體用法在代碼中體現。整個代碼實現分爲如下幾步完成: 1.圖像二值化
# 二值化圖像 print("start to detect lines...\n") 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.輪廓發現
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]) + ")"
運行顯示原圖如下:
分析結果:
控制檯輸出: