基于OpenCV和pyqt5的人脸录入技术：简单高效的个人信息记录方式

人脸识别技术在现代社会中得到了广泛的应用。它可以用于安全监控、人脸支付、人脸解锁等众多场合。在这些应用中，最关键的环节就是人脸录入。本文将介绍一种基于摄像头的人脸录入技术。

文章目录

- GUI界面
- 人脸识别

大致分为两部分，一部分是业务逻辑，一部分是GUI界面。

GUI界面

使用PyQt5制作GUI界面。

# 录入窗口
from PyQt5.QtCore import Qt
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
import sys
# 窗口类
class UI_FaceRec(QWidget):
    def __init__(self):
        super(UI_FaceRec, self).__init__()
        self.resize(1000,600)
        leftw = QWidget()
        rightw= QWidget()
        # 布局
        layout=QHBoxLayout(self)
        layout.addWidget(leftw)
        layout.addWidget(rightw)
        # 左侧区域
        formlayout=QFormLayout(leftw)
        #学号 姓名
        self.account=QLineEdit()
        self.username=QLineEdit()
        formlayout.addRow("学号",self.account)
        formlayout.addRow("姓名", self.username)


        self.btnsave=QPushButton("保存")
        self.btnsave.clicked.connect(self.save)
        btnlay = QHBoxLayout()
        btnlay.addWidget(self.btnsave,alignment=Qt.AlignCenter)
        formlayout.addRow(btnlay)


        # 摄像头区域
        vied = QVBoxLayout(rightw)
        self.img = QLabel("头像")
        self.img.setStyleSheet("background-color:#00ff00")
        vied.addWidget(self.img)


if __name__=="__main__":
    app=QApplication(sys.argv)
    w=UI_FaceRec()
    w.show()
    sys.exit(app.exec_())

人脸识别

将录入的人脸信息保存到文件夹和数据库中。

# 录入业务逻辑
import sys

from threading import Thread

import face_recognition
from PyQt5.QtGui import QImage,QPixmap
import os



from day2.day23 import UI_FaceRec,QApplication
from PIL import Image,ImageDraw,ImageFont
import numpy as np
import cv2
# 录入业务类
from utils.DBHelper import Dbutil


class LuruApp(UI_FaceRec):
    def __init__(self):
        super().__init__()
        self.open()
        self.db = Dbutil()


    def open(self):
        # 全0三维数组
        frame = np.zeros((750,600,3),np.uint8)
        # 灰化
        img = Image.fromarray(cv2.cvtColor(frame,cv2.COLOR_BGR2RGB))
        draw = ImageDraw.Draw(img)
        font = ImageFont.truetype("simkai.ttf",40,encoding="utf-8")
        draw.text((200,200),"请正视摄像头",(200,0,0),font=font)
        frame = cv2.cvtColor(np.asarray(img),cv2.COLOR_BGR2RGB)
        frame = cv2.resize(frame,(self.img.width()-10,self.img.height()-10))
        h,w,d = frame.shape
        #从内存创建图像
        frame = QImage(frame.data,w,h,w*d,QImage.Format_RGB888)
        self.img.setPixmap(QPixmap.fromImage(frame))
        #开启子线程
        self.thread=Thread(target=self.doLuru)
        self.thread.start()

     #录入
    def doLuru(self):

        self.cap=cv2.VideoCapture(0,cv2.CAP_DSHOW)
        while True:
            ok,frame=self.cap.read()
            frame = cv2.flip(frame, 1)
            # 复制图像
            raw = frame.copy()
            # 人脸检测
            faces = face_recognition.face_locations(frame)
            if len(faces)==0:
                self.msg="没人录入"
            elif len(faces)>1:
                self.msg="一次只能录入一次"
            else:
                self.msg=""
                marks=face_recognition.face_landmarks(frame,faces)
                facecode=face_recognition.face_encodings(frame,faces)[0]
                top, right, bottom, left = faces[0]
                # 脸部图片
                self.face_img = cv2.resize(frame,(0,0),fx=0.25,fy=0.25)
                # 脸部图片编码
                self.face_img_code = facecode.tolist()
            img = Image.fromarray(cv2.cvtColor(raw,cv2.COLOR_BGR2RGB))
            draw=ImageDraw.Draw(img)
            for x1, y1, x2, y2 in faces:
                draw.rectangle([(y2, x1), (y1, x2)], outline=(200, 0, 0), width=2)
            font = ImageFont.truetype("simkai.ttf",40,encoding="utf-8")
            draw.text((200, 200),self.msg, (200, 0, 0), font=font)
            frame = cv2.cvtColor(np.asarray(img), cv2.COLOR_BGR2RGB)
            h, w, c= frame.shape
            frame = QImage(frame.data, w, h, w * c, QImage.Format_RGB888)
            self.img.setPixmap(QPixmap.fromImage(frame))

    def save(self):
        # 获取学号和姓名
        account=self.account.text()
        username=self.username.text()
        path=os.path.join("../faces",account)
        if not os.path.exists(path):
            # 创建目录
            os.mkdir(path)
        try:
            # 保存图像
            cv2.imencode(".jpg", self.face_img)[1].tofile(os.path.join(path, username + ".jpg"))
            sql = "insert into user values(%s,%s)"

            self.db.save(sql, [[account, username]])

        except Exception as ex:
            print(ex)








if __name__=="__main__":
    app = QApplication(sys.argv)
    w = LuruApp()
    w.show()
    sys.exit(app.exec_())