ZF/OpenFace
1
0
Fork 0
mirror of https://gitcode.com/gh_mirrors/ope/OpenFace.git synced 2026-05-16 04:08:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Output similarity aligned faces through recorder.

Browse Source
This commit is contained in:
Tadas Baltrusaitis
2017-11-05 08:51:27 +00:00
parent db1fef46eb
commit 1159caba75
3 changed files with 41 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
lib/local/Recorder/src/RecorderOpenFace.cpp
View File

@@ -97,12 +97,26 @@ RecorderOpenFace::RecorderOpenFace(const std::string out_directory, const std::s
this->video_filename = (path(record_root) / path(filename).replace_extension(".avi")).string();
}
// TODO aligned Prepare image recording
// Prepare image recording
if (params.outputAlignedFaces())
{
aligned_output_directory = (path(record_root) / path(filename + "_aligned")).string();
CreateDirectory(aligned_output_directory);
}
observation_count = 0;
}
// TODO move to actual write
void RecorderOpenFace::SetObservationFaceAlign(const cv::Mat& aligned_face)
{
this->aligned_face = aligned_face;
}
void RecorderOpenFace::SetObservationVisualization(const cv::Mat &vis_track)
{
if (params.outputTrackedVideo())
@@ -171,6 +185,28 @@ void RecorderOpenFace::WriteObservation()
this->hog_recorder.Write();
}
// Write aligned faces
if (params.outputAlignedFaces())
{
char name[100];
// Filename is based on frame number
std::sprintf(name, "frame_det_%06d.bmp", observation_count);
// Construct the output filename
boost::filesystem::path slash("/");
std::string preferredSlash = slash.make_preferred().string();
string out_file = aligned_output_directory + preferredSlash + string(name);
bool write_success = cv::imwrite(out_file, aligned_face);
if (!write_success)
{
WARN_STREAM("Could not output similarity aligned image image");
}
}
if(params.outputTrackedVideo())
{
if (vis_to_out.empty())