Dlib —— 对视频流进行人脸识别（附C++部分源码、完整源码下载）

news2024/11/27 12:30:01

照片效果

在这里插入图片描述

视频效果

注意：Dlib检测人脸在Release版耗时与CPU有关,本人I7 10代约100ms左右，这里本人将人脸检测用Yolov5对人脸简单抠图训练后使用yolov5推理检测人脸，之后将检测到的人脸输入给Dlib做特征，发现人脸特征部分耗时也较高约200ms，之后考虑分线程做检测或查找如何降低耗时。

本源码运行环境为Vs2017+OpenCV+Dlib+Qt，功能为获取视频流或mp4视频进行人脸检测，检测后可进行人脸注册，注册后在之后的检测中会识别出人名与置信度。代码在识别时耗时约100ms，在实际生产中只需要检测到正确人脸即可，所以该耗时满足使用。

源码

void Class::writeSaveMatrix(dlib::matrix<float, 0, 1> Matrix)
{
	QFile wMatrixFile(regString);
	if (wMatrixFile.open(QFile::WriteOnly))
	{
		QByteArray data;
		for (int i = 0; i < Matrix.size(); i++)
		{
			QByteArray ba = QByteArray::number(Matrix(i, 0), 'g', 10);
			data.push_back(ba);
			if (i + 1 != Matrix.size()) { data.push_back(","); }
		}
		wMatrixFile.write(data.toHex());
		wMatrixFile.flush();
		wMatrixFile.close();
	}

	sltDirectoryChanged();
}

void Class::sltDirectoryChanged()
{
	localFaceInfo.clear();

	QDir dir(FACEDIR);
	dir.setFilter(QDir::Files | QDir::NoSymLinks);
	QFileInfoList list = dir.entryInfoList();
	for (unsigned short index = 0; index < list.size(); ++index)
	{
		QFileInfo file_info = list.at(index);
		QString pt = file_info.absoluteFilePath();
		QString ptName = file_info.baseName();

		QFile wMatrixFile(pt);
		if (wMatrixFile.open(QFile::ReadOnly))
		{
			QByteArray data = wMatrixFile.readAll();
			data = QByteArray::fromHex(data);
			QByteArrayList dataList = data.split(',');
			if (dataList.size() == 128)
			{
				dlib::matrix<float, 0, 1> Matrix(128, 1);
				for (unsigned short index = 0; index < dataList.size(); ++index)
				{
					Matrix(index, 0) = dataList[index].toDouble();
				}
				localFaceInfo.push_back(QPair<QString, dlib::matrix<float, 0, 1>>(ptName, Matrix));
			}
			wMatrixFile.close();
		}
	}
}