一、视频流人脸识别系统的系统函数的构建(Python)(system.py)

本文链接：https://blog.csdn.net/mahoon411/article/details/106182219

1. 代码部分

此代码块为system.py内的代码

# -*- coding: utf-8 -*-
"""
Created on Fri Mar 15 11:23:52 2019

@author: Administrator
"""
import tkinter as tk
import os
from build_face_dataset import build_face_dataset
from extract_embeddings import extract_embeddings
from train_model import train_model
from recognize_video import recognize_video


os.chdir(r'E:\python_project\face_recognize\half_change_build_v6_note')

win = tk.Tk()
win.title('基于深度学习的人脸识别系统')
win.geometry('310x200')
tk.Label(win, text = '人脸识别系统v6').place(x=190, y=180)

# 登记窗口函数
def register():
    def yes():
        # 创建文件夹 
        os.makedirs("E:/python_project/face_recognize/half_change_build_v6_note/dataset/" + entry_name.get())
        # 创建人脸数据库
        build_face_dataset(entry_name.get())
        # 关闭窗口
        win_reg.destroy()
    # 重输    
    def reenter():
        name.set('')
        ID.set('')
        
    win_reg = tk.Toplevel()
    win_reg.title('信息登记')
    win_reg.geometry('320x200')
    
    name = tk.StringVar()
    tk.Label(win_reg, text = '姓名: ').place(x = 10, y = 10)
    entry_name = tk.Entry(win_reg, textvariable = name)
    entry_name.place(x = 150, y = 10)

    ID = tk.StringVar()
    tk.Label(win_reg, text = '证件号码: ').place(x = 10, y = 110)
    entry_usr_pwd = tk.Entry(win_reg, textvariable=ID)
    entry_usr_pwd.place(x = 150, y = 110)        
    
    tk.Label(win_reg, text = '证件类型: ').place(x = 10,  y = 50)
    var = tk.StringVar()
    var.set(('居民身份证', '护照'))
    lb = tk.Listbox(win_reg, listvariable = var, height = 2)
    lb.place(x = 150, y = 50)
    
    tk.Label(win_reg, text = '人脸识别系统v6').place(x=180, y=180)

    
    tk.Button(win_reg, text = '确定', command = yes).place(x = 67, y = 150)
    
    tk.Button(win_reg, text = '重输', command = reenter).place(x = 243, y = 150)
    
# 识别窗口函数
def recognize():
    # 提取128维特征向量 
    extract_embeddings()    
 
    # 训练SVM（支持向量机）
    train_model()

    # 人脸识别
	# 必须先关闭主窗口，然后此函数内的人脸识别窗口才可以正常显示
    win.destroy()
    recognize_video()   
        
tk.Button(win, text = '信息登记', height = 2, width = 15, 
          command = register).place(x = 45, y = 30)

tk.Button(win, text = '人员识别', height = 2, width = 15, 
          command = recognize).place(x = 45, y = 100)


win.mainloop()