import cv2 as cv
import numpy as np
global a
class ShapeAnalysis:def __init__(self):self.shapes = {'three': 0, 'four': 0, 'five':0, 'six': 0, 'circles': 0}def analysis(self, frame):h, w, ch = frame.shaperesult = np.zeros((h, w, ch), dtype=np.uint8)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)#函数cv.findContours有三个参数，第一个是输入图像，第二个是轮廓检索模式，第三个是轮廓近似方法out_binary, contours, hierarchy = cv.findContours(binary, cv.RETR_EXTERNAL, cv.CHAIN_APPROX_SIMPLE)#shape数组为面积范围areas = np.zeros(shape=[1, 7000])areaj = np.zeros(shape=[1, 7000])aread = np.zeros(shape=[1, 7000])areaw = np.zeros(shape=[1, 7000])areal = np.zeros(shape=[1, 7000])areayuan = np.zeros(shape=[1, 7000])i = 1global afor cnt in range(len(contours)):# 提取与绘制轮廓cv.drawContours(result, contours, cnt, (0, 255, 0), 2)# 轮廓逼近#epsilon，它是从原始轮廓到近似轮廓的最大距离。它是一个准确度参数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['three']count = count+1self.shapes['three'] = count#shape_type = "三角形"area_3 = cv.contourArea(contours[cnt])areas[0, i] = area_3if corners == 4:count = self.shapes['four']count = count + 1self.shapes['four'] = count#shape_type = "four形"area_4 = cv.contourArea(contours[cnt])areaj[0, i] = area_4if corners == 5:count = self.shapes['five']count = count + 1self.shapes['five'] = count#shape_type = "five形"area_5 = cv.contourArea(contours[cnt])areaw[0, i] = area_5#if corners == 6:#     count = self.shapes['six']#    count = count + 1#    self.shapes['six'] = count#   #shape_type = "six形"#    area_6 = cv.contourArea(contours[cnt])#    areal[0, i] = area_6if corners >= 50:count = self.shapes['circles']count = count + 1self.shapes['circles'] = count#shape_type = "圆形"area_y = cv.contourArea(contours[cnt])areayuan[0, i] = area_yi = i + 1area_S = areas.max()area_J = areaj.max()area_Y = areayuan.max()area_D = aread.max()area_W = areaw.max()area_L = areal.max()shape_types = "T"  #三角形shape_typej = "F"  #矩形shape_typey = "Y"  #圆形shape_typew = "F"  #five形shape_typel = "S"  #six形if area_S >5000:a = shape_typesif area_J >5000:a = shape_typejif area_Y >5000:a = shape_typeyif area_W>5000:a = shape_typewif area_L >5000:a = shape_typelreturn self.shapesif __name__ == "__main__":cap = cv.VideoCapture(0)success, img = cap.read()cv.imshow('frame',img)cv.imwrite("/home/car/AI_match/4.jpg", img)src = cv.imread("/home/car/AI_match/4.jpg")ld = ShapeAnalysis()ld.analysis(src)print(a)# cv.waitKey(0)#  cv.destroyAllWindows()