print('[!] Loaded model')
print('[!] Press "q" to quit')
print('[.] Please get ready')
print('[!] Press "c" to continue')
while not keyevents.pressed_c:
pass
print('[!] Starting...')
keyevents.end = False
# Time used to display 'jump'
start_time = time.time()
start_time -= 32767
while not keyevents.end:
ann_input = eye.roi_to_grid().flatten()
img_preview = eye.img_preview
if clf.predict([ann_input])[0] == 1:
start_time = time.time()
mouse.click(eye._click_x, eye._click_y)
time.sleep(0.06)
# Shows text for 0.5 seconds
if time.time() - start_time < 0.5:
cv2.putText(img_preview, 'Jump!', (65, 65), font, 2, (0, 0, 0), 2)
cv2.imshow('preview', img_preview)
cv2.waitKey(1)
print('[X] Finished')
NN_INPUT = []
NN_OUTPUT = []
start_time = time.time()
while not keyevents.end:
# To prevent data from being too bias to not clicking
if time.time() - mouseevents.clicked_time > 0.08:
mouseevents.clicked = False
# 26 fps
if time.time() - start_time < 0.038:
continue
# Append to total data
NN_INPUT.append(eye.roi_to_grid().flatten())
NN_OUTPUT.append(1 if mouseevents.clicked else 0)
start_time = time.time()
# What filename to save as
file_outname = get_new_filename()
print('[S] Saving data...')
raw_data = {'input': NN_INPUT, 'output': NN_OUTPUT}
# Save our data
df = pd.DataFrame(raw_data, columns=['input', 'output'])
with open(file_outname, 'w') as f:
df.to_csv(f)
print('[X] Finished')
