OpenCV-入门笔记：检测简单的几何形状（多边形逼近）

目标：

import numpy as np
import cv2 as cv

# 1.载入图片
img = cv.imread("shapes.jpg")
# 2.灰度处理
imgGray = cv.cvtColor(img,cv.COLOR_BGR2GRAY)

# 3.二值化，查找轮廓
_,thresh = cv.threshold(imgGray,170,255,cv.THRESH_BINARY_INV)
contours,_ = cv.findContours(thresh,cv.RETR_EXTERNAL,cv.CHAIN_APPROX_NONE)

# 4.多边形逼近
for contour in contours:
    approx = cv.approxPolyDP(contour,0.01*cv.arcLength(contour,True),True)
    cv.drawContours(img,[approx],0,(0,0,0),5)
    x = approx.ravel()[0]
    y = approx.ravel()[1] - 5
    if len(approx) == 3:
        cv.putText(img,"Triangle",(x,y),cv.FONT_HERSHEY_SIMPLEX,0.5,(0,0,0))
    elif len(approx) == 4: 
        x1,y1,w,h = cv.boundingRect(approx)
        aspectRatio = float(w) / h
        print(aspectRatio)