百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术分类 > 正文

使用Python+OpenCV代码制作贪吃蛇小游戏,你也可以打造自己的AI

ztj100 2024-10-29 18:22 39 浏览 0 评论

前言

今天和大家分享一下如何使用 mediapipe+opencv 自制贪吃蛇小游戏。先放张图看效果。

规则:食指指尖控制蛇头,指尖每接触到黄色方块,计数加一,蛇身变长,方块随机切换位置。如果指尖停止移动,或者移动过程中蛇头撞到蛇身,那么游戏结束。点击键盘上的R键重新开始游戏。

游戏进行时:

游戏结束界面:

1. 安装工具包

pip install opencv_python==4.2.0.34  # 安装opencv
pip install mediapipe  # 安装mediapipe
# pip install mediapipe --user  #有user报错的话试试这个
pip install cvzone  # 安装cvzone
 
# 导入工具包
import cv2
import cvzone
import numpy as np
from cvzone.HandTrackingModule import HandDetector  # 导入手部检测模块
import math
import random

21个手部关键点信息如下,本节我们主要研究食指指尖'8'的坐标(x,y)信息。

2. 检测手部关键点

(1)cvzone.HandTrackingModule.HandDetector()是手部关键点检测方法

参数:

mode: 默认为 False,将输入图像视为视频流。它将尝试在第一个输入图像中检测手,并在成功检测后进一步定位手的坐标。在随后的图像中,一旦检测到所有 maxHands 手并定位了相应的手的坐标,它就会跟踪这些坐标,而不会调用另一个检测,直到它失去对任何一只手的跟踪。这减少了延迟,非常适合处理视频帧。如果设置为 True,则在每个输入图像上运行手部检测,用于处理一批静态的、可能不相关的图像。

maxHands: 最多检测几只手,默认为 2

detectionCon: 手部检测模型的最小值(0-1之间),超过阈值则检测成功。默认为 0.5

minTrackingCon: 坐标跟踪模型的最小置信值 (0-1之间),用于将手部坐标视为成功跟踪,不成功则在下一个输入图像上自动调用手部检测。将其设置为更高的值可以提高解决方案的稳健性,但代价是更高的延迟。如果 mode 为 True,则忽略这个参数,手部检测将在每个图像上运行。默认为 0.5

它的参数和返回值类似于官方函数 mediapipe.solutions.hands.Hands()

MULTI_HAND_LANDMARKS: 被检测/跟踪的手的集合,其中每只手被表示为21个手部地标的列表,每个地标由x, y, z组成。x和y分别由图像的宽度和高度归一化为[0,1]。Z表示地标深度。

MULTI_HANDEDNESS: 被检测/追踪的手是左手还是右手的集合。每只手由label(标签)和score(分数)组成。 label 是 'Left' 或 'Right' 值得字符串。 score 是预测左右手的估计概率。

(2)cvzone.HandTrackingModule.HandDetector.findHands()找到手部关键点并绘图

参数:

img: 需要检测关键点的帧图像,格式为BGR

draw: 是否需要在原图像上绘制关键点及识别框

flipType: 图像是否需要翻转,当视频图像和我们自己不是镜像关系时,设为True就可以了

返回值:

hands: 检测到的手部信息,由0或1或2个字典组成的列表。如果检测到两只手就是由两个字典组成的列表。字典中包含:21个关键点坐标(x,y,z),检测框左上坐标及其宽高,检测框中心点坐标,检测出是哪一只手。

img: 返回绘制了关键点及连线后的图像

代码如下:

import cv2
import cvzone
import numpy as np
from cvzone.HandTrackingModule import HandDetector  # 导入手部检测模块
 
#(1)获取摄像头
cap = cv2.VideoCapture(0) # 0代表电脑自带的摄像头
# 设置显示窗口的size
cap.set(3, 1280)  # 窗口宽1280
cap.set(4, 720)   # 窗口高720
 
#(2)模型配置
detector = HandDetector(maxHands=1,  # 最多检测1只手
                        detectionCon=0.8)  # 最小检测置信度0.8
 
