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()
2. 文件结构图
3. 程序流程图
4. 小结
本博客是整个视频流人脸识别系统的第一大部分:系统函数的构建。其余部分包括:构建人脸数据库、提取人脸的面部嵌入、训练SVM分类器、视频流人脸识别的实现。其余博客将在更新后发出。
另外,本系统中用到的一些文件的下载地址为:人脸识别集成包