创建黑色画板，并支持两种画图功能

import mathimport cv2import numpy as np"""1. 创建一个黑色画板2. 输入q退出3. 输入m切换画图模式两种模式，画矩形和画圆形。用户按住鼠标左键到一个位置然后释放就可以画出对应的图像"""px, py, rx, ry = 0, 0, 0, 0# 临时保存左键按下时的x，y坐标drawing = False# 绑定窗体，窗体名为imagecv2.namedWindow("image")canvas = np.zeros((400, 400, 3))# 创建黑色画板def draw_shape(event, x, y, flags, param):global px, py, rx, ry, canvas, draw_line# 设置全局变量，这样才能够访问方法体外部的变量if not drawing:if event == cv2.EVENT_LBUTTONDOWN:# 判断当前是否为按下左键px, py = x, y# 保存按下左键的坐标if event == cv2.EVENT_LBUTTONUP:rx, ry = x, y# 保存释放左键的坐标cv2.rectangle(canvas, [px, py], [rx, ry], color=(255, 255, 255))# 根据坐标信息，画出一个黑色的矩形else:if event == cv2.EVENT_LBUTTONDOWN:# 如果用户按下了m键，则切换到画圆形模式px, py = x, yif event == cv2.EVENT_LBUTTONUP:# 计算当前释放点和上次点击的圆心距离多远dis = math.sqrt((x - px) ** 2 + (y - py) ** 2)# 两点之间距离公式cv2.circle(canvas, center=(px, py), radius=int(dis), color=(0, 0, 255))# 根据离圆心的距离画出圆cv2.setMouseCallback('image', draw_shape)# 设置opencv鼠标事件while 1:# 死循环等待用户cv2.imshow("image", canvas)# 显示当前的画板数据k = cv2.waitKey(10)# 等待10ms，判断用户的键盘数据if k == ord("q"):# 如果输入q则退出程序breakif k == ord('m'):# 如果输入m则切换模型，修改drawing变量drawing = not drawing

这段代码使用 OpenCV 创建了一个简单的画板应用，支持矩形和圆形的绘制。用户可以按下左键拖动以绘制矩形，按下右键拖动以绘制圆形。按下 ‘q’ 键退出程序，按下 ‘m’ 键切换绘制模式（矩形或圆形）。

以下是代码的主要步骤：

  1. 初始化全局变量：px, py, rx, ry 用于保存左键按下和释放时的坐标，drawing 用于标记当前绘制模式。

  2. 创建一个黑色画板：canvas = np.zeros((400, 400, 3))。

  3. 定义鼠标事件处理函数 draw_shape，在函数中根据事件类型和坐标信息绘制矩形或圆形。

  4. 设置鼠标事件回调函数：cv2.setMouseCallback('image', draw_shape)。

  5. 进入死循环，不断显示当前画板，并等待用户的键盘输入。

  6. 如果用户按下 ‘q’ 键，则退出程序；如果按下 ‘m’ 键，则切换绘制模式。

这个简单的画板演示了如何使用 OpenCV 处理鼠标事件来实现基本的图形绘制功能。