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

Towards getting hog recording in GUI image processing.

Browse source
This commit is contained in:
Tadas Baltrusaitis 2018-01-18 21:55:15 +00:00
parent df23dc2eb5
commit 3b21213f41
3 changed files with 24 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
gui/OpenFaceOffline/MainWindow.xaml.cs
View file

@ -458,10 +458,12 @@ namespace OpenFaceOffline
recorder.SetObservationFaceAlign(face_analyser.GetLatestAlignedFace());
var hog_feature = face_analyser.GetLatestHOGFeature();
//open_face_rec.SetObservationHOG(face_model.detection_success, hog_descriptor, num_hog_rows, num_hog_cols, 31); // The number of channels in HOG is fixed at the moment, as using FHOG
recorder.WriteObservation();
// TODO
//open_face_rec.SetObservationHOG(face_model.detection_success, hog_descriptor, num_hog_rows, num_hog_cols, 31); // The number of channels in HOG is fixed at the moment, as using FHOG
//open_face_rec.SetObservationVisualization(visualizer.GetVisImage());
}

22
lib/local/CppInerop/FaceAnalyserInterop.h
View file

@ -85,7 +85,7 @@ private:
FaceAnalysis::FaceAnalyser* face_analyser;
// The actual descriptors (for visualisation and output)
cv::Mat_<double>* hog_features;
cv::Mat_<float>* hog_features;
cv::Mat* aligned_face;
cv::Mat* visualisation;
@ -111,7 +111,7 @@ public:
params.setAlignedOutput(output_width);
face_analyser = new FaceAnalysis::FaceAnalyser(params);
hog_features = new cv::Mat_<double>();
hog_features = new cv::Mat_<float>();
aligned_face = new cv::Mat();
visualisation = new cv::Mat();
@ -172,7 +172,7 @@ public:
hog_output_file->write((char*)(&good_frame_float), 4);
cv::MatConstIterator_<double> descriptor_it = hog_features->begin();
cv::MatConstIterator_<float> descriptor_it = hog_features->begin();
for(int y = 0; y < *num_cols; ++y)
{
@ -181,7 +181,7 @@ public:
for(unsigned int o = 0; o < 31; ++o)
{
float hog_data = (float)(*descriptor_it++);
float hog_data = (*descriptor_it++);
hog_output_file->write((char*)&hog_data, 4);
}
}
@ -208,7 +208,9 @@ public:
face_analyser->AddNextFrame(frame->Mat, landmarks_mat, success, 0, online);
face_analyser->GetLatestHOG(*hog_features, *num_rows, *num_cols);
cv::Mat_<double> hog_d;
face_analyser->GetLatestHOG(hog_d, *num_rows, *num_cols);
hog_d.convertTo(*hog_features, CV_64F);
face_analyser->GetLatestAlignedFace(*aligned_face);
@ -237,7 +239,10 @@ public:
face_analyser->PredictStaticAUsAndComputeFeatures(frame->Mat, landmarks_mat);
// Set the computed appearance features
face_analyser->GetLatestHOG(*hog_features, *num_rows, *num_cols);
cv::Mat_<double> hog_tmp;
face_analyser->GetLatestHOG(hog_tmp, *num_rows, *num_cols);
hog_tmp.convertTo(*hog_features, CV_32F);
face_analyser->GetLatestAlignedFace(*aligned_face);
if (vis_hog)
@ -330,6 +335,11 @@ public:
return HOG_vis_image;
}
OpenCVWrappers::RawImage^ GetLatestHOGFeature() {
OpenCVWrappers::RawImage^ HOG_feature = gcnew OpenCVWrappers::RawImage(*hog_features);
return HOG_feature;
}
void Reset()
{
face_analyser->Reset();

5
lib/local/CppInerop/RecorderInterop.h
View file

@ -176,6 +176,11 @@ namespace UtilitiesOF {
m_recorder->SetObservationFaceAlign(aligned_face_image->Mat);
}
void SetObservationHOG(bool success, OpenCVWrappers::RawImage^ aligned_face_image, int num_cols, int num_rows, int num_channels)
{
m_recorder->SetObservationHOG(success, aligned_face_image->Mat, num_cols, num_rows, num_channels);
}
void SetObservationLandmarks(List<System::Tuple<double, double>^>^ landmarks_2D, List<System::Tuple<double, double, double>^>^ landmarks_3D, List<double>^ params_global, List<double>^ params_local, double confidence, bool success)
{
// Construct an OpenCV matrix from the landmarks