python 检测四边形顶点 代码

以下是Python代码，可以检测四边形顶点并将其可视化。

import cv2
import numpy as np

image = cv2.imread('image.png')
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

# Canny边缘检测
edges = cv2.Canny(gray, 50, 150, apertureSize=3)

# 轮廓检测
contours, hierarchy = cv2.findContours(edges, cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE)

for contour in contours:
    approx = cv2.approxPolyDP(contour, 0.01 * cv2.arcLength(contour, True), True)
    # 如果轮廓有四个点，我们将其视为四边形
    if len(approx) == 4:
        # 画出四边形
        cv2.drawContours(image, [approx], 0, (0, 0, 255), 2)
        # 画出每个顶点
        for point in approx:
            x, y = point[0]
            cv2.circle(image, (x, y), 5, (0, 255, 0), -1)

cv2.imshow('image', image)
cv2.waitKey(0)
cv2.destroyAllWindows()

在代码中，我们首先加载图像并将其转换为灰度图像。然后，我们使用Canny边缘检测算法来查找图像中的边缘。接下来，我们使用findContours函数查找图像中的所有轮廓。

对于每个找到的轮廓，我们使用approxPolyDP函数来近似该轮廓。如果该轮廓近似为四个点，我们将其视为四边形。然后，我们在图像中绘制该四边形，并使用圆圈表示每个四边形的顶点。

最后，我们显示包含四边形和顶点的图像并等待按键。