pyautogui.press('0')
print('FRAME NUM: {}'.format(index))
print('TOTALS ----- HOT {0}, NOT HOT {1}'.format(hot, not_hot))
def label(dir_path, img_arr):
for index, img in enumerate(img_arr):
frame = cv2.imread('{}{}'.format(dir_path, img))
cv2.imshow('FRAME', np.array(frame))
keyboard.on_press_key('h',
lambda _: post_press('h', index),
suppress=True)
keyboard.on_press_key('n',
lambda _: post_press('n', index),
suppress=True)
cv2.waitKey(0)
if __name__ == '__main__':
# sort images
processing = Processing()
processing.order_images(r'../images/')
processing.total_images()
# apply labels
hot = 0
not_hot = 0
label(r'../images/', processing.sorted_images)
