thomas.perin
/
Lab1_PhotoPlethysmoGraphy
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Final version

This commit is contained in:
Thomas PÉRIN 2023-03-01 23:35:58 +01:00
parent d90d3ae2b4
commit 0f0fc3f502
1 changed files with 24 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
ppg.cpp
View File

@ -2,12 +2,17 @@
#include "opencv2/videoio.hpp" #include "opencv2/videoio.hpp"
#include "opencv2/highgui.hpp" #include "opencv2/highgui.hpp"
const int FPS = 15;
const int FPS = 30;
bool isDiscardData = true; bool isDiscardData = true;
int countDiscard = 0; int countDiscard = 0;
const int DISCARD_DURATION = 5; const int DISCARD_DURATION = 5;
const int BUFFER_DURATION = 30; const int BUFFER_DURATION = 30 ;
const int I = 1;
bool isBufferFull = false;
int sampleIdBuffer = 0;
std::vector<double> greenFFTModule;
template <typename T> template <typename T>
cv::Mat plotGraph(std::vector<T>& vals, int YRange[2]) cv::Mat plotGraph(std::vector<T>& vals, int YRange[2])
@ -79,8 +84,7 @@ int main()
cv::Mat frame_forehead = frame(foreheadROI); cv::Mat frame_forehead = frame(foreheadROI);
cv::Scalar avg_forehead = mean(frame_forehead); cv::Scalar avg_forehead = mean(frame_forehead);
bool isBufferFull = false; //
int sampleIdBuffer = 0;
cv::Mat greenSignal(1, FPS*BUFFER_DURATION, CV_64F); cv::Mat greenSignal(1, FPS*BUFFER_DURATION, CV_64F);
if (!isBufferFull) if (!isBufferFull)
{ {
@ -103,23 +107,33 @@ int main()
int range[2] = {0, (int)(FPS*BUFFER_DURATION)}; int range[2] = {0, (int)(FPS*BUFFER_DURATION)};
cv::imshow("green", plotGraph(greenSignalNormalized, range));
cv::Mat greenFFT;
///
cv::dft(greenSignalNormalized,greenFFT,cv::DFT_ROWS|cv::DFT_COMPLEX_OUTPUT);
cv::Mat planes[] = {cv::Mat::zeros(greenSignalNormalized.size(),1, CV_64F),
cv::Mat::zeros(greenSignalNormalized.size(),1, CV_64F)};
cv::split(greenFFT, planes); //planes[0] = Re(DFT(I),
//planes[1] = Im(DFT(I))
greenFFTModule.clear();
for (int l=0; l < planes[1].cols; l++)
{
double moduleFFT = pow(planes[1].at<double>(0,l),2) + pow(planes[0].at<double>(0,l),2);
greenFFTModule.push_back(sqrt(moduleFFT));
}
// display green FFT
cv::imshow("FFT module green", plotGraph(greenFFTModule, range));
} }
} }
cv::imshow("Your Face PLS", frame); cv::imshow("Your Face PLS", frame);
if (cv::waitKey(1000.0/FPS) >= 0) if (cv::waitKey(1000.0/FPS) >= 0)
break; break;
} }
} }
return 0; return 0;
} }