[Python] 纯文本查看 复制代码
#边缘检测
import cv2
import numpy as np
#定义形状检测函数
def ShapeDetection(img):
contours,hierarchy = cv2.findContours(img,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_NONE) #寻找轮廓点
for obj in contours:
area = cv2.contourArea(obj) #计算轮廓内区域的面积
if area>=5:
cv2.drawContours(imgContour, obj, -1, (255, 0, 0), 4) #绘制轮廓线
perimeter = cv2.arcLength(obj,True) #计算轮廓周长
approx = cv2.approxPolyDP(obj,0.02*perimeter,True) #获取轮廓角点坐标
CornerNum = len(approx) #轮廓角点的数量
x, y, w, h = cv2.boundingRect(approx) #获取坐标值和宽度、高度
if w>200 and h>200:
if CornerNum == 4:
if w==h:
objType= "Square"
else:objType="N"
print(str(x)+" "+str(y))
cv2.rectangle(imgContour,(x,y),(x+w,y+h),(0,0,255),2) #绘制边界框
cv2.putText(imgContour,objType,(x+(w//2),y+(h//2)),cv2.FONT_HERSHEY_COMPLEX,0.6,(0,0,0),1) #绘制文字
oblk = 'backImg.png' #大图
template = cv2.imread(oblk) # 使用灰度图方式加载大图
img_1 = cv2.resize(template,(1000,500))
cv2.imwrite('backImg.png', img_1)
path = 'backImg.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("shape Detection", imgContour)
cv2.waitKey(0)