opencv期末练习题（3）附带解析

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

python 复制代码

import math

import cv2
import numpy as np
"""
1. 创建一个黑色画板
2. 输入q退出
3. 输入m切换画图模式
    两种模式，画矩形和画圆形。
    用户按住鼠标左键到一个位置然后释放就可以画出对应的图像
"""
px, py, rx, ry = 0, 0, 0, 0  # 临时保存左键按下时的x，y坐标

drawing = False
# 绑定窗体，窗体名为image
cv2.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, y
        if 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则退出程序
        break
    if k == ord('m'):  # 如果输入m则切换模型，修改drawing变量
        drawing = not drawing

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

以下是代码的主要步骤：

初始化全局变量：px, py, rx, ry 用于保存左键按下和释放时的坐标，drawing 用于标记当前绘制模式。
创建一个黑色画板：canvas = np.zeros((400, 400, 3))。
定义鼠标事件处理函数 draw_shape，在函数中根据事件类型和坐标信息绘制矩形或圆形。
设置鼠标事件回调函数：cv2.setMouseCallback('image', draw_shape)。
进入死循环，不断显示当前画板，并等待用户的键盘输入。
如果用户按下 'q' 键，则退出程序；如果按下 'm' 键，则切换绘制模式。

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