#(3)图像处理
while True:
 
    # 每次读取一帧相机图像,返回是否读取成功success,读取的帧图像img
    success, img = cap.read()
 
    # 图像翻转,使图像和自己呈镜像关系
    img = cv2.flip(img, 1)  # 0代表上下翻转,1代表左右翻转
 
    # 检测手部关键点。返回手部信息hands,绘制关键点后的图像img
    hands, img = detector.findHands(img, flipType=False)  # 由于上一行翻转过图像了,这里就不用翻转了
 
    # 查看关键点信息
    print(hands)
 
    #(4)关键点处理
    if hands:  # 如果检测到手了,那就处理关键点
 
        # 获得食指指尖坐标(x,y)
        hand = hands[0]  # 获取一只手的全部信息
        lmList = hand['lmList']  # 获得这只手的21个关键点的坐标(x,y,z)
        pointIndex = lmList[8][0:2]  # 只获取食指指尖关键点的(x,y)坐标
 
        # 以食指指尖为圆心画圈(圆心坐标是元组类型),半径为15,青色填充
        cv2.circle(img, tuple(pointIndex), 15, (255,0,0), cv2.FILLED)
 
    #(5)显示图像
    cv2.imshow('img', img)  # 输入图像显示窗口的名称及图像
     # 每帧滞留1毫秒后消失,并且按下ESC键退出
    if cv2.waitKey(1) & 0xFF == 27:
        break
 
# 释放视频资源
cap.release()
cv2.destroyAllWindows()

效果图如下:

打印手部关键点信息如下:

[{'lmList': [[1152, 675, 0], [1085, 693, -37], [1030, 698, -68], [1003, 698, -97], [1003, 679, -122], [1001, 511, -48], [1041, 546, -81], [1093, 608, -102], [1134, 652, -110], [1075, 484, -46], [1119, 534, -84], [1171, 605, -101], [1217, 659, -103], [1141, 481, -45], [1177, 529, -83], [1219, 590, -84], [1253, 642, -73], [1195, 494, -47], [1221, 521, -73], [1245, 566, -65], [1267, 602, -49]], 
'bbox': (1001, 481, 266, 217),
'center': (1134, 589), 
'type': 'Right'}]

3. 蛇身移动

构造一个处理蛇身移动的类,要求在没吃食物时,蛇身保持固定的长度跟随食指指尖移动。

举个例子,如果当前的蛇身节点列表 self.points 包含 [a, b, c, d] 这四个节点,a节点代表蛇尾,d节点代表蛇头。在下一帧,食指指尖移动到 e 点,将 e 节点追加到蛇身节点列表中,那么现在的列表包含 [a, b, c, d, e] 节点,其中 e 节点为新的蛇头。

此时判断当前蛇身总长度 self.currentLength(列表中所有节点之间的长度之和)是否大于蛇身固定长度 self.allowedLength,保证在移动过程中蛇身长度不变。

如果当前蛇身总长度 self.currentLength 大于固定长度 self.allowedLength,那么在节点列表中从尾到头依次删除节点,列表 [a, b, c, d, e] 中 a 表示蛇尾节点,先删除,判断列表 [b, c, d, e] 的节点之间的总长度是否满足要求。若仍大于固定长度,那么就再删除 b 节点,再判断。

如果当前蛇身总长度 self.currentLength 小于固定长度 self.allowedLength,那么就不做任何处理。

在上述代码中补充:

import cv2
import cvzone
import numpy as np
from cvzone.HandTrackingModule import HandDetector  # 导入手部检测模块
import math
 
