wip

main.py

@@ -3,63 +3,100 @@
 from time import perf_counter
 from os import walk
 from random import randint
+import numpy as np
 
 gaze = GazeTracking()
 webcam = cv2.VideoCapture(0)
 
-mypath = r'logo'
-filenames = list(walk(mypath))[0][2]
+time_to_show = 1  # sec
+
+dir_path = r'logo_new_scaled'
+filenames = list(walk(dir_path))[0][2]
+print(filenames)
 logos_time = [0] * len(filenames)
-time_old = perf_counter()
-photos_shows = []
-
-for i in range(5):
-    r1 = randint(0, len(filenames))
-    r2 = randint(0, len(filenames))
-    while True:
-        if r1 not in photos_shows:
-            photos_shows.append(r1)
-            break
-        r1 = randint(0, len(filenames))
-    while True:
-        if r2 not in photos_shows:
-            photos_shows.append(r2)
-            break
-        r2 = randint(0, len(filenames))
-
-    image1 = cv2.imread(filenames[r1])
-    image2 = cv2.imread(filenames[r2])
-    while True:
-        # We get a new frame from the webcam
-        _, frame = webcam.read()
-        # We send this frame to GazeTracking to analyze it
-        gaze.refresh(frame)
-        frame = gaze.annotated_frame()
-
-        pos = gaze.horizontal_ratio()
-        print(pos)
-
-        if pos is None:
-            # "Blinking"
-            pass
-        elif pos <= 0.6:
-            # "Looking right"
-            logos_time[r2] += perf_counter() - time_old
-            time_old = perf_counter()
-        elif pos > 0.6:
-            # "Looking left"
-            logos_time[r1] += perf_counter() - time_old
+logos_already_shows = []
+logos_pair = []
+
+for round in range(5):
+    r1 = 0
+    r2 = 0
+    for i in range(len(filenames)//2):
+        r1 = randint(0, len(filenames)-1)
+        r2 = randint(0, len(filenames)-1)
+        while True:
+            if r1 not in logos_already_shows:
+                logos_already_shows.append(r1)
+                break
+            r1 = randint(0, len(filenames)-1)
+
+        while True:
+            if r2 not in logos_already_shows:
+                logos_already_shows.append(r2)
+                break
+            r2 = randint(0, len(filenames)-1)
+
+        logos_pair.append((r1, r2))
+        print(r1, r2)
+        image1 = cv2.imread(f'{dir_path}/{filenames[r1]}')
+        image2 = cv2.imread(f'{dir_path}/{filenames[r2]}')
+
+        time_start_10_sec = perf_counter()
+        while perf_counter() - time_start_10_sec < time_to_show:
             time_old = perf_counter()
+            # show 2 logo
+            hori = np.concatenate((image1, image2), axis=1)
+
+            cv2.namedWindow("programs", cv2.WINDOW_NORMAL)
+            cv2.resizeWindow("programs", 1680, 1000)
+            cv2.imshow("programs", hori)
+            if cv2.waitKey(1) == 27:
+                break
+
+            # We get a new frame from the webcam
+            _, frame = webcam.read()
+            # We send this frame to GazeTracking to analyze it
+            gaze.refresh(frame)
+            frame = gaze.annotated_frame()
+
+            pos = gaze.horizontal_ratio()
+            # print(pos)
+
+            if pos is None:
+                # "Blinking"
+                pass
+            elif pos <= 0.6:
+                # "Looking right"
+                logos_time[r2] += perf_counter() - time_old
+                time_old = perf_counter()
+            elif pos > 0.6:
+                # "Looking left"
+                logos_time[r1] += perf_counter() - time_old
+                time_old = perf_counter()
+
+    print(logos_time)
+    # удалить логотипы с наименьшем временем
+    win_logos = []
+    for item in logos_pair:
+        if item[0] > item[1]:
+            win_logos.append(filenames[item[1]])
+        else:
+            win_logos.append(filenames[item[0]])
 
-        left_pupil = gaze.pupil_left_coords()
-        right_pupil = gaze.pupil_right_coords()
+    # print(win_logos)
+    filenames = win_logos
+    print(filenames)
+    logos_time = [0] * len(filenames)
+    logos_already_shows = []
+    logos_pair = []
 
-        cv2.namedWindow("programs", cv2.WINDOW_NORMAL)
-        cv2.resizeWindow("programs", 1680, 1000)
-        cv2.imshow("programs", frame)
+image = cv2.imread(f'{dir_path}/{filenames[0]}')
+cv2.namedWindow("programs", cv2.WINDOW_NORMAL)
+cv2.resizeWindow("programs", 1680, 1000)
+cv2.imshow("programs", image)
+while True:
+    if cv2.waitKey(1) == 27:
+        break
 
-        if cv2.waitKey(1) == 27:
-            break
 
-    webcam.release()
-    cv2.destroyAllWindows()
+webcam.release()
+cv2.destroyAllWindows()

scale_logos.py

@@ -0,0 +1,15 @@
+from PIL import Image, ImageGrab, ImageOps
+from os import walk
+
+
+dir_path = r'logo_new'
+filenames = list(walk(dir_path))[0][2]
+print(filenames)
+
+for i in range(len(filenames)):
+    img_orig = Image.open(f'logo_new/{filenames[i]}')
+    # впишем скриншот в квадрат 1000x1000 пикселей
+    # img = ImageOps.contain(img_orig, (900, 900), method=Image.LANCZOS)
+    img = ImageOps.pad(img_orig, (840, 840), color='#ffffff', method=Image.LANCZOS, centering=(0.5, 0.5))
+
+    img.save(f'logo_new_scaled/{filenames[i]}')