需要三个库
1.cuda直接下载
2.opencv3.0重新编译
3.caffe编译库
#include<opencv2/opencv.hpp>
#include <opencv2/imgproc.hpp>#include <opencv2/highgui.hpp>
#include <caffe/caffe.hpp>
#include "type_recognition_ver2_api_gpu.h"
#include <fstream>
#include <iostream>
#include <cstdlib>
using namespace std;
using namespace cv;
using namespace caffe;
int tired_num = 0;
//人脸检测的类
CascadeClassifier faceCascade;
int main(){
Caffe::set_mode(Caffe::GPU);
std::string model_file("deploy.prototxt");
std::string trained_file("caffenet_train_iter_10000.caffemodel");
std::string mean_file("imagenet_mean.binaryproto");
std::string label_file("words.txt");
Classifier myclassifier(model_file, trained_file, mean_file, label_file);
faceCascade.load("haarcascade_frontalface_alt2.xml"); //加载分类器,注意文件路径
double scale = 1.3;
float tired_sum = 1.0;
VideoCapture cap;
//cap.open("tt.avi");//打开视频
cap.open(0); //打开摄像头
if (!cap.isOpened()){
return 1;
}
double rate = cap.get(CV_CAP_PROP_FPS);
bool stop(false);
Mat img, imgGray;
vector<Rect> faces;
int c = 0;
int sum = 0;
for (;;){
if (!cap.read(img)) {
std::cout << "No frame" << std::endl;
cv::waitKey();
}
cv::imshow("Output Window", img);
if (img.channels() == 3)
{
cvtColor(img, imgGray, CV_RGB2GRAY);
}
else
{
imgGray = img;
}
faceCascade.detectMultiScale(imgGray, faces, 1.1, 3, 0, Size(0, 0)); //检测人脸 faces为检测到目标区域列表
printf("检测到人脸个数:%d\n", faces.size());
Mat srcROI;
if (faces.size() > 0)
{
for (int i = 0; i < faces.size(); i++)//faces.size()检测到人脸的个数
{
srcROI = img(Rect(faces[i].x, faces[i].y, faces[i].width, faces[i].height));
vector<int> compression_params;
compression_params.push_back(CV_IMWRITE_PNG_COMPRESSION);
//rectangle(img, Point(faces[i].x, faces[i].y), Point(faces[i].x + faces[i].width, faces[i].y + faces[i].height), Scalar(0, 255, 0), 1, 8); //框出人脸位置
imshow("video111", srcROI);
std::vector<Prediction> result = myclassifier.Classify(srcROI);
Prediction p = result[0];
std::cout << "类别:" << p.first << "确信度:" << p.second * 100 << '%' << "\n";
if (p.first == 0){ tired_num++; }
}
sum++;
}
if (cv::waitKey(1) >= 0) break;
}
//计算视频中疲劳指数
if (sum != 0){
tired_sum = tired_num / sum;
std::cout << "疲劳状态图像张数:" << tired_num << "\n";
if (tired_sum >= 0.30){
std::cout << "该学习者现在正处于疲劳状态,请注意休息" << "\n";
}
else{
std::cout << "该学习者现在正处于正常状态" << "\n";
}
}
else{
std::cout << "未在视频中捕获到人脸!!" << "\n";
}
waitKey(0);
return 0;
}