# 构造一个贪吃蛇移动的类
class SnakeGameClass:
 
    #(一)初始化
    def __init__(self):
 
        self.points = []  # 蛇的身体的节点坐标
        self.lengths = []  # 蛇身各个节点之间的坐标
        self.currentLength = 0  # 当前蛇身长度
        self.allowedLength = 150  # 没吃东西时,蛇的总长度
        self.previousHead = (0,0)  # 前一个蛇头节点的坐标
 
    #(二)更新增加蛇身长度
    def update(self, imgMain, currentHead): # 输入图像,当前蛇头的坐标
 
        px, py = self.previousHead  # 获得前一个蛇头的x和y坐标
        cx, cy = currentHead  # 当前蛇头节点的x和y坐标
        
        # 添加当前蛇头的坐标到蛇身节点坐标列表中
        self.points.append([cx,cy])
 
        # 计算两个节点之间的距离
        distance = math.hypot(cx-px, cy-py)  # 计算平方和开根
        # 将节点之间的距离添加到蛇身节点距离列表中
        self.lengths.append(distance)
        # 增加当前蛇身长度
        self.currentLength += distance
 
        # 更新蛇头坐标
        self.previousHead = (cx,cy)
 
        #(三)减少蛇尾长度,即移动过程中蛇头到蛇尾的长度不大于150
        if self.currentLength > self.allowedLength:
 
            # 遍历所有的节点线段长度。新更新的蛇头索引在列表后面,蛇尾的索引在列表前面
            for i, length in enumerate(self.lengths):
 
                # 从蛇尾到蛇头依次减线段长度,得到的长度是否满足要求
                self.currentLength -= length
 
                # 从列表中删除蛇尾端的线段长度,以及蛇尾节点
                self.lengths.pop(i)
                self.points.pop(i)
 
                # 如果当前蛇身长度小于规定长度,满足要求,退出循环
                if self.currentLength < self.allowedLength:
                    break
 
        #(四)绘制蛇
        # 当节点列表中有值了,才能绘制
        if self.points:
 
            # 遍历蛇身节点坐标
            for i, point in enumerate(self.points):  
                # 绘制前后两个节点之间的连线
                if i != 0:
                    cv2.line(imgMain, tuple(self.points[i-1]), tuple(self.points[i]), (0,255,0), 20)
 
            # 在蛇头的位置画个圆
            cv2.circle(imgMain, tuple(self.points[-1]), 20, (255,0,0), cv2.FILLED)
 
        # 返回更新后的图像
        return imgMain
 
 
#(1)获取摄像头
cap = cv2.VideoCapture(0) # 0代表电脑自带的摄像头
# 设置显示窗口的size
cap.set(3, 1280)  # 窗口宽1280
cap.set(4, 720)   # 窗口高720
 
#(2)模型配置
detector = HandDetector(maxHands=1,  # 最多检测1只手
                        detectionCon=0.8)  # 最小检测置信度0.8
 
# 接收创建贪吃蛇的类
game = SnakeGameClass()
 
#(3)图像处理
while True:
 
    # 每次读取一帧相机图像,返回是否读取成功success,读取的帧图像img
    success, img = cap.read()
 
    # 图像翻转,使图像和自己呈镜像关系
    img = cv2.flip(img, 1)  # 0代表上下翻转,1代表左右翻转
 
    # 检测手部关键点。返回手部信息hands,绘制关键点后的图像img
    hands, img = detector.findHands(img, flipType=False)  # 由于上一行翻转过图像了,这里就不用翻转了
 
    # 查看关键点信息
    print(hands)
 
    #(4)关键点处理
    if hands:  # 如果检测到手了,那就处理关键点
 
        # 获得食指指尖坐标(x,y)
        hand = hands[0]  # 获取一只手的全部信息
        lmList = hand['lmList']  # 获得这只手的21个关键点的坐标(x,y,z)
        pointIndex = lmList[8][0:2]  # 只获取食指指尖关键点的(x,y)坐标
 
        # 更新贪吃蛇的节点,给出蛇头节点坐标。返回更新后的图像
        img = game.update(img, pointIndex)
 
    #(5)显示图像
    cv2.imshow('img', img)  # 输入图像显示窗口的名称及图像
     # 每帧滞留1毫秒后消失,并且按下ESC键退出
    if cv2.waitKey(1) & 0xFF == 27:
        break
 
# 释放视频资源
cap.release()
cv2.destroyAllWindows()

