我给公司写了一款基于Python的人脸签到系统!给了奖金五千!

本文实例为大家分享了python实现人脸签到系统的具体代码,供大家参考,具体内容如下

简易版人脸签到/签退系统

管理员可进行录入人脸操作,以及导出各类签到情况表;

普通学生只可人脸识别进行签到签退操作。

face_Recognizer.py

import cv2
import os
import numpy as np
from PIL import Image #pillow
import pyttsx3
import sys
import test.student_sql as connect #导入py
import test.recognizer_sql as baseConnect #导入py
import time
import json

def makeDir(engine):
  flag= 0
  if not os.path.exists("face_trainer"):
    print("创建预训练环境")
    engine.say('检测到第一次启动,未检测到环境,正在创建环境')
    engine.say('正在创建预训练环境')
    os.mkdir("face_trainer")
    engine.say('创建成功')
    engine.runAndWait()
    flag=1
  if not os.path.exists("Facedata"):
    print("创建训练环境")
    engine.say('正在创建训练环境')
    os.mkdir("Facedata")
    engine.say('创建成功')
    engine.runAndWait()
    flag=1
  return flag

def getFace(cap,path_id):
  # 调用笔记本内置摄像头,所以参数为0,如果有其他的摄像头可以调整参数为1,2
  #cap = cv2.VideoCapture(0)
  #xml文件为自己的文件路径
  face_detector = cv2.CascadeClassifier(r'F:\npyWorkspace\venv\Lib\site-packages\cv2\data\haarcascade_frontalface_default.xml')
  #face_id = input('\n enter user id:')
  print('\n Initializing face capture. Look at the camera and wait ...')
  count = 0
  while True:
    # 从摄像头读取图片
    sucess, img = cap.read()
    # 转为灰度图片
    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    # 检测人脸
    faces = face_detector.detectMultiScale(gray, 1.3, 5)
    for (x, y, w, h) in faces:
      cv2.rectangle(img, (x, y), (x+w, y+w), (255, 0, 0))
      count += 1
      # 保存图像
      cv2.imwrite("Facedata/User." + str(path_id) + '.' + str(count) + '.jpg', gray[y: y + h, x: x + w])
      cv2.imshow('image', img)
    # 保持画面的持续。
    k = cv2.waitKey(1)
    if k == 27:  # 通过esc键退出摄像
      break
    elif count >= 100: # 得到1000个样本后退出摄像
      break
  cv2.destroyAllWindows()

def getImagesAndLabels(path, detector):
  imagePaths = [os.path.join(path, f) for f in os.listdir(path)] # join函数的作用
  faceSamples = []
  ids = []
  for imagePath in imagePaths:
    PIL_img = Image.open(imagePath).convert('L') # convert it to grayscale
    img_numpy = np.array(PIL_img, 'uint8')
    id = int(os.path.split(imagePath)[-1].split(".")[1])
    faces = detector.detectMultiScale(img_numpy)
    for (x, y, w, h) in faces:
      faceSamples.append(img_numpy[y:y + h, x: x + w])
      ids.append(id)
  return faceSamples, ids

def trainFace():
  # 人脸数据路径
  path = 'Facedata'
  recognizer = cv2.face.LBPHFaceRecognizer_create()
  detector = cv2.CascadeClassifier(r'F:\npyWorkspace\venv\Lib\site-packages\cv2\data\haarcascade_frontalface_default.xml')
  print('Training faces. It will take a few seconds. Wait ...')
  faces, ids = getImagesAndLabels(path, detector)
  recognizer.train(faces, np.array(ids))
  recognizer.write(r'face_trainer\trainer.yml')
  print("{0} faces trained. Exiting Program".format(len(np.unique(ids))))

