def main():
last_time = time.time()
for i in list(range(4))[::-1]:
print(i + 1)
time.sleep(1)
paused = False
mode_choice = 0
screen = grab_screen(region = (0, 40, GAME_WIDTH, GAME_HEIGHT + 40))
screen = cv2.cvtColor(screen, cv2.COLOR_BGR2RGB)
prev = cv2.resize(screen, (WIDTH,HEIGHT))
t_minus, t_now, t_plus = prev, prev, prev
while(True):
if not paused:
screen = grab_screen(region = (0, 40, GAME_WIDTH, GAME_HEIGHT + 40))
screen = cv2.cvtColor(screen, cv2.COLOR_BGR2RGB)
last_time = time.time()
screen = cv2.resize(screen, (WIDTH,HEIGHT))
delta_count_last = motion_detection(t_minus, t_now, t_plus)
t_minus = t_now
t_now = t_plus
t_plus = screen
t_plus = cv2.blur(t_plus, (4, 4))
prediction = model.predict([screen.reshape(WIDTH,HEIGHT, 3)])[0]
prediction = np.array(prediction) * np.array([4.5, 0.1, 0.1, 0.1, 1.8, 1.8, 0.5, 0.5, 0.2])
mode_choice = np.argmax(prediction)
if mode_choice == 0:
straight()
choice_picked = 'straight'
elif mode_choice == 1:
reverse()
choice_picked = 'reverse'
elif mode_choice == 2:
left()
choice_picked = 'left'
elif mode_choice == 3:
right()
choice_picked = 'right'
elif mode_choice == 4:
forward_left()
choice_picked = 'forward+left'
elif mode_choice == 5:
forward_right()
choice_picked = 'forward+right'
elif mode_choice == 6:
reverse_left()
choice_picked = 'reverse+left'
elif mode_choice == 7:
reverse_right()
choice_picked = 'reverse+right'
elif mode_choice == 8:
no_keys()
choice_picked = 'nokeys'
motion_log.append(delta_count)
motion_avg = round(mean(motion_log), 3)
print('loop took {0} seconds. Motion: {1}. Choice: {2}'.format(round(time.time() - last_time, 3), motion_avg, choice_picked))
if motion_avg < motion_req and len(motion_log) >= log_len:
print('WERE PROBABLY STUCK FFS, initiating some evasive maneuvers.')
# 0 = reverse straight, turn left out
# 1 = reverse straight, turn right out
# 2 = reverse left, turn right out
# 3 = reverse right, turn left out
quick_choice = random.randrange(0, 4)
if quick_choice == 0:
reverse()
time.sleep(random.uniform(1, 2))
forward_left()
time.sleep(random.uniform(1, 2))
elif quick_choice == 1:
reverse()
time.sleep(random.uniform(1, 2))
forward_right()
time.sleep(random.uniform(1, 2))
elif quick_choice == 2:
reverse_left()
time.sleep(random.uniform(1, 2))
forward_right()
time.sleep(random.uniform(1, 2))
elif quick_choice == 3:
reverse_right()
time.sleep(random.uniform(1, 2))
forward_left()
time.sleep(random.uniform(1, 2))
for i in range(log_len - 2):
del motion_log[0]
keys = key_check()
# p pauses game and can get annoying.
if 'T' in keys:
if paused:
paused = False
time.sleep(1)
else:
paused = True
ReleaseKey(A)
ReleaseKey(W)
ReleaseKey(D)
time.sleep(1)
def main():
last_time = time.time()
for i in list(range(4))[::-1]:
print(i + 1)
time.sleep(1)
paused = False
mode_choice = 0
screen = grab_screen(region=(0, 40, GAME_WIDTH, GAME_HEIGHT + 40))
screen = cv2.cvtColor(screen, cv2.COLOR_BGR2RGB)
prev = cv2.resize(screen, (WIDTH, HEIGHT))
t_minus = prev
t_now = prev
t_plus = prev
while (True):
if not paused:
screen = grab_screen(region=(0, 40, GAME_WIDTH, GAME_HEIGHT + 40))
screen = cv2.cvtColor(screen, cv2.COLOR_BGR2RGB)
last_time = time.time()
screen = cv2.resize(screen, (WIDTH, HEIGHT))
delta_count_last = motion_detection(t_minus, t_now, t_plus, screen)
t_minus = t_now
t_now = t_plus
t_plus = screen
t_plus = cv2.blur(t_plus, (4, 4))
prediction = model.predict([screen.reshape(WIDTH, HEIGHT, 3)])[0]
prediction = np.array(prediction) * np.array(
[4.5, 0.1, 0.1, 0.1, 1.8, 1.8, 0.5, 0.5, 0.2])
mode_choice = np.argmax(prediction)
if mode_choice == 0:
straight()
choice_picked = 'straight'
elif mode_choice == 1:
reverse()
choice_picked = 'reverse'
elif mode_choice == 2:
left()
choice_picked = 'left'
elif mode_choice == 3:
right()
choice_picked = 'right'
elif mode_choice == 4:
forward_left()
choice_picked = 'forward+left'
elif mode_choice == 5:
forward_right()
choice_picked = 'forward+right'
elif mode_choice == 6:
reverse_left()
choice_picked = 'reverse+left'
elif mode_choice == 7:
reverse_right()
choice_picked = 'reverse+right'
elif mode_choice == 8:
no_keys()
choice_picked = 'nokeys'
motion_log.append(delta_count)
motion_avg = round(mean(motion_log), 3)
print('loop took {} seconds. Motion: {}. Choice: {}'.format(
round(time.time() - last_time, 3), motion_avg, choice_picked))
if motion_avg < motion_req and len(motion_log) >= log_len:
print(
'WERE PROBABLY STUCK FFS, initiating some evasive maneuvers.'
)
# 0 = reverse straight, turn left out
# 1 = reverse straight, turn right out
# 2 = reverse left, turn right out
# 3 = reverse right, turn left out
quick_choice = random.randrange(0, 4)
if quick_choice == 0:
reverse()
time.sleep(random.uniform(1, 2))
forward_left()
time.sleep(random.uniform(1, 2))
elif quick_choice == 1:
reverse()
time.sleep(random.uniform(1, 2))
forward_right()
time.sleep(random.uniform(1, 2))
elif quick_choice == 2:
reverse_left()
time.sleep(random.uniform(1, 2))
forward_right()
time.sleep(random.uniform(1, 2))
elif quick_choice == 3:
reverse_right()
time.sleep(random.uniform(1, 2))
forward_left()
time.sleep(random.uniform(1, 2))
for i in range(log_len - 2):
del motion_log[0]
keys = key_check()
# p pauses game and can get annoying.
if 'T' in keys:
if paused:
paused = False
time.sleep(1)
else:
paused = True
ReleaseKey(A)
ReleaseKey(W)
ReleaseKey(D)
time.sleep(1)
def main():
last_time = time.time()
for i in list(range(4))[::-1]:
print(i + 1)
time.sleep(1)
paused = False
mode_choice = 0
screen = grab_screen(region=(0, 40, GAME_WIDTH, GAME_HEIGHT + 40))
screen = cv2.cvtColor(screen, cv2.COLOR_BGR2RGB)
prev = cv2.resize(screen, (WIDTH, HEIGHT))
t_minus = prev
t_now = prev
t_plus = prev
while (True):
if not paused:
screen = grab_screen(region=(0, 40, GAME_WIDTH, GAME_HEIGHT + 40))
screen = cv2.cvtColor(screen, cv2.COLOR_BGR2RGB)
last_time = time.time()
screen = cv2.resize(screen, (WIDTH, HEIGHT))
delta_count_last = motion_detection(t_minus, t_now, t_plus)
t_minus = t_now
t_now = t_plus
t_plus = screen
t_plus = cv2.blur(t_plus, (4, 4))
prediction = model.predict([screen.reshape(WIDTH, HEIGHT, 3)])[0]
prediction = np.array(prediction) * np.array(
[4.5, 0.1, 0.1, 0.1, 1.8, 1.8, 0.5, 0.5, 0.2])
mode_choice = np.argmax(prediction)
if mode_choice == 0:
straight()
choice_picked = '직진'
elif mode_choice == 1:
reverse()
choice_picked = '후진'
elif mode_choice == 2:
left()
choice_picked = '왼쪽'
elif mode_choice == 3:
right()
choice_picked = '오른쪽'
elif mode_choice == 4:
forward_left()
choice_picked = '앞+왼쪽'
elif mode_choice == 5:
forward_right()
choice_picked = '앞+오른쪽'
elif mode_choice == 6:
reverse_left()
choice_picked = '뒤+왼쪽'
elif mode_choice == 7:
reverse_right()
choice_picked = '뒤+오른쪽'
elif mode_choice == 8:
no_keys()
choice_picked = '없음'
motion_log.append(delta_count)
motion_avg = round(mean(motion_log), 3)
print('반복 {} sec. 키: {}. 선택: {}'.format(
round(time.time() - last_time, 3), motion_avg, choice_picked))
if motion_avg < motion_req and len(motion_log) >= log_len:
quick_choice = random.randrange(0, 4)
if quick_choice == 0:
reverse()
time.sleep(random.uniform(1, 2))
forward_left()
time.sleep(random.uniform(1, 2))
elif quick_choice == 1:
reverse()
time.sleep(random.uniform(1, 2))
forward_right()
time.sleep(random.uniform(1, 2))
elif quick_choice == 2:
reverse_left()
time.sleep(random.uniform(1, 2))
forward_right()
time.sleep(random.uniform(1, 2))
elif quick_choice == 3:
reverse_right()
time.sleep(random.uniform(1, 2))
forward_left()
time.sleep(random.uniform(1, 2))
for i in range(log_len - 2):
del motion_log[0]
keys = key_check()
# p > 정지
if 'T' in keys:
if paused:
paused = False
time.sleep(1)
else:
paused = True
ReleaseKey(A)
ReleaseKey(W)
ReleaseKey(D)
time.sleep(1)