效果图如下,蛇身保持默认固定长度随着指尖而移动。

4. 蛇进食增加身体长度

先看下面代码 SnakeGameClass 类中的第(五)步。给食物(即绘制的矩形)随机给出一个中心点坐标,自定义的类方法 randomFoodLocation(),执行该方法则食物的中心点坐标的x在[100,1000]中随机取一个数,y在[100,600]中随机取一个数。

下面代码定义的类中的第(七)步。判断食指指尖(即蛇头节点坐标)是否在矩形内部,如果在内部,那么蛇身移动过程中的固定长度 self.allowedLength 增加50个像素值。得分 self.score 加一。并在下一帧随机改变食物的位置。

在上述代码中补充:

import cv2
import cvzone
import numpy as np
from cvzone.HandTrackingModule import HandDetector  # 导入手部检测模块
import math
import random
 
# 构造一个贪吃蛇移动的类
class SnakeGameClass:
 
    #(一)初始化
    def __init__(self):
 
        self.score = 0  # 积分器
        self.points = []  # 蛇的身体的节点坐标
        self.lengths = []  # 蛇身各个节点之间的坐标
        self.currentLength = 0  # 当前蛇身长度
        self.allowedLength = 150  # 没吃东西时,蛇的总长度
        self.previousHead = (0,0)  # 前一个蛇头节点的坐标
 
        self.foodPoint = (0,0)  # 食物的起始位置
        self.randomFoodLocation()  # 随机改变食物的位置
 
    #(五)食物随机出现的位置
    def randomFoodLocation(self):
        # x在100至1000之间,y在100至600之间,随机取一个整数
        self.foodPoint = random.randint(100, 1000),  random.randint(100, 600)
 
    #(二)更新增加蛇身长度
    def update(self, imgMain, currentHead): # 输入图像,当前蛇头的坐标
 
        px, py = self.previousHead  # 获得前一个蛇头的x和y坐标
        cx, cy = currentHead  # 当前蛇头节点的x和y坐标
        
        # 添加当前蛇头的坐标到蛇身节点坐标列表中
        self.points.append([cx,cy])
 
        # 计算两个节点之间的距离
        distance = math.hypot(cx-px, cy-py)  # 计算平方和开根
        # 将节点之间的距离添加到蛇身节点距离列表中
        self.lengths.append(distance)
        # 增加当前蛇身长度
        self.currentLength += distance
 
        # 更新蛇头坐标
        self.previousHead = (cx,cy)
 
        #(三)减少蛇尾长度,即移动过程中蛇头到蛇尾的长度不大于150
        if self.currentLength > self.allowedLength:
 
            # 遍历所有的节点线段长度。新更新的蛇头索引在列表后面,蛇尾的索引在列表前面
            for i, length in enumerate(self.lengths):
 
                # 从蛇尾到蛇头依次减线段长度,得到的长度是否满足要求
                self.currentLength -= length
 
                # 从列表中删除蛇尾端的线段长度,以及蛇尾节点
                self.lengths.pop(i)
                self.points.pop(i)
 
                # 如果当前蛇身长度小于规定长度,满足要求,退出循环
                if self.currentLength < self.allowedLength:
                    break
        
        #(七)检查蛇是否吃了食物
        rx, ry = self.foodPoint  # 得到食物的中心点坐标位置
        
        # 绘制矩形作为蛇的食物
        cv2.rectangle(imgMain, (rx-20, ry-20), (rx+20, ry+20), (255,255,0), cv2.FILLED)
        cv2.rectangle(imgMain, (rx-20, ry-20), (rx+20, ry+20), (0,255,255), 5)        
        cv2.rectangle(imgMain, (rx-5, ry-5), (rx+5, ry+5), (0,0,255), cv2.FILLED)  
 
        # 检查指尖(即蛇头cx,cy)是否在矩形内部
        if rx-20 < cx < rx+20 and ry-20< cy < ry+20:
 
            # 随机更换食物的位置
            self.randomFoodLocation()
 
            # 增加蛇身的限制长度,每吃1个食物就能变长50
            self.allowedLength += 50
 
            # 吃食物的计数加一
            self.score += 1
 
            print('eat!', f'score:{self.score}')
 
        #(四)绘制蛇
        # 当节点列表中有值了,才能绘制
        if self.points:
 
            # 遍历蛇身节点坐标
            for i, point in enumerate(self.points):  
                # 绘制前后两个节点之间的连线
                if i != 0:
                    cv2.line(imgMain, tuple(self.points[i-1]), tuple(self.points[i]), (0,255,0), 20)
                    cv2.line(imgMain, tuple(self.points[i-1]), tuple(self.points[i]), (0,0,255), 15)
 
            # 在蛇头的位置画个圆
            cv2.circle(imgMain, tuple(self.points[-1]), 20, (255,255,0), cv2.FILLED)
            cv2.circle(imgMain, tuple(self.points[-1]), 18, (255,0,0), 3)            
            cv2.circle(imgMain, tuple(self.points[-1]), 5, (0,0,0), cv2.FILLED)
 
        # 返回更新后的图像
        return imgMain
 
 