def checkFace(cam,names,engine,sign_flag):
  sex = {"female":"女士","male":"先生"}
  recognizer = cv2.face.LBPHFaceRecognizer_create()
  recognizer.read('face_trainer/trainer.yml')
  cascadePath = r"F:\npyWorkspace\venv\Lib\site-packages\cv2\data\haarcascade_frontalface_default.xml"
  faceCascade = cv2.CascadeClassifier(cascadePath)
  font = cv2.FONT_HERSHEY_SIMPLEX
  idnum = 0
  names = ['yumengzhen', 'dujuanjuan','litingting','kangming','wangyizhe']
  #cam = cv2.VideoCapture(0)
  minW = 0.1 * cam.get(3)
  minH = 0.1 * cam.get(4)
  while True:
    ret, img = cam.read()
    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    faces = faceCascade.detectMultiScale(
      gray,
      scaleFactor=1.2,
      minNeighbors=5,
      minSize=(int(minW), int(minH))
    )
    for (x, y, w, h) in faces:
      cv2.rectangle(img, (x, y), (x + w, y + h), (0, 255, 0), 2)
      idnum, confidence = recognizer.predict(gray[y:y + h, x:x + w])
      if confidence < 100:
        Name =connect.readName(idnum) #connect 传入ID 学生信息表找到 返回 name
        Sex =connect.readSex(idnum) #connect ID 学生信息表找到 返回 Sex
        StudentID =connect.readStudentID(idnum) #connect ID 学生信息表找到 返回 studentID
        #idnum = names[idnum]  #利用数据库 读取学生信息表 该id 对应的name
        confidence = "{0}%".format(round(100 - confidence))
        if sign_flag=='0': #签到
          say(engine, "欢迎     "+Name+ sex[Sex]+"     签到成功 ")
          baseConnect.insertd(idnum,Name,StudentID,Sex) #签到表中 插入签到信息
          print("欢迎   "+Name+ sex[Sex] + "签到成功 ")
        else :
          say(engine, "欢迎     "+Name+ sex[Sex]+"     签退成功 ")
          baseConnect.insertt(idnum,Name,StudentID,Sex) #签到表中 插入签退信息
          print("欢迎   "+Name+ sex[Sex] + "签退成功 ")

        cv2.imshow("img",img)
        os.system("pause")
        return
      else:
        idnum = "unknown"
        confidence = "{0}%".format(round(100 - confidence))
      cv2.putText(img, str(idnum), (x + 5, y - 5), font, 1, (0, 0, 255), 1)
      cv2.putText(img, str(confidence), (x + 5, y + h - 5), font, 1, (0, 0, 0), 1)
    cv2.imshow('camera', img)
    k = cv2.waitKey(10)
    if k == 27:
      break
  cam.release()
  cv2.destroyAllWindows()

def say(engine,str):
  engine.say(str)
  engine.runAndWait()

def admission(): #录入信息模块
  #names = {"yumengzhen":0,"dujuanjuan":1,"litingting":2}
  say(engine, "请输入您的学号      ")
  StudentID = input("请输入学号:")
  # 读取数据库信息表 取出Name 对应ID
  ID=connect.readIDbaseStudentID(StudentID) #connect 传入name 学生信息表找到 返回 ID
  if ID==-1:#没有找到该学生插入学生信息
    while True:
      say(engine,"没有找到该学生信息 输人 0 注册 1重新输入")
      op=input("\n 没有找到该学生信息 输人数字 0 注册学生信息 1重新输入")
      if op=='0':
        Name,studentID,Sex=input("输入学生信息: Name studentID Sex").split()
        connect.insert(Name,studentID,Sex) #插入学生信息信息
      else:
        StudentID = input("请输入学号:")
      ID=connect.readIDbaseStudentID(StudentID) #connect 传入name 学生信息表找到 返回 ID
      if ID!=-1 :
        break
  say(engine, "正在打开摄像头")
  cam = cv2.VideoCapture(0)
  say(engine, "注视摄像头,开始采集人脸数据")
  getFace(cam, ID) # 实际传入的是id
  cam.release()

