video_to_heartpulserate_cpp/ppg.cpp

68 lines
2.0 KiB
C++

#include <iostream>
#include "opencv2/opencv.hpp"
#include "opencv2/videoio.hpp"
#include "opencv2/highgui.hpp"
#define FPS 30.0
#define temps 1.0
bool isDiscardData = true;
int countDiscard = 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));
for (size_t i = 0; i < faceRectangles.size(); i++) {
cv::rectangle(frame, faceRectangles[i], cv::Scalar(255, 0, 0), 2);
cv::Rect foreheadROI = faceRectangles[i];
foreheadROI.height *= 0.3;
cv::rectangle(frame, foreheadROI, cv::Scalar(0, 255, 0), 2);
}
cv::imshow("Color", frame);
}
if (cv::waitKey(1000.0/FPS) >= 0) {
break;
}
}
return 0;
}