#(1)获取摄像头
cap = cv2.VideoCapture(0) # 0代表电脑自带的摄像头
# 设置显示窗口的size
cap.set(3, 1280)  # 窗口宽1280
cap.set(4, 720)   # 窗口高720
 
#(2)模型配置
detector = HandDetector(maxHands=1,  # 最多检测1只手
                        detectionCon=0.8)  # 最小检测置信度0.8
 
# 接收创建贪吃蛇的类
game = SnakeGameClass()
 
#(3)图像处理
while True:
 
    # 每次读取一帧相机图像,返回是否读取成功success,读取的帧图像img
    success, img = cap.read()
 
    # 图像翻转,使图像和自己呈镜像关系
    img = cv2.flip(img, 1)  # 0代表上下翻转,1代表左右翻转
 
    # 检测手部关键点。返回手部信息hands,绘制关键点后的图像img
    hands, img = detector.findHands(img, flipType=False)  # 由于上一行翻转过图像了,这里就不用翻转了
 
    #(4)关键点处理
    if hands:  # 如果检测到手了,那就处理关键点
 
        # 获得食指指尖坐标(x,y)
        hand = hands[0]  # 获取一只手的全部信息
        lmList = hand['lmList']  # 获得这只手的21个关键点的坐标(x,y,z)
        pointIndex = lmList[8][0:2]  # 只获取食指指尖关键点的(x,y)坐标
 
        # 更新贪吃蛇的节点,给出蛇头节点坐标。返回更新后的图像
        img = game.update(img, pointIndex)
 
    #(5)显示图像
    cv2.imshow('img', img)  # 输入图像显示窗口的名称及图像
     # 每帧滞留1毫秒后消失,并且按下ESC键退出
    if cv2.waitKey(1) & 0xFF == 27:
        break
 
# 释放视频资源
cap.release()
cv2.destroyAllWindows()

效果图如下:

5. 自身碰撞及界面的处理

先看到自定义类 SnakeGameClass 中的第(八)步,蛇身节点列表 self.points 中包含从蛇头到蛇尾的所有节点。如 [a, b, c, d, e] 节点,a 节点代表蛇尾,e 节点代表蛇头。这里我就粗糙地判断一下是否碰撞,如果大家有更好的判断方法可以改动这第(八)步。计算蛇头 e 节点到所有节点之间的距离,如果小于某个值就代表碰撞了,游戏结束 self.gameover = True

再看到自定义类中的第(三)步。如果游戏结束 self.gameover = True,那就在下一帧中绘制结算界面,不再执行蛇身移动程序。

再看到主程序中的第(5)步。其中 k == ord('r'),按下键盘上的 r 键来重新游戏。将所有蛇身变量初始化,并将 self.gameover = False,退出结算界面,使得下一帧能执行蛇身移动操作。

在上述代码中补充:

import cv2
import cvzone
from matplotlib.cbook import pts_to_midstep
import numpy as np
from cvzone.HandTrackingModule import HandDetector  # 导入手部检测模块
import math
import random
 
# 构造一个贪吃蛇移动的类
class SnakeGameClass:
 
    #(一)初始化
    def __init__(self):
 
        self.score = 0  # 积分器
        self.points = []  # 蛇的身体的节点坐标
        self.lengths = []  # 蛇身各个节点之间的坐标
        self.currentLength = 0  # 当前蛇身长度
        self.allowedLength = 150  # 没吃东西时,蛇的总长度
        self.previousHead = (0,0)  # 前一个蛇头节点的坐标
 
        self.foodPoint = (0,0)  # 食物的起始位置
        self.randomFoodLocation()  # 随机改变食物的位置
 
        self.gameover = False  # 蛇头撞到蛇身,变成True,游戏结束
 
    #(二)食物随机出现的位置
    def randomFoodLocation(self):
        # x在100至1000之间,y在100至600之间,随机取一个整数
        self.foodPoint = random.randint(100, 1000),  random.randint(100, 600)
 
    #(三)更新增加蛇身长度
    def update(self, imgMain, currentHead): # 输入图像,当前蛇头的坐标
 
        # 游戏结束,显示文本
        if self.gameover:
            cvzone.putTextRect(imgMain, 'GameOver', [400,300], 5, 3, colorR=(0,255,255), colorT=(0,0,255))
            cvzone.putTextRect(imgMain, f'Score:{self.score}', [450,400], 5, 3, colorR=(255,255,0))
            cvzone.putTextRect(imgMain, f"Press Key 'R' to Restart", [230,500], 4, 3, colorR=(0,255,0), colorT=(255,0,0))
 
        else:
            px, py = self.previousHead  # 获得前一个蛇头的x和y坐标
            cx, cy = currentHead  # 当前蛇头节点的x和y坐标
            
            # 添加当前蛇头的坐标到蛇身节点坐标列表中
            self.points.append([cx,cy])
 
            # 计算两个节点之间的距离
            distance = math.hypot(cx-px, cy-py)  # 计算平方和开根
            # 将节点之间的距离添加到蛇身节点距离列表中
            self.lengths.append(distance)
            # 增加当前蛇身长度
            self.currentLength += distance
 
            # 更新蛇头坐标
            self.previousHead = (cx,cy)
 
            #(四)减少蛇尾长度,即移动过程中蛇头到蛇尾的长度不大于150
            if self.currentLength > self.allowedLength:
 
                # 遍历所有的节点线段长度。新更新的蛇头索引在列表后面,蛇尾的索引在列表前面
                for i, length in enumerate(self.lengths):
 
                    # 从蛇尾到蛇头依次减线段长度,得到的长度是否满足要求
                    self.currentLength -= length
 
                    # 从列表中删除蛇尾端的线段长度,以及蛇尾节点
                    self.lengths.pop(i)
                    self.points.pop(i)
 
                    # 如果当前蛇身长度小于规定长度,满足要求,退出循环
                    if self.currentLength < self.allowedLength:
                        break
            
            #(五)绘制得分板
            cvzone.putTextRect(imgMain, f'Score:{self.score}', [50,80], 4, 3, colorR=(255,255,0))
 
            #(六)检查蛇是否吃了食物
            rx, ry = self.foodPoint  # 得到食物的中心点坐标位置
            
            # 绘制矩形作为蛇的食物
            cv2.rectangle(imgMain, (rx-20, ry-20), (rx+20, ry+20), (255,255,0), cv2.FILLED)
            cv2.rectangle(imgMain, (rx-20, ry-20), (rx+20, ry+20), (0,255,255), 5)        
            cv2.rectangle(imgMain, (rx-5, ry-5), (rx+5, ry+5), (0,0,255), cv2.FILLED)  
 
            # 检查指尖(即蛇头cx,cy)是否在矩形内部
            if rx-20 < cx < rx+20 and ry-20< cy < ry+20:
 
                # 随机更换食物的位置
                self.randomFoodLocation()
 
                # 增加蛇身的限制长度,每吃1个食物就能变长50
                self.allowedLength += 50
 
                # 吃食物的计数加一
                self.score += 1
 
                print('eat!', f'score:{self.score}')
 
            #(七)绘制蛇
            # 当节点列表中有值了,才能绘制
            if self.points:
 
                # 遍历蛇身节点坐标
                for i, point in enumerate(self.points):  
                    # 绘制前后两个节点之间的连线
                    if i != 0:
                        cv2.line(imgMain, tuple(self.points[i-1]), tuple(self.points[i]), (0,255,0), 20)
                        cv2.line(imgMain, tuple(self.points[i-1]), tuple(self.points[i]), (0,0,255), 15)
 
                # 在蛇头的位置画个圆
                cv2.circle(imgMain, tuple(self.points[-1]), 20, (255,255,0), cv2.FILLED)
                cv2.circle(imgMain, tuple(self.points[-1]), 18, (255,0,0), 3)            
                cv2.circle(imgMain, tuple(self.points[-1]), 5, (0,0,0), cv2.FILLED)
 
            #(八)检查蛇头碰撞到自身        
            for point in self.points[:-2]:  # 不算蛇头到自身的距离
 
                # 计算蛇头和每个节点之间的距离
                dist = math.hypot(cx-point[0], cy-point[1])
 
                # 如果距离小于1.8,那么就证明碰撞了
                if dist < 1.8:
 
                    # 游戏结束
                    self.gameover = True
 
        # 返回更新后的图像
        return imgMain
 
 