if __name__ == '__main__':
  names = {"yumengzhen":0,"dujuanjuan":1,"litingting":  2}
  password="123456" #密码
  engine = pyttsx3.init()
  rate = engine.getProperty('rate')
  engine.setProperty('rate', rate - 20)
  flag=makeDir(engine)
  #trainFace()
  while True:
    if flag==1 :
      flag = 0
      say(engine, "首次使用 没有人脸信息 ")
      say(engine, "是否要录入新的人脸信息   ")
      say(engine, "输入0 代表是 输入其他表示退出")
      value = input("0:是 or other:否")
      if value=='0':
        while True:
          admission()
          say(engine, "是否要继续录入新的人脸信息   ")
          say(engine, "输入0 代表是 输入其他表示退出")
          firstflag = input("0:是  其他:退出")
          if firstflag != '0':
            break
        say(engine, "采集完毕,开始训练")
        trainFace()
        say(engine, "训练完毕     ")

    #say(engine, "请选择登录方式  ")
    say(engine, "输入 0管理人员模式    1 进入签到/签退模式   2 退出学生签到系统 ")
    user=input("\n0:管理人员模式    1:进入签到/签退模式  2:退出学生签到系统\n")

    if user=='0':
      say(engine, "输入管理员密码  ")
      pd=input("\n输入管理员密码 :\n")
      count=1
      while True:
        if count==3:
          say(engine," 输入密码错误超过3次 强制退出输入   ")
          break

        if password == pd:
          say(engine, "管理员模式  ")
          #say(engine, "输入数字 0 导出签到表   1 导出个人签到表   2 导出时长表    3 导出信息表    4 录入人脸信息  5 退出")
          op = input("\n0:导出所有同学签到表  1:导出个人签到表 2:导出所有人员时长表  3:导出学生信息表 4 录入人脸信息  5 退出\n")
          if op == '0':
            baseConnect.sign()#导出签到表
            say(engine, "导出签到表成功  ")
            pass
          elif op == '1':
            say(engine,"输入导出学生的学号")
            StudentID=input("输入导出学生的学号")
            ID=connect.readIDbaseStudentID(StudentID)
            if ID==-1:
              say(engine, "没有该学生信息 ")
            else:
              baseConnect.peoson_sign(StudentID)#导出个人签到表
              Name =connect.readName(ID) #connect 传入ID 学生信息表找到 返回 name
              say(engine, "导出 "+Name+"  信息成功")

          elif op == '2':
            baseConnect.total_time()#导出时长表
            say(engine,"导出时长表成功   ")
          elif op == '3':
            #导出学生信息表
            connect.find_student_all()
            say(engine, "导出学生信息成功   ")
          elif op == '4':
            while True:
              admission()
              say(engine, "是否要继续录入新的人脸信息   ")
              say(engine, "输入0 代表是 输入其他表示退出")
              secondflag = input("0:是  其他:退出")
              if secondflag != '0':
                break
            say(engine, "采集完毕,开始训练")
            trainFace()
            say(engine, "训练完毕     ")
          elif op == '5':
            say(engine, "已退出 管理员模式  ")
            break
          else:
            say(engine, "输入形式错误 请重新输入   ")
        else:
          say(engine, "输入密码错误   请重新输入 ")
          pd = input("\n输入管理员密码 :\n")
          count += 1;

    elif user=='1':
      say(engine, "欢迎进入学生系统签到/签退模式   ")
      sign_flag=0;
      while True:
        say(engine, "输入数字 0 签到   1 签退")
        sign_flag = input("\n0: 签到   1 签退\n")
        if sign_flag=='1' or sign_flag=='0' :
          break
        else :
          say(engine,"  请输入正确的输入形式")
      say(engine, "开始人脸识别")
      say(engine, "正在打开摄像头")
      cam = cv2.VideoCapture(0)
      checkFace(cam, names, engine,sign_flag)

    elif user=='2':
      say(engine, "信息已保存")
      say(engine, "再见")
      sys.exit(0)
    else:
      say(engine, "输入错误请重新输入   ")

student_sql.py

学生

