-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
108 lines (85 loc) · 3.64 KB
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
import cv2
import cvzone
import random
from cvzone.HandTrackingModule import HandDetector
import time
cap = cv2.VideoCapture(0)
# scale down and crop it from size
cap.set(3,640)
cap.set(4,480)
detector = HandDetector(maxHands=1)
timer = 0
stateResult = False
startGame = False
scores = [0,0] # [AI,player]
print("Welcome to Rock Paper Scissors")
print("Press s to play game and press b to exit match\n")
while True:
imgBG = cv2.imread("Resources/BG.png")
success,img = cap.read()
imgScaled = cv2.resize(img,(0,0),None,0.875,0.875) # 420/480
imgScaled = imgScaled[:, 80:480] # height:width
# find hands
hands, img = detector.findHands(imgScaled) # with draw
if startGame:
if stateResult is False:
timer = time.time()-initialTime # initial time is when key s is pressed
cv2.putText(imgBG,str(int(timer)),(605,435),cv2.FONT_HERSHEY_PLAIN,6,(255,0,255),4)
# 6 is font and 4 is thickness. () is the position
if timer>3:
stateResult = True
timer = 0
if hands:
playerMove=None
hand = hands[0]
fingers = detector.fingersUp(hand)
if fingers == [0,0,0,0,0]:
playerMove = 1 # rock
if fingers == [1,1,1,1,1]:
playerMove = 2 # paper
if fingers == [0,1,1,0,0]:
playerMove = 3 # scissor
randomNumber = random.randint(1,3)
imgAI=cv2.imread(f'Resources/{randomNumber}.png',cv2.IMREAD_UNCHANGED) # alpha value should be unchanged
imgBG = cvzone.overlayPNG(imgBG,imgAI,(149,310))
# Player Wins
if(playerMove == 1 and randomNumber == 3) or \
(playerMove == 2 and randomNumber == 1) or \
(playerMove == 3 and randomNumber == 2):
scores[1] += 1
# AI wins
if (playerMove == 3 and randomNumber == 1) or \
(playerMove == 1 and randomNumber == 2) or \
(playerMove == 2 and randomNumber == 3):
scores[0] += 1
if (playerMove == 1): print("Rock")
elif (playerMove == 2): print("Paper")
elif (playerMove == 3): print("Scissor")
else: print("Invalid Sign")
# print(fingers)
# display an array with 0 and 1 of size 5 [thumb,index,middle,ring,pinky]
imgBG[234:654, 795:1195] = imgScaled
if stateResult:
imgBG = cvzone.overlayPNG(imgBG, imgAI, (149, 310))
cv2.putText(imgBG, str(scores[0]), (410, 215), cv2.FONT_HERSHEY_PLAIN, 4, (255, 255, 255), 6)
cv2.putText(imgBG, str(scores[1]), (1112, 215), cv2.FONT_HERSHEY_PLAIN, 4, (255, 255, 255), 6)
# cv2.imshow("Image",img)
cv2.imshow("BG",imgBG)
# cv2.imshow("Scaled",imgScaled)
key = cv2.waitKey(1)
if key == ord('s'):
startGame = True
initialTime = time.time()
stateResult = False
# press b to break out
if key == ord('b'):
print("\nFinal Score")
print(str(scores[0]) + "-" + str(scores[1]))
if scores[0] > scores[1]:
print("AI won")
elif scores[0] < scores[1]:
print("You won")
else:
print("It's a Tie")
break
# we can make AI win as per symbol of user (cheating)