def main():
for i in range(3, 0, -1):
print(i)
time.sleep(1)
tracker = cv2.TrackerMOSSE_create()
game_frame = grab_screen()
bbox = cv2.selectROI("game_frame", game_frame, False)
tracker.init(game_frame, bbox)
while True:
game_frame = grab_screen()
lane = perspective_transform(game_frame)
success, bbox = tracker.update(game_frame)
if success:
x, y, w, h = int(bbox[0]), int(bbox[1]), int(bbox[2]), int(bbox[3])
pt1 = (x + int(w/2), y)
cv2.circle(game_frame, pt1, 3, (0, 0, 255), -1)
cv2.rectangle(game_frame, (x, y), (x+w, y+h), (0, 255, 0), 2)
cv2.imshow("game_frame", game_frame)
if cv2.waitKey(1) == 27:
break
cv2.destroyAllWindows()
def main():
image_counter = 4802
for i in range(5, 0, -1):
print(i)
time.sleep(1)
while True:
game_frame = grab_screen()
game_frame = cv2.resize(game_frame, (320, 160))
game_frame = cv2.cvtColor(game_frame, cv2.COLOR_BGR2GRAY)
val = (pygame.joystick.Joystick(0).get_axis(0))
image_path = DATADIR+str(image_counter)+".jpg"
cv2.imwrite(image_path, game_frame)
writer.writerow([image_path, val])
image_counter = image_counter + 1
cv2.imshow('game_frame', game_frame)
if cv2.waitKey(1) == 27:
break
f.close()
cv2.destroyAllWindows()
def main():
A_counter = 10674
W_counter = 21641
D_counter = 14518
for i in range(5, 0, -1):
print(i)
time.sleep(1)
while True:
game_frame = grab_screen()
game_frame = cv2.cvtColor(game_frame, cv2.COLOR_BGR2GRAY)
key_pressed = key_check()
if key_pressed == 'A':
cv2.imwrite(DATADIR + "A/" + str(A_counter) + ".jpg", game_frame)
A_counter = A_counter + 1
elif key_pressed == 'W':
cv2.imwrite(DATADIR + "W/" + str(W_counter) + ".jpg", game_frame)
W_counter = W_counter + 1
if key_pressed == 'D':
cv2.imwrite(DATADIR + "D/" + str(D_counter) + ".jpg", game_frame)
D_counter = D_counter + 1
else:
pass
cv2.imshow('game_frame', game_frame)
if cv2.waitKey(1) == 27:
break
cv2.destroyAllWindows()
def main():
for i in range(3, 0, -1):
print(i)
time.sleep(1)
while True:
game_frame = grab_screen()
cv2.imshow("game_frame", game_frame)
if cv2.waitKey(1) == 27:
break
cv2.destroyAllWindows()
def main():
try:
for i in range(3, 0, -1):
print(i)
time.sleep(1)
while True:
game_frame = grab_screen()
game_frame = cv2.cvtColor(game_frame, cv2.COLOR_BGR2GRAY)
game_frame = cv2.resize(game_frame, (100, 100))
prediction = model.predict(game_frame.reshape(-1, 100, 100, 1))
print(np.argmax(prediction[0]))
if cv2.waitKey(1) == 27:
break
except KeyboardInterrupt:
pass
def get_screen():
screen = grab_screen(region=(0, 40, 1310, 835))
screen = cv2.resize(screen, (320, 180))
screen = cv2.cvtColor(screen, cv2.COLOR_BGR2RGB)
return screen