import pymssql as py
import pandas as pd
# 连接数据库,创建学生表,进行表查询,表录入
server = "DESKTOP-XXX"# 连接自己数据库的服务器地址
user = "sa"# 连接帐号
password = "123"# 连接密码
conn = py.connect(server, user, password, "student_message") #获取连接
cursor = conn.cursor() # 获取光标

# 创建表
cursor.execute("""
IF OBJECT_ID('students', 'U') IS NOT NULL
  DROP TABLE students
CREATE TABLE students (
  ID INT NOT NULL,
  name VARCHAR(100),
  StudentID INT,
  Sex VARCHAR(100)
  )
""")
conn.commit()
#第一次运行时建立表,之后再运行无需再建

def insert(Name, studentID, Sex):
    count_students = 0
    try:
      conn = py.connect(server, user, password, "student_message") # 获取连接
      cursor =conn.cursor()
      cursor.execute(' select count(ID) from students')
      for row in cursor:
        count_students = row[0]
        print(row[0])
      cursor.executemany(
        "INSERT INTO students VALUES (%d, %s, %d,%s)",
        [(count_students+1, Name, studentID, Sex)])
    # 你必须调用 commit() 来保持你数据的提交如果你没有将自动提交设置为true
      conn.commit()
      conn.close()
    except py.InterfaceError:
      print("数据库连接出错")
    except py.ProgrammingError:
      print("数据错误,请检查输入的数据")
    except py.OperationalError:
      print("数据错误,请检查输入的数据")

# 导出学生信息表
def find_student_all():
  try:
    conn = py.connect(server, user, password, "student_message") # 获取连接
    cursor =conn.cursor()
    sql = "select * from students"
    df = pd.read_sql(sql, conn)
    df.to_excel('all.xlsx',index=False)
    print('ok')
    conn.commit()
    conn.close()
  except py.InterfaceError:
    print("数据库连接出错")
  except py.ProgrammingError:
    print("数据错误,请检查输入的数据")

#find_student_all()

def readName(idnum):
  Name = -1
  try:
    conn = py.connect(server, user, password, "student_message") # 获取连接
    cursor =conn.cursor()
    cursor.execute(' select Name from students where ID='+str(idnum))
    for row in cursor:
      if row[0]!=[]:
        Name = row[0]
    conn.commit()
    conn.close()
  except py.InterfaceError:
    print("数据库连接出错")
  except py.ProgrammingError:
    print("数据错误,请检查输入的数据")
  return Name

def readIDbaseStudentID(StudentID):
  ID = -1
  try:
    conn = py.connect(server, user, password, "student_message") # 获取连接
    cursor =conn.cursor()
    cursor.execute(' select ID from students where StudentID='+str(StudentID))
    for row in cursor:
      if row[0]!=[]:
        ID = row[0]
    conn.commit()
    conn.close()
  except py.InterfaceError:
    print("数据库连接出错")
  except py.ProgrammingError:
    print("数据错误,请检查输入的数据")
  return ID

def readSex(idnum):
  Sex = -1
  try:
    conn = py.connect(server, user, password, "student_message") # 获取连接
    cursor =conn.cursor()
    cursor.execute(' select Sex from students where ID='+str(idnum))
    for row in cursor:
       if row[0]!=[]:
          Sex = row[0]
    conn.commit()
    conn.close()
  except py.InterfaceError:
    print("数据库连接出错")
  except py.ProgrammingError:
    print("数据错误,请检查输入的数据")
  return Sex

def readID(name):
  # 多个id
  ID = -1
  try:
    conn = py.connect(server, user, password, "student_message") # 获取连接
    cursor =conn.cursor()
    cursor.execute(' select ID from students where name='+'\''+str(name)+'\'')
    for row in cursor:
      if row[0]!=[]:
        ID = row[0]
    conn.commit()
    conn.close()
  except py.InterfaceError:
    print("数据库连接出错")
  except py.ProgrammingError:
    print("数据错误,请检查输入的数据")
  return ID