#(1)获取摄像头
cap = cv2.VideoCapture(0) # 0代表电脑自带的摄像头
# 设置显示窗口的size
cap.set(3, 1280)  # 窗口宽1280
cap.set(4, 720)   # 窗口高720
 
#(2)模型配置
detector = HandDetector(maxHands=1,  # 最多检测1只手
                        detectionCon=0.8)  # 最小检测置信度0.8
 
# 接收创建贪吃蛇的类
game = SnakeGameClass()
 
#(3)图像处理
while True:
 
    # 每次读取一帧相机图像,返回是否读取成功success,读取的帧图像img
    success, img = cap.read()
 
    # 图像翻转,使图像和自己呈镜像关系
    img = cv2.flip(img, 1)  # 0代表上下翻转,1代表左右翻转
 
    # 检测手部关键点。返回手部信息hands,绘制关键点后的图像img
    hands, img = detector.findHands(img, flipType=False)  # 由于上一行翻转过图像了,这里就不用翻转了
 
    #(4)关键点处理
    if hands:  # 如果检测到手了,那就处理关键点
 
        # 获得食指指尖坐标(x,y)
        hand = hands[0]  # 获取一只手的全部信息
        lmList = hand['lmList']  # 获得这只手的21个关键点的坐标(x,y,z)
        pointIndex = lmList[8][0:2]  # 只获取食指指尖关键点的(x,y)坐标
 
        # 更新贪吃蛇的节点,给出蛇头节点坐标。返回更新后的图像
        img = game.update(img, pointIndex)
 
    #(5)显示图像
    cv2.imshow('img', img)  # 输入图像显示窗口的名称及图像
 
    # 重新开始游戏
    k = cv2.waitKey(1)  # 每帧滞留1毫秒后消失
    if k == ord('r'):  # 键盘'r'键代表重新开始游戏
        game.gameover = False
        game.score = 0  # 积分器
        game.points = []  # 蛇的身体的节点坐标
        game.lengths = []  # 蛇身各个节点之间的坐标
        game.currentLength = 0  # 当前蛇身长度
        game.allowedLength = 150  # 没吃东西时,蛇的总长度
        game.previousHead = (0,0)  # 前一个蛇头节点的坐标                
        game.randomFoodLocation()  # 随机改变食物的位置
    
    if k & 0xFF == 27:  # 键盘ESC键退出程序
        break
 
# 释放视频资源
cap.release()
cv2.destroyAllWindows()

