QT实现人脸识别
Face.pro文件:
QT += core gui
greaterThan(QT_MAJOR_VERSION, 4): QT += widgets
CONFIG += c++11
# The following define makes your compiler emit warnings if you use
# any Qt feature that has been marked deprecated (the exact warnings
# depend on your compiler). Please consult the documentation of the
# deprecated API in order to know how to port your code away from it.
DEFINES += QT_DEPRECATED_WARNINGS
# You can also make your code fail to compile if it uses deprecated APIs.
# In order to do so, uncomment the following line.
# You can also select to disable deprecated APIs only up to a certain version of Qt.
#DEFINES += QT_DISABLE_DEPRECATED_BEFORE=0x060000 # disables all the APIs deprecated before Qt 6.0.0
SOURCES += \
main.cpp \
widget.cpp
HEADERS += \
widget.h
FORMS += \
widget.ui
INCLUDEPATH += D:/opencv/opencv3.4-qt-intall/install/include
INCLUDEPATH += D:/opencv/opencv3.4-qt-intall/install/include/opencv
INCLUDEPATH += D:/opencv/opencv3.4-qt-intall/install/include/opencv2
LIBS += D:/opencv/opencv3.4-qt-intall/install/x86/mingw/lib/libopencv_*.a
# Default rules for deployment.
qnx: target.path = /tmp/$${TARGET}/bin
else: unix:!android: target.path = /opt/$${TARGET}/bin
!isEmpty(target.path): INSTALLS += target
widget.h文件:
#ifndef WIDGET_H
#define WIDGET_H
#include <QWidget>
#include <QWidget>
#include <opencv2/opencv.hpp>
#include <iostream>
#include <math.h>
#include<opencv2/face.hpp>
#include <vector>
#include <map>
#include <QMessageBox>
#include <QDebug>
#include <QFile>
#include <QTextStream>
#include <QDateTime>
#include <QTimerEvent>
#include<QtSerialPort/QtSerialPort>
#include<QtSerialPort/QSerialPortInfo>
using namespace cv;
using namespace cv::face;
using namespace std;
QT_BEGIN_NAMESPACE
namespace Ui { class Widget; }
QT_END_NAMESPACE
class Widget : public QWidget
{
Q_OBJECT
public:
Widget(QWidget *parent = nullptr);
~Widget();
private slots:
void on_openBtn_clicked();
void on_closeBtn_clicked();
void on_faceBtn_clicked();
private:
Ui::Widget *ui;
//=========摄像头相关成员的设置=========
VideoCapture v; //摄像头容器
Mat src; //存放 原图 的容器
Mat gray; //存放 灰度图 的容器
Mat dst; //存放 直方图 的容器
Mat rgb; //存放 rgb图 的容器
CascadeClassifier c; //定义一个级联分类器容器
vector<Rect> faces; // 定义一个数组存放人脸矩形框
int camera_id; //打开摄像头的定时器id
void timerEvent(QTimerEvent *e); //定时器事件 重写函数事件声明
//=========人脸录入相关成员的设置=========
Ptr<LBPHFaceRecognizer> recognizer; //人脸 识别器指针
vector<Mat> study_faces; //定义一个存放录入人脸的数组
vector<int> study_lables; //定义一个存放人脸对应的标签数组
int count; //记录录入人脸的次数
int flag; //用来区别是人脸录入还是人脸识别
int face_id; // 人脸录入定时器id
//=========人脸检测相关成员的设置=========
int check_id; //
};
#endif // WIDGET_H
widget.cpp文件:
#include "widget.h"
#include "ui_widget.h"
Widget::Widget(QWidget *parent)
: QWidget(parent)
, ui(new Ui::Widget)
{
ui->setupUi(this);
ui->weChatBtn->setEnabled(false);
//打开系统摄像头
if(!v.open(0))
{
QMessageBox::information(this, "", "打开系统摄像头失败!");
return;
}
//配置 级联分类器
if(!c.load("D:\\opencv\\image\\haarcascade_frontalface_alt.xml"))
{
QMessageBox::information(this, "", "配置级联分类器失败!");
return;
}
//判断是否录入过人脸
QFile file("D:\\opencv\\image\\my_face.xml");
//判断文件是否存在
if(file.exists())
{
//表示之前录入过人脸 则下载文件
recognizer = LBPHFaceRecognizer::load<LBPHFaceRecognizer>("D:\\opencv\\image\\my_face.xml");
}else {
//表示之前没有录入过人脸
recognizer = LBPHFaceRecognizer::create();
}
//启动一个人脸检测定时器
check_id = startTimer(3000);
flag = 1; //可以人脸检测
recognizer->setThreshold(100);
}
Widget::~Widget()
{
delete ui;
}
//打开摄像头按钮对应的 槽函数
void Widget::on_openBtn_clicked()
{
//启动一个定时器
camera_id = startTimer(30);
}
//定时器重写的功能函数
void Widget::timerEvent(QTimerEvent *e)
{
//判断是否是 摄像头定时器 超时
if(e->timerId() == camera_id)
{
//读取系统摄像头中的图像
v.read(src);
//图像反转
flip(src, src, 1);
//将 bgr 转换成 rgb
cvtColor(src, rgb, CV_BGR2RGB);
//将图像重新设置大小 适应lab
cv::resize(rgb, rgb, Size(301,251));
//灰度处理
cvtColor(rgb, gray, CV_RGB2GRAY);
//均衡化处理
equalizeHist(gray, dst);
//锁定人脸矩形框位置
c.detectMultiScale(dst, faces);
//将矩形框绘制到人脸上
for(uint i=0; i<faces.size(); i++)
{
rectangle(rgb, faces[i], Scalar(255,0,0),2);
}
//将图像放入 lab 中
QImage img(rgb.data, rgb.cols, rgb.rows, rgb.cols*rgb.channels(), QImage::Format_RGB888);
ui->label->setPixmap(QPixmap::fromImage(img));
}
//判断是否是 人脸录入定时器 超时
if(e->timerId() == face_id)
{
if(0 == flag)
{
qDebug() << "人脸录入中,请正视摄像头!";
Mat face = src(faces[0]); //将摄像头当前的一帧图像上的一个人脸给face
//灰度处理
cvtColor(face, face, CV_BGR2GRAY);
//均衡化处理
equalizeHist(face, face);
//将人脸放入数组中
study_faces.push_back(face);
study_lables.push_back(1);
count++;
if(50 == count)
{
//将图像模型转换成数据模型
recognizer->update(study_faces, study_lables);
recognizer->save("D:\\opencv\\image\\my_face.xml");
QMessageBox::information(this, "", "录入人脸成功!");
//关闭定时器
killTimer(face_id);
flag = 1; //表示可以人脸识别
study_faces.clear(); //将存放人脸数组清空,方便下次录入
study_lables.clear();
}
}
}
//判断是否是 人脸检测定时器 超时
if(e->timerId() == check_id)
{
if(1 == flag)
{
QFile file("D:\\opencv\\image\\my_face.xml");
if(file.exists())
{
if(recognizer.empty() || faces.empty())
{
return;
}
Mat face = src(faces[0]);
//灰度处理
cvtColor(face, face, CV_BGR2GRAY);
//
equalizeHist(face, face);
int lab = -1;
double cof = 0.0;
recognizer->predict(face, lab, cof);
//根据 lab 判断是否识别成功
if(lab != -1)
{
QMessageBox::information(this, "", "人脸识别成功!");
ui->weChatBtn->setEnabled(true);
killTimer(check_id);
}
}
}
}
}
//关闭摄像头按钮对应的槽函数
void Widget::on_closeBtn_clicked()
{
killTimer(camera_id);
}
//录入人脸按钮对应的槽函数
void Widget::on_faceBtn_clicked()
{
count = 0; // 将录入人脸的次数设置为0
flag = 0; //表示只能做人脸录入,不能做人脸检测
face_id = startTimer(50);
}
main.c文件:
#include "widget.h"
#include <QApplication>
int main(int argc, char *argv[])
{
QApplication a(argc, argv);
Widget w;
w.show();
return a.exec();
}
ui布局: