用python编写一个绘制棋盘并用Harris()函数进行角点检测的代码

当然，我可以帮你回答这个问题！

首先，我们需要导入必要的库和模块：

import cv2
import numpy as np

然后，我们可以定义棋盘的大小和每个方格的大小（这里假设是8x8的棋盘）：

board_size = (8,8)
square_size = 50 # 50 pixels

接下来，我们可以创建一个棋盘图像的矩阵，并用cv2.rectangle函数绘制每个方格：

board_width = board_size[0] * square_size
board_height = board_size[1] * square_size

board = np.zeros((board_height, board_width), dtype=np.uint8)

for i in range(board_size[0]):
    for j in range(board_size[1]):
        if (i+j) % 2 == 0:
            color = 255
        else:
            color = 0
        x1 = i * square_size
        y1 = j * square_size
        x2 = (i+1) * square_size
        y2 = (j+1) * square_size
        cv2.rectangle(board, (x1, y1), (x2, y2), color, -1)

现在我们已经创建了棋盘图像。接下来，我们可以使用Harris函数进行角点检测。首先，我们需要将图像转换为灰度，并使用cv2.cornerHarris函数计算角点响应：

gray_board = cv2.cvtColor(board, cv2.COLOR_BGR2GRAY)

dst = cv2.cornerHarris(gray_board, 2, 3, 0.04)

然后，我们可以定义一个阈值来找到角点，并使用cv2.findContours函数找到这些角点的位置：

threshold = 0.1 * dst.max()

board[dst > threshold] = [0, 0, 255]

_, contours, hierarchy = cv2.findContours(board, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)

最后，我们可以在图像上绘制找到的角点：

board_with_corners = board.copy()

for cnt in contours:
    x,y,w,h = cv2.boundingRect(cnt)
    cv2.rectangle(board_with_corners,(x,y),(x+w,y+h),(0,255,0),2)

完整的代码如下：

import cv2
import numpy as np

board_size = (8,8)
square_size = 50 # 50 pixels

board_width = board_size[0] * square_size
board_height = board_size[1] * square_size

board = np.zeros((board_height, board_width), dtype=np.uint8)

for i in range(board_size[0]):
    for j in range(board_size[1]):
        if (i+j) % 2 == 0:
            color = 255
        else:
            color = 0
        x1 = i * square_size
        y1 = j * square_size
        x2 = (i+1) * square_size
        y2 = (j+1) * square_size
        cv2.rectangle(board, (x1, y1), (x2, y2), color, -1)

gray_board = cv2.cvtColor(board, cv2.COLOR_BGR2GRAY)

dst = cv2.cornerHarris(gray_board, 2, 3, 0.04)

threshold = 0.1 * dst.max()

board[dst > threshold] = [0, 0, 255]

_, contours, hierarchy = cv2.findContours(board, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)

board_with_corners = board.copy()

for cnt in contours:
    x,y,w,h = cv2.boundingRect(cnt)
    cv2.rectangle(board_with_corners,(x,y),(x+w,y+h),(0,255,0),2)

cv2.imshow("board", board)
cv2.imshow("board with corners", board_with_corners)
cv2.waitKey(0)

这个代码应该可以实现你想要的功能。如果有任何问题，请告诉我。