opencv轮廓检测正方形
时间: 2024-05-28 11:08:11 浏览: 12
在OpenCV中,可以通过轮廓检测来检测出图像中的正方形。具体步骤如下:
1. 将图像进行灰度化和二值化处理,使图像变成黑白的二值图像。
2. 使用findContours()函数对图像进行轮廓检测,得到所有的轮廓。
3. 对于每个轮廓,使用approxPolyDP()函数将其近似为一个多边形。
4. 对于每个近似得到的多边形,判断其是否为正方形,可以通过计算其周长和面积来进行判断。
下面是一个示例代码:
```python
import cv2
# 读取图像
img = cv2.imread('square.jpg')
# 灰度化和二值化处理
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
ret, thresh = cv2.threshold(gray, 127, 255, cv2.THRESH_BINARY)
# 轮廓检测
contours, hierarchy = cv2.findContours(thresh, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
# 遍历所有轮廓
for cnt in contours:
# 近似为多边形
approx = cv2.approxPolyDP(cnt, 0.01 * cv2.arcLength(cnt, True), True)
# 判断是否为正方形
if len(approx) == 4:
x, y, w, h = cv2.boundingRect(cnt)
if abs(w-h) < 20:
cv2.rectangle(img, (x,y), (x+w,y+h), (0,255,0), 2)
# 显示图像
cv2.imshow('image', img)
cv2.waitKey(0)
cv2.destroyAllWindows()
```