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