效果图如下,在移动过程中,蛇头每碰到一个食物,蛇身就会变长,如果 停止移动 或 蛇头节点距离蛇身节点过近 就会结束游戏。

相关推荐

离谱!写了5年Vue,还不会自动化测试?

前言大家好,我是倔强青铜三。是一名热情的软件工程师,我热衷于分享和传播IT技术,致力于通过我的知识和技能推动技术交流与创新,欢迎关注我,微信公众号:倔强青铜三。Playwright是一个功能强大的端到...

package.json 与 package-lock.json 的关系

模块化开发在前端越来越流行,使用node和npm可以很方便的下载管理项目所需的依赖模块。package.json用来描述项目及项目所依赖的模块信息。那package-lock.json和...

Github 标星35k 的 SpringBoot整合acvtiviti开源分享,看完献上膝盖

前言activiti是目前比较流行的工作流框架,但是activiti学起来还是费劲,还是有点难度的,如何整合在线编辑器,如何和业务表单绑定,如何和系统权限绑定,这些问题都是要考虑到的,不是说纯粹的把a...

Vue3 + TypeScript 前端研发模板仓库

我们把这个Vue3+TypeScript前端研发模板仓库的初始化脚本一次性补全到可直接运行的状态,包括:完整的目录结构所有配置文件研发规范文档示例功能模块(ExampleFeature)...

Vue 2迁移Vue 3:从响应式到性能优化

小伙伴们注意啦!Vue2已经在2023年底正式停止维护,再不升级就要面临安全漏洞没人管的风险啦!而且Vue3带来的性能提升可不是一点点——渲染速度快40%,内存占用少一半,更新速度直接翻倍!还在...

VUE学习笔记:声明式渲染详解,对比WEB与VUE

声明式渲染是指使用简洁的模板语法,声明式的方式将数据渲染进DOM系统。声明式是相对于编程式而言,声明式是面向对象的,告诉框架做什么,具体操作由框架完成。编程式是面向过程思想,需要手动编写代码完成具...

苏州web前端培训班, 苏州哪里有web前端工程师培训

前端+HTML5德学习内容:第一阶段:前端页面重构:PC端网站布局、HTML5+CSS3基础项目、WebAPP页面布局;第二阶段:高级程序设计:原生交互功能开发、面向对象开发与ES5/ES6、工具库...

跟我一起开发微信小程序——扩展组件的代码提示补全

用户自定义代码块步骤:1.HBuilderX中工具栏:工具-代码块设置-vue代码块2.通过“1”步骤打开设置文件...

JimuReport 积木报表 v1.9.3发布,免费可视化报表

项目介绍积木报表JimuReport,是一款免费的数据可视化报表,含报表、大屏和仪表盘,像搭建积木一样完全在线设计!功能涵盖:数据报表、打印设计、图表报表、门户设计、大屏设计等!...

软开企服开源的无忧企业文档(V2.1.3)产品说明书

目录1....

一款面向 AI 的下一代富文本编辑器,已开源

简介AiEditor是一个面向AI的下一代富文本编辑器。开箱即用、支持所有前端框架、支持Markdown书写模式什么是AiEditor?AiEditor是一个面向AI的下一代富文本编辑...

玩转Markdown(2)——抽象语法树的提取与操纵

上一篇玩转Markdown——数据的分离存储与组件的原生渲染发布,转眼已经鸽了大半年了。最近在操纵mdast生成md文件的时候,心血来潮,把玩转Markdown(2)给补上了。...

DeepseekR1+ollama+dify1.0.0搭建企业/个人知识库(入门避坑版)

找了网上的视频和相关文档看了之后,可能由于版本不对或文档格式不对,很容易走弯路,看完这一章,可以让你少踩三天的坑。步骤和注意事项我一一列出来:1,前提条件是在你的电脑上已配置好ollama,dify1...

升级JDK17的理由,核心是降低GC时间

升级前后对比升级方法...

一个vsCode格式化插件_vscode格式化插件缩进量

ESlint...

取消回复欢迎 发表评论: