在Python中将Tkinter Canvas与眼睛检测集成

时间:2019-03-10 19:07:20

标签: python opencv canvas tkinter

我正在做一个关于眼睛检测的项目。我使用tkinter和眼睛检测代码创建了一个4象限的画布。当我运行代码时,将打开两个窗口。一个是我的Canvas窗口,另一个是我的摄像头检测我的视线的框架。我想以某种方式将我的视线与画布整合在一起,这样,如果我凝视一个象限一段时间,就会执行一些操作,例如象限的颜色变化。

附加的是我的眼睛检测代码,其中我调用了包含我的tkinter Canvas窗口的“ test.py ”脚本。 Test.py代码在下面进一步附件。

import cv2
import numpy as np
import dlib
from math import hypot
import os

command = "test.py"
os.system(command)

cap = cv2.VideoCapture(0)

detector = dlib.get_frontal_face_detector()
predictor = dlib.shape_predictor("shape_predictor_68_face_landmarks.dat")

def midpoint(p1 ,p2):
    return int((p1.x + p2.x)/2), int((p1.y + p2.y)/2)

font = cv2.FONT_HERSHEY_PLAIN


def get_blinking_ratio(eye_points, facial_landmarks):
    left_point = (facial_landmarks.part(eye_points[0]).x, facial_landmarks.part(eye_points[0]).y)
    right_point = (facial_landmarks.part(eye_points[3]).x, facial_landmarks.part(eye_points[3]).y)
    center_top = midpoint(facial_landmarks.part(eye_points[1]), facial_landmarks.part(eye_points[2]))
    center_bottom = midpoint(facial_landmarks.part(eye_points[5]), facial_landmarks.part(eye_points[4]))

hor_line_lenght = hypot((left_point[0] - right_point[0]), (left_point[1] - right_point[1]))
ver_line_lenght = hypot((center_top[0] - center_bottom[0]), (center_top[1] - center_bottom[1]))

ratio = hor_line_lenght / ver_line_lenght
return ratio


def get_gaze_ratio(eye_points, facial_landmarks):
    left_eye_region = np.array([(facial_landmarks.part(eye_points[0]).x, facial_landmarks.part(eye_points[0]).y),
                                (facial_landmarks.part(eye_points[1]).x, facial_landmarks.part(eye_points[1]).y),
                                (facial_landmarks.part(eye_points[2]).x, facial_landmarks.part(eye_points[2]).y),
                                (facial_landmarks.part(eye_points[3]).x, facial_landmarks.part(eye_points[3]).y),
                                (facial_landmarks.part(eye_points[4]).x, facial_landmarks.part(eye_points[4]).y),
                                (facial_landmarks.part(eye_points[5]).x, facial_landmarks.part(eye_points[5]).y)], np.int32)
    #cv2.polylines(frame, [left_eye_region], True, (0, 0, 255), 2)


height, width, _ = frame.shape
mask = np.zeros((height, width), np.uint8)
cv2.polylines(mask, [left_eye_region], True, 255, 2)
cv2.fillPoly(mask, [left_eye_region], 255)
eye = cv2.bitwise_and(gray, gray, mask=mask)

min_x = np.min(left_eye_region[:, 0])
max_x = np.max(left_eye_region[:, 0])
min_y = np.min(left_eye_region[:, 1])
max_y = np.max(left_eye_region[:, 1])

gray_eye = eye[min_y: max_y, min_x: max_x]
_, threshold_eye = cv2.threshold(gray_eye, 70, 255, cv2.THRESH_BINARY)
height, width = threshold_eye.shape
left_side_threshold = threshold_eye[0: height, 0: int(width / 2)]
left_side_white = cv2.countNonZero(left_side_threshold)

right_side_threshold = threshold_eye[0: height, int(width / 2): width]
right_side_white = cv2.countNonZero(right_side_threshold)

if left_side_white == 0:
    gaze_ratio = 1
elif right_side_white == 0:
    gaze_ratio = 4
else:
    gaze_ratio = left_side_white / right_side_white
return gaze_ratio


while True:
    _, frame = cap.read()
    new_frame = np.zeros((500, 500, 3), np.uint8)
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)


faces = detector(gray)
for face in faces:

    landmarks = predictor(gray, face)

    """
        # Detect blinking
    left_eye_ratio = get_blinking_ratio([36, 37, 38, 39, 40, 41], landmarks)
    right_eye_ratio = get_blinking_ratio([42, 43, 44, 45, 46, 47], landmarks)
    blinking_ratio = (left_eye_ratio + right_eye_ratio) / 2

    if blinking_ratio > 5.7:
        cv2.putText(frame, "BLINKING", (50, 170), font, 5, (255, 0, 0))
        print("Blinking")
    """


    # Gaze detection
    gaze_ratio_left_eye = get_gaze_ratio([36, 37, 38, 39, 40, 41], landmarks)
    gaze_ratio_right_eye = get_gaze_ratio([42, 43, 44, 45, 46, 47], landmarks)
    gaze_ratio = (gaze_ratio_right_eye + gaze_ratio_left_eye) / 2

    cv2.putText(frame, str(gaze_ratio), (50, 100), font, 2, (0, 0, 255), 3)


    if gaze_ratio <= 0:
        cv2.putText(frame, "RIGHT", (50, 230), font, 2, (0, 0, 255), 3)
        print("RIGHT")
        new_frame[:] = (0, 0, 255)
    elif 0 <= gaze_ratio <= 1:
        cv2.putText(frame, "CENTRE", (50, 230), font, 2, (0, 0, 255), 3)
        print("CENTRE")
        new_frame[:] = (0, 0, 0)

    else:
        new_frame[:] = (255, 0, 0)
        cv2.putText(frame, "LEFT", (50, 230), font, 2, (0, 0, 255), 3)
        print("LEFT")





cv2.imshow("Frame", frame) 
#cv2.imshow("New frame", new_frame)

key = cv2.waitKey(1)
if key == 27:
    break


cap.release()
cv2.destroyAllWindows()

以下是test.py的代码

import tkinter as tk
import time

root = tk.Tk()
Canvas = tk.Canvas(root, width = 400, height=400)
Canvas.pack()
first = Canvas.create_rectangle(0,0,200,200, fill ='lightgrey')
second = Canvas.create_rectangle(200,0,400,200, fill ='lightgrey')
third = Canvas.create_rectangle(0,200,200,400, fill ='lightgrey')
fourth = Canvas.create_rectangle(200,200,400,400, fill ='lightgrey')
Canvas.create_text(100,100, text="Drinking", font = "Times 25 bold") 
Canvas.create_text(300,100, text="Food", font = "Times 25 bold")
Canvas.create_text(100,300, text="Sleep", font = "Times 25 bold")
Canvas.create_text(300,300, text="Washroom", font = "Times 25 bold")
root.mainloop()

请指导我。预先感谢您!

0 个答案:

没有答案