def readStudentID(idnum):
  StudentID = -1
  try:
    conn = py.connect(server, user, password, "student_message") # 获取连接
    cursor =conn.cursor()
    cursor.execute(' select StudentID from students where ID='+str(idnum))
    for row in cursor:
      if row[0]!=[]:
        StudentID = row[0]
    conn.commit()
    conn.close()
  except py.InterfaceError:
    print("数据库连接出错")
  except py.ProgrammingError:
    print("数据错误,请检查输入的数据")
  return StudentID

# 关闭连接
# conn.close()
#
# # 注:在任何时候,在一个连接下,一次正在执行的数据库操作只会出现一个cursor对象

recognizer_sql.py

import pymssql as py
import time
import pandas as pd

server = "DESKTOP-XXXX"# 连接服务器地址
user = "sa" # 连接帐号
password = "123" # 连接密码
conn = py.connect(server, user, password, "student_message") #获取连接
cursor = conn.cursor() # 获取光标

def insertd(idnum,Name,StudentID,Sex):  # 签到
  conn = py.connect(server, user, password, "student_message") # 获取连接
  timenow = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
  cursor = conn.cursor()
  cursor.execute("INSERT INTO qiandao VALUES (%d, %s,%d,%s, %s, %s, %s,%d )", (idnum,Name,StudentID,Sex,timenow, '0', '0', 0))
  conn.commit()
  # 必须调用 commit() 来保持数据的提交
def insertt(idnum,Name,StudentID,Sex):  # 签退
  conn = py.connect(server, user, password, "student_message") # 获取连接
  cursor = conn.cursor()
  cursor.execute("SELECT starttime FROM qiandao WHERE ID=%s and flag=%d",(idnum,0))
  starttimeget = cursor.fetchone()
  sat = str(tuple(starttimeget))
  timenow = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
  timeArray = time.strptime(sat, "('%Y-%m-%d %H:%M:%S',)")
  timeStamp = int(time.mktime(timeArray))
  timecout = time.time() - timeStamp
  m, s = divmod(timecout, 60)
  h, m = divmod(m, 60)
  timepass = str(h) + '小时 ' + str(m) + '分钟 ' + str(s) + '秒'
  print(timepass)
  cursor.executemany("INSERT INTO qiandao VALUES (%d, %s,%d,%s, %s, %s, %s,%d )",
    [(idnum,Name,StudentID,Sex,'0', timenow, timepass, 1)])
  conn.commit()

def peoson_sign(StudentID):# 导出学生信息表_按照学号
  conn = py.connect(server, user, password, "student_message") # 获取连接
  cursor = conn.cursor()
  sql = "select * from qiandao where StudentID=" + str(StudentID)
  df = pd.read_sql(sql, conn)
  df.to_excel(r'E:\01STUDY\20190701\work\openVersion\excel\studentID_sign.xlsx',index=False)
  print('ok')
  conn.commit()

#peoson_sign(2016002105)

def sign():# 导出签到表
  conn = py.connect(server, user, password, "student_message") # 获取连接
  cursor = conn.cursor()
  sql = "select * from qiandao"
  df = pd.read_sql(sql, conn)
  df.to_excel(r'E:\01STUDY\20190701\work\openVersion\excel\sign_all.xlsx', index=False)
  print('ok')
  conn.commit()
#sign()

def total_time():# 导出时长表#sign()
  conn = py.connect(server, user, password, "student_message") # 获取连接
  cursor = conn.cursor()
  sql = "select * from qiandao where convert(nvarchar(max),count) != convert(nvarchar(max),0)"
  df = pd.read_sql(sql, conn)
  df.to_excel(r'E:\01STUDY\20190701\work\openVersion\excel\total_time.xlsx', index=False)
  print('ok')
  conn.commit()

#
# if __name__=='__main':
#   sign()
#   #peoson_sign(2016002105)

conn.close()

源码获取加群:850591259

猜你喜欢

转载自www.cnblogs.com/Py1233/p/12698375.html
今日推荐