import cv2 import numpy as np #定义形状检测函数 def ShapeDetection(img): contours, _ = cv2.findContours(img,cv2.RETR_LIST,cv2.CHAIN_APPROX_NOME) #寻找轮廓点 for obj in contours: area = cv2.contourArea(obj) #计算轮廓内区域的面积 # cv2.drawContours(imgContour, obj, -1, (255, 0, 0), 4) #绘制轮廓线 perimeter = cv2.arcLength(obj,True) #计算轮廓周长 approx = cv2.approxPolyDP(obj,0.03*perimeter,True) #获取轮廓角点坐标 CornerNum = len(approx) #轮廓角点的数量 #轮廓对象分类 if (CornerNum == 4) : x, y, w, h = cv2.boundingRect(approx) # 获取坐标值和宽度、高度 if w > 40 and h > 30: cv2.drawContours(imgContour, [approx], -1, (255, 0, 0), 4) #绘制轮廓线 for i in range(4): # 计算角点的坐标 pt = tuple(approx[i][0]) cv2.circle(imgContour, pt, 5, (0, 255, 0), -1) # 绘制角点 # 可选:显示角点的坐标 cv2.putText(imgContour, str(pt), (pt[0] + 10, pt[1] + 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 255), 1) print(approx[i][0]) # 绘制矩形类型和名称 cv2.putText(imgContour, "Rectangel", (x + (w // 2), y + (h // 2)), cv2.FONT_HERSHEY_COMPLEX, 0.6, (0, 0, 0), 1) path = 'B.png' img = cv2.imread(path) imgContour = img.copy() imgGray = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY) #转灰度图 imgBlur = cv2.GaussianBlur(imgGray,(5,5),1) #高斯模糊 imgCanny = cv2.Canny(imgBlur,60,60) #Canny算子边缘检测 ShapeDetection(imgCanny) #形状检测 # cv2.imshow("Original img", img) # cv2.imshow("imgGray", imgGray) # cv2.imshow("imgBlur", imgBlur) # cv2.imshow("imgCanny", imgCanny) cv2.imshow("shape Detection", imgContour) cv2.waitKey(0) 结果如下
02-22
7452