28.05.24 Создание скриншотов для последующего анализа алгоритмами машинного обучения

import os
import cv2
import datetime
import time
 
class utils:
 
    def __init__(self):
 
        #папка, в которой мы запуситили нашу программу
        mDir = os.path.dirname(os.path.realpath('__file__'))
        print("mDir=", mDir)        
        #папка Data, то есть мы запустили файл в некой папке, и в ней находится папка data
        self.DATA_DIR = os.path.join(mDir, "data")
        #папка Photo, то есть мы запустили файл в некой папке, и в ней неходится папка photo
        self.PHOTO_DIR = os.path.join(mDir, "photo") 
        #папка внутри папки photo с названием orig
        self.PHOTO_DIR_ORIG = os.path.join(self.PHOTO_DIR, "orig")
        #папка внутри папки photo с названием compr
        self.PHOTO_DIR_COMPR = os.path.join(self.PHOTO_DIR, "compr"  )
 
        #если этих папок нет, то мы автоматически их создаём
        for d in [self.DATA_DIR, self.PHOTO_DIR, self.PHOTO_DIR_ORIG, self.PHOTO_DIR_COMPR]:
            if not os.path.exists(d):
                os.mkdir(d)
 
    def give_data_dir(self):
 
        return self.DATA_DIR
 
    def give_photo_dir(self):
 
        return self.PHOTO_DIR
 
    def give_photo_dir_orig(self):
 
        return self.PHOTO_DIR_ORIG
 
    def give_photo_dir_compr(self):
 
        return self.PHOTO_DIR_COMPR
 
    #сохраняем наш файл на диск   
    def save_photo_disk(self, frame, path):
        
        print("Start")
        now = datetime.datetime.now()
 
        mstr = str(now)
        nstr = ""
        count = 0
        
        for m in mstr:
            if m == ":":
                nstr += "_"
            else:
                nstr += m
 
            if "." in nstr:
                count += 1
            if count ==3:
                break
 
        #print("mstr=", mstr)
        #print("nstr=", nstr)
 
        nstr += ".jpg"
 
        new_name = os.path.join(path,nstr)
        print("path=", path)
        print("nstr=", nstr)
        cv2.imwrite(new_name, frame)

import numpy as np
#вот эта библиотека и позволит нам снимать скриншоты нашео экрана
import pyautogui
import imutils
import cv2
 
from utils import utils
 
import time
 
mDir = utils()
 
while True:
    #снимает скриншотв экрана
    image = pyautogui.screenshot()
    #сохраняем в более привычный массив через библиотеку OpenCV
    image = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
    #сохраняем наш файл на диск
    mDir.save_photo_disk(image, mDir.give_photo_dir_orig())
    print(mDir.give_photo_dir_orig())
    #Берём паузу, чем больше она будет, тем реже будет делать снимки нашего экране
    #Чем меньше паузу, тем чаще
    time.sleep(300)

model.fit(X_train, y_train, batch_size = 200, nb_epoch = 100, validation_split = 0.2, verbose = 1)

from keras.callbacks import ReduceLROnPlateau

learning_rate_reduction = ReduceLROnPlateau(monitor='val_acc', 
                                            patience=3, 
                                            verbose=1, 
                                            factor=0.5, 
                                            min_lr=0.00001)