90 lines
2.8 KiB
C++
90 lines
2.8 KiB
C++
#include <iostream>
|
|
#include "opencv2/opencv.hpp"
|
|
#include "opencv2/videoio.hpp"
|
|
#include "opencv2/highgui.hpp"
|
|
|
|
|
|
#define BUFFER_DURATION 30.0
|
|
#define FPS 30.0
|
|
#define temps 1.0
|
|
|
|
bool isDiscardData = true;
|
|
int countDiscard = 0;
|
|
bool isBufferFull = false;
|
|
int sampleIdBuffer = 0;
|
|
|
|
int main() {
|
|
std::cout << "PPG, j'adore gitbash même si avec github desktop c'est mieux" << std::endl;
|
|
|
|
cv::VideoCapture cap;
|
|
cap.open(0);
|
|
cv::CascadeClassifier faceDetector;
|
|
if (!faceDetector.load("./haarcascade_frontalface_alt.xml")) {
|
|
std::cerr << "[ERROR] Unable to load face cascade" << std::endl;
|
|
return -1;
|
|
}
|
|
|
|
if (!cap.isOpened()) {
|
|
std::cerr << "[ERROR] Unable to open camera!" << std::endl;
|
|
return -2;
|
|
}
|
|
|
|
while (true) {
|
|
if (isDiscardData) {
|
|
countDiscard++;
|
|
std::cout << "countDiscard = " << countDiscard << std::endl;
|
|
if (countDiscard == temps*FPS) {
|
|
isDiscardData = false;
|
|
}
|
|
} else {
|
|
// create a matrix to store the image from the cam
|
|
cv::Mat frame;
|
|
// wait for a new frame from camera and store it into 'frame'
|
|
cap.read(frame);
|
|
// check if we succeeded
|
|
if (frame.empty()) {
|
|
std::cerr << "[ERROR] blank frame grabbed" << std::endl;
|
|
break;
|
|
}
|
|
|
|
cv::Mat frame_gray;
|
|
cv::cvtColor(frame, frame_gray, cv::COLOR_BGR2GRAY);
|
|
std::vector<cv::Rect> faceRectangles;
|
|
faceDetector.detectMultiScale(frame_gray, faceRectangles, 1.1, 3, 0, cv::Size(20, 20));
|
|
cv::rectangle(frame, faceRectangles[0], cv::Scalar(255, 0, 0), 2);
|
|
cv::Rect foreheadROI = faceRectangles[0];
|
|
foreheadROI.height *= 0.3;
|
|
cv::rectangle(frame, foreheadROI, cv::Scalar(0, 255, 0), 2);
|
|
cv::imshow("Color", frame);
|
|
cv::Mat frame_forehead = frame(foreheadROI);
|
|
cv::Scalar avg_forehead = mean(frame_forehead);
|
|
cv::Mat greenSignal(1, FPS*BUFFER_DURATION, CV_64F);
|
|
|
|
|
|
if (!isBufferFull)
|
|
{
|
|
greenSignal.at<double>(0, sampleIdBuffer) = avg_forehead[1] ;
|
|
sampleIdBuffer++;
|
|
if (sampleIdBuffer == FPS*BUFFER_DURATION)
|
|
{
|
|
isBufferFull = true;
|
|
}
|
|
std::vector<double> greenSignalNormalized;
|
|
cv::Scalar mean, stddev;
|
|
cv::meanStdDev(greenSignal, mean, stddev);
|
|
for (int l_sample=0; l_sample < FPS*BUFFER_DURATION; l_sample++)
|
|
{
|
|
greenSignalNormalized.push_back((greenSignal.at<double>(0, l_sample) -mean[0])/stddev[0]);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (cv::waitKey(1000.0/FPS) >= 0) {
|
|
break;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|