6.形状识别 - python+opencv计算机视觉人工智能

课程实现目标

利用USB摄像头，识别图像中的物体形状，并做出框选标识
打开pycharm开发工具，在项目中新建 shapedetector.py 文件作为定义形状的类，文件代码如下：

# coding=utf-8
import cv2

# 创建形状检测类
class ShapeDetector:
	def __init__(self):
		pass

	def detect(self, c):
		# 初始化形状名和近似的轮廓
		shape = "unidentified"
		peri = cv2.arcLength(c, True)
		approx = cv2.approxPolyDP(c, 0.04 * peri, True)

		# 如果当前的轮廓含有3个顶点，则其为三角形
		if len(approx) == 3:
			shape = "triangle"

		# 如果当前的轮廓含有4个顶点，则其可能是矩形或者正方形
		elif len(approx) == 4:
			# 获取轮廓的边界框并计算长和宽的比例
			(x, y, w, h) = cv2.boundingRect(approx)
			ar = w / float(h)

			shape = "square" if ar >= 0.95 and ar <= 1.05 else "rectangle"

		# 如果这个轮廓含有5个顶点，则它是一个多边形
		elif len(approx) == 5:
			shape = "pentagon"