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

Some small fixes on recording in feature extraction.

Browse source
This commit is contained in:
Tadas Baltrusaitis 2017-11-05 08:56:25 +00:00
parent 1159caba75
commit 3b084fa4f3
1 changed files with 8 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
exe/FeatureExtraction/FeatureExtraction.cpp
View file

@ -157,11 +157,6 @@ void visualise_tracking(cv::Mat& captured_image, const LandmarkDetector::CLNF& f
fpsSt += fpsC; fpsSt += fpsC;
cv::putText(captured_image, fpsSt, cv::Point(10, 20), CV_FONT_HERSHEY_SIMPLEX, 0.5, CV_RGB(255, 0, 0), 1, CV_AA); cv::putText(captured_image, fpsSt, cv::Point(10, 20), CV_FONT_HERSHEY_SIMPLEX, 0.5, CV_RGB(255, 0, 0), 1, CV_AA);
if (!det_parameters.quiet_mode)
{
cv::namedWindow("tracking_result", 1);
cv::imshow("tracking_result", captured_image);
}
} }
int main (int argc, char **argv) int main (int argc, char **argv)
@ -415,7 +410,7 @@ int main (int argc, char **argv)
// Work out the pose of the head from the tracked model // Work out the pose of the head from the tracked model
cv::Vec6d pose_estimate = LandmarkDetector::GetPose(face_model, fx, fy, cx, cy); cv::Vec6d pose_estimate = LandmarkDetector::GetPose(face_model, fx, fy, cx, cy);
// Visualising the tracker, TODO this should be in utility // Drawing the visualization on the captured image
if (recording_params.outputTrackedVideo() || (visualize_track && !det_parameters.quiet_mode)) if (recording_params.outputTrackedVideo() || (visualize_track && !det_parameters.quiet_mode))
{ {
visualise_tracking(captured_image, face_model, det_parameters, gazeDirection0, gazeDirection1, frame_count, fx, fy, cx, cy); visualise_tracking(captured_image, face_model, det_parameters, gazeDirection0, gazeDirection1, frame_count, fx, fy, cx, cy);
@ -431,6 +426,13 @@ int main (int argc, char **argv)
open_face_rec.SetObservationTimestamp(time_stamp); open_face_rec.SetObservationTimestamp(time_stamp);
open_face_rec.WriteObservation(); open_face_rec.WriteObservation();
// Visualize the image if desired
if (visualize_track && !det_parameters.quiet_mode)
{
cv::namedWindow("tracking_result", 1);
cv::imshow("tracking_result", captured_image);
}
// Grabbing the next frame (todo this should be part of capture) // Grabbing the next frame (todo this should be part of capture)
if(video_input) if(video_input)
{ {