初学者 Python 项目：使用 OpenCV 和 Mediapipe 构建增强现实绘图应用程序

学习知识要善于思考，思考，再思考！今天golang学习网小编就给大家带来《初学者 Python 项目：使用 OpenCV 和 Mediapipe 构建增强现实绘图应用程序》，以下内容主要包含等知识点，如果你正在学习或准备学习文章，就都不要错过本文啦~让我们一起来看看吧，能帮助到你就更好了！

本Python项目构建一个简单的增强现实(AR)绘图应用程序。利用摄像头和手势，您可以在屏幕上进行虚拟绘画，自定义画笔，甚至保存您的作品！

项目设置

首先，创建一个新文件夹，并使用以下命令初始化新的虚拟环境：

import cv2
import mediapipe as mp
import math

mp_hands = mp.solutions.hands
hands = mp_hands.Hands(min_detection_confidence=0.9, min_tracking_confidence=0.9)

draw_points = []
reset_drawing = False
brush_color = (0, 0, 255)
brush_size = 5

cap = cv2.VideoCapture(0)
while True:
    ret, frame = cap.read()
    if not ret:
        break

    frame = cv2.flip(frame, 1)
    frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
    results = hands.process(frame_rgb)

    if results.multi_hand_landmarks:
        for hand_landmarks in results.multi_hand_landmarks:
            h, w, _ = frame.shape
            cx, cy = int(hand_landmarks.landmark[mp_hands.HandLandmark.INDEX_FINGER_TIP].x * w), \
                     int(hand_landmarks.landmark[mp_hands.HandLandmark.INDEX_FINGER_TIP].y * h)
            mx, my = int(hand_landmarks.landmark[mp_hands.HandLandmark.MIDDLE_FINGER_TIP].x * w), \
                     int(hand_landmarks.landmark[mp_hands.HandLandmark.MIDDLE_FINGER_TIP].y * h)
            distance = math.sqrt((mx - cx) ** 2 + (my - cy) ** 2)
            threshold = 40

            if distance > threshold:
                if reset_drawing:
                    draw_points.append(None)
                    reset_drawing = False
                draw_points.append((cx, cy))
            else:
                reset_drawing = True

    for i in range(1, len(draw_points)):
        if draw_points[i - 1] and draw_points[i]:
            cv2.line(frame, draw_points[i - 1], draw_points[i], brush_color, brush_size)

    cv2.imshow('摄像头', frame)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()

步骤四：改进方向

• 使用OpenCV的cv2.rectangle()和cv2.putText()创建按钮来切换画笔大小和颜色。
• 添加保存图像的功能。
• 添加橡皮擦工具，使用新的坐标修改draw_points数组。

这个改进的版本提供了更完整的功能，并对代码进行了更清晰的组织和注释。 记住安装必要的库才能运行代码。

今天关于《初学者 Python 项目：使用 OpenCV 和 Mediapipe 构建增强现实绘图应用程序》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！