当前热点-在 Python 中使用 OpenCV 构建 Color Catcher 游戏

维科号   2023-05-19 11:40:58

介绍


【资料图】

你是否曾经想在 Python 中使用 OpenCV 创建自己的游戏?

今天我们将构建一个名为 Color Catcher 的游戏,该游戏挑战玩家使用手部跟踪机制接住从屏幕顶部掉落的彩球。

设置游戏窗口

构建游戏的第一步是使用 OpenCV 设置游戏窗口。我们将定义窗口大小、创建窗口并设置其在屏幕上的位置:

#Setupthegamewindow

window_size=(640,480)

window_name="ColorCatcher"

cv2.namedWindow(window_name)

cv2.moveWindow(window_name,0,0)

定义游戏对象

接下来,我们将定义游戏对象。在 Color Catcher 中,我们有两个主要的游戏对象:捕手和球。

捕手是玩家用手移动的矩形,而球是从屏幕顶部落下的随机生成的圆圈。我们将定义这些游戏对象的属性:

#Setupthegameobjects

catcher_color=(0,0,255)

catcher_width=100

catcher_height=20

catcher_position=np.array([window_size[0]//2,window_size[1]-catcher_height],dtype=int)

catcher_velocity=np.array([10,0],dtype=int)

ball_radius=20

ball_speed=5

ball_colors=[(0,255,0),(255,0,0),(0,0,255)]

balls=[]

score=0

从网络摄像头捕获视频

为了跟踪玩家的手部动作,我们需要使用 OpenCV 从网络摄像头捕获视频。

我们将创建一个视频捕获设备并循环播放视频的每一帧:

#Setupthevideocapturedevice

cap=cv2.VideoCapture(0)

whileTrue:

#Readaframefromthevideocapturedevice

ret,frame=cap.read()

ifnotret:

break

检测玩家的手

为了跟踪玩家的手部动作,我们将使用 OpenCV 的轮廓检测功能。首先,我们将每个帧转换为灰度并应用阈值以便更容易检测轮廓:

#Converttheframetograyscaleandapplyathreshold

gray=cv2.cvtColor(frame,cv2.COLOR_BGR2GRAY)

_,thresh=cv2.threshold(gray,0,255,cv2.THRESH_BINARY+cv2.THRESH_OTSU)

接下来,我们将在阈值图像中找到轮廓,并确定面积最大的轮廓,这应该是玩家的手:

#Findthecontoursinthethresholdedimage

contours,_=cv2.findContours(thresh,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)

#Findthecontourwiththelargestarea,whichshouldbethehand

ifcontours:

hand_contour=max(contours,key=cv2.contourArea)

hand_hull=cv2.convexHull(hand_contour)

hand_center=np.mean(hand_contour,axis=0,dtype=int)[0]

移动捕手

一旦我们检测到玩家手的位置,我们就可以相应地移动捕手。

在代码中,我们检查手中心的 x 坐标并将其与接球手位置的 x 坐标进行比较。如果手在接球手的左侧,我们通过从接球手的当前位置减去接球手的速度,将接球手向左移动。如果手在捕手的右侧,我们通过将捕手的速度添加到其当前位置来将捕手移动到右侧。

ifhand_center[0]0:

catcher_position-=catcher_velocity

elifhand_center[0]>catcher_position[0]+catcher_widthandcatcher_position[0]+catcher_width

catcher_position+=catcher_velocity

生成和移动球:

如果当前比赛中的球数少于五个,我们将生成具有随机颜色和位置的新球。我们将包含球的颜色、位置和速度的元组附加到balls列表中。

iflen(balls)<5:

ball_color=random.choice(ball_colors)

ball_position=np.array([random.randint(ball_radius,window_size[0]-ball_radius),0],dtype=int)

ball_velocity=np.array([0,ball_speed],dtype=int)

balls.append((ball_color,ball_position,ball_velocity))

然后我们遍历balls列表中的每个球,通过将其速度添加到其当前位置来更新其位置,并检查它是否与接球手发生碰撞或击中游戏窗口的底部。

如果球与接球手发生碰撞,我们将其从balls列表中移除,增加玩家的得分,然后跳出循环。

如果球击中了游戏窗口的底部,我们将其从balls列表中移除并跳出循环。

foriinrange(len(balls)):

balls[i]=(balls[i][0],balls[i][1]+balls[i][2],balls[i][2])

ball_position=balls[i][1]

ifball_position[1]+ball_radius>=catcher_position[1]and

ball_position[0]>=catcher_position[0]and

ball_position[0]<=catcher_position[0]+catcher_width:

balls.pop(i)

score+=1

break

elifball_position[1]+ball_radius>=window_size[1]:

balls.pop(i)

break

绘制游戏对象:

最后,我们使用cv2.rectangle()和cv2.circle()函数在框架上绘制游戏对象。我们使用np.zeros()创建一个黑色框架,将捕手绘制为红色矩形,并将每个球绘制为彩色圆圈。

我们还使用cv2.putText()函数在框架的左上角显示玩家的分数。

frame=np.zeros(window_size+(3,),dtype=np.uint8)

cv2.rectangle(frame,tuple(catcher_position),tuple(catcher_position+np.array([catcher_width,catcher_height])),catcher_color,-1)

forballinballs:

cv2.circle(frame,tuple(ball[1]),ball_radius,ball[0],-1)

cv2.putText(frame,"Score:{}".format(score),(10,30),cv2.FONT_HERSHEY_SIMPLEX,1,(255,255,255),2)

cv2.imshow(window_name,frame)

退出游戏:

最后,游戏循环包含在while循环中。在这个循环中,执行以下任务:

使用cap.read()方法从视频捕获设备读取新帧。处理框架以检测手区域并相应地移动捕手。如有必要,将生成一个具有随机颜色和位置的新球。球被移动并检查是否与接球手发生碰撞。游戏对象绘制在框架上。框架显示在屏幕上。循环继续,直到用户按下“q”键。

按“q”键可以退出游戏。这是使用cv2.waitKey()方法完成的,该方法等待键盘事件的给定毫秒数。如果按键被按下,该方法返回按键的 ASCII 代码,如果没有按键被按下,则返回 -1。

我们使用按位与运算符 ( &) 提取结果的最低有效 8 位,这为我们提供了按 256 模的按下键的 ASCII 代码。我们将此值与 "q" 键 ( ord("q")) 的 ASCII 代码进行比较,如果匹配则退出循环。

#Exitthegameiftheuserpressesthe"q"key

ifcv2.waitKey(1)&0xFF==ord("q"):

break

退出循环后,我们分别使用cap.release()和cv2.destroyAllWindows()方法释放视频捕获设备并关闭游戏窗口。

#Releasethevideocapturedeviceandclosethegamewindow

cap.release()

cv2.destroyAllWindows()

就是这样!你现在应该能够运行代码并玩游戏了。这个游戏是一个简单的例子,说明了如何使用计算机视觉来实时控制游戏对象的移动。当然,还有很大的改进和优化空间,但这应该足以让你入门。

编码愉快!

完整代码:

https://github.com/Yaga987/Computer-Vision/blob/main/CompVisGame.py