r/sustaining_gazes
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Fix with reading multiple sequences and visualizing data in multi-face setting.

Browse source
This commit is contained in:
Tadas Baltrusaitis 2017-11-22 17:37:26 +00:00
parent bafbfdca29
commit 41af428f5c
2 changed files with 11 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
exe/FaceLandmarkVid/FaceLandmarkVid.cpp
View file

@ -106,6 +106,8 @@ int main (int argc, char **argv)
Utilities::FpsTracker fps_tracker; Utilities::FpsTracker fps_tracker;
fps_tracker.AddFrame(); fps_tracker.AddFrame();
int sequence_number = 0;
while (true) // this is not a for loop as we might also be reading from a webcam while (true) // this is not a for loop as we might also be reading from a webcam
{ {
@ -113,7 +115,7 @@ int main (int argc, char **argv)
if(!sequence_reader.Open(arguments)) if(!sequence_reader.Open(arguments))
{ {
// If failed to open because no input files specified, attempt to open a webcam // If failed to open because no input files specified, attempt to open a webcam
if (sequence_reader.no_input_specified) if (sequence_reader.no_input_specified && sequence_number == 0)
{ {
// If that fails, revert to webcam // If that fails, revert to webcam
INFO_STREAM("No input specified, attempting to open a webcam 0"); INFO_STREAM("No input specified, attempting to open a webcam 0");
@ -190,6 +192,8 @@ int main (int argc, char **argv)
// Reset the model, for the next video // Reset the model, for the next video
face_model.Reset(); face_model.Reset();
sequence_number++;
} }
return 0; return 0;
} }

8
exe/FaceLandmarkVidMulti/FaceLandmarkVidMulti.cpp
View file

@ -152,6 +152,8 @@ int main (int argc, char **argv)
Utilities::FpsTracker fps_tracker; Utilities::FpsTracker fps_tracker;
fps_tracker.AddFrame(); fps_tracker.AddFrame();
int sequence_number = 0;
while(true) // this is not a for loop as we might also be reading from a webcam while(true) // this is not a for loop as we might also be reading from a webcam
{ {
@ -159,7 +161,7 @@ int main (int argc, char **argv)
if (!sequence_reader.Open(arguments)) if (!sequence_reader.Open(arguments))
{ {
// If failed to open because no input files specified, attempt to open a webcam // If failed to open because no input files specified, attempt to open a webcam
if (sequence_reader.no_input_specified) if (sequence_reader.no_input_specified && sequence_number == 0)
{ {
// If that fails, revert to webcam // If that fails, revert to webcam
INFO_STREAM("No input specified, attempting to open a webcam 0"); INFO_STREAM("No input specified, attempting to open a webcam 0");
@ -290,10 +292,10 @@ int main (int argc, char **argv)
{ {
visualizer.SetObservationLandmarks(face_models[model].detected_landmarks, face_models[model].detection_certainty, face_models[model].detection_success); visualizer.SetObservationLandmarks(face_models[model].detected_landmarks, face_models[model].detection_certainty, face_models[model].detection_success);
visualizer.SetObservationPose(LandmarkDetector::GetPose(face_models[model], sequence_reader.fx, sequence_reader.fy, sequence_reader.cx, sequence_reader.cy), face_models[model].detection_certainty); visualizer.SetObservationPose(LandmarkDetector::GetPose(face_models[model], sequence_reader.fx, sequence_reader.fy, sequence_reader.cx, sequence_reader.cy), face_models[model].detection_certainty);
visualizer.SetFps(fps_tracker.GetFPS());
visualizer.ShowObservation();
} }
} }
visualizer.SetFps(fps_tracker.GetFPS());
visualizer.ShowObservation();
// detect key presses // detect key presses
char character_press = cv::waitKey(1); char character_press = cv::waitKey(1);