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

Starting to introduce recording parameters for simplicity.

Browse source
This commit is contained in:
Tadas Baltrusaitis 2017-11-03 21:35:55 +00:00
parent 7f3ed9c855
commit 8beb7e58c4
3 changed files with 7 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
lib/local/Recorder/include/RecorderOpenFace.h
View file

@ -59,10 +59,6 @@ namespace Recorder
// The constructor for the recorder, need to specify if we are recording a sequence or not // The constructor for the recorder, need to specify if we are recording a sequence or not
RecorderOpenFace(const std::string out_directory, const std::string in_filename, RecorderOpenFaceParameters parameters); RecorderOpenFace(const std::string out_directory, const std::string in_filename, RecorderOpenFaceParameters parameters);
// Simplified constructor that records all, TODO implement
RecorderOpenFace(const std::string out_directory, const std::string in_filename, bool sequence, int num_face_landmarks, int num_model_modes, int num_eye_landmarks,
const std::vector<std::string>& au_names_class, const std::vector<std::string>& au_names_reg);
~RecorderOpenFace(); ~RecorderOpenFace();
// TODO copy, assignment and move operators? Do not allow // TODO copy, assignment and move operators? Do not allow

2
lib/local/Recorder/include/RecorderOpenFaceParameters.h
View file

@ -66,6 +66,8 @@ namespace Recorder
bool outputHOG() const { return output_hog; } bool outputHOG() const { return output_hog; }
bool outputTrackedVideo() const { return output_tracked_video; } bool outputTrackedVideo() const { return output_tracked_video; }
bool outputAlignedFaces() const { return output_aligned_faces; } bool outputAlignedFaces() const { return output_aligned_faces; }
std::string outputCodec() const { return output_codec; }
double outputFps() const { return fps_vid_out; }
private: private:

7
lib/local/Recorder/src/RecorderOpenFace.cpp
View file

@ -86,7 +86,7 @@ RecorderOpenFace::RecorderOpenFace(const std::string out_directory, const std::s
// Consruct HOG recorder here // Consruct HOG recorder here
if(output_hog) if(params.outputHOG())
{ {
std::string hog_filename = (path(record_root) / path(filename).replace_extension(".hog")).string(); std::string hog_filename = (path(record_root) / path(filename).replace_extension(".hog")).string();
hog_recorder.Open(hog_filename); hog_recorder.Open(hog_filename);
@ -95,7 +95,7 @@ RecorderOpenFace::RecorderOpenFace(const std::string out_directory, const std::s
// TODO construct a video recorder // TODO construct a video recorder
// saving the videos // saving the videos
if (output_tracked_video) if (params.outputTrackedVideo())
{ {
this->video_filename = (path(record_root) / path(filename).replace_extension(".avi")).string(); this->video_filename = (path(record_root) / path(filename).replace_extension(".avi")).string();
} }
@ -109,12 +109,13 @@ RecorderOpenFace::RecorderOpenFace(const std::string out_directory, const std::s
void RecorderOpenFace::SetObservationVisualization(const cv::Mat &vis_track) void RecorderOpenFace::SetObservationVisualization(const cv::Mat &vis_track)
{ {
if (output_tracked_video) if (params.outputTrackedVideo())
{ {
// Initialize the video writer if it has not been opened yet // Initialize the video writer if it has not been opened yet
if(!video_writer.isOpened()) if(!video_writer.isOpened())
{ {
std::string video_filename = (path(record_root) / path(filename).replace_extension(".avi")).string(); std::string video_filename = (path(record_root) / path(filename).replace_extension(".avi")).string();
std::string output_codec = params.outputCodec();
try try
{ {
video_writer.open(video_filename, CV_FOURCC(output_codec[0], output_codec[1], output_codec[2], output_codec[3]), fps_vid_out, vis_track.size(), true); video_writer.open(video_filename, CV_FOURCC(output_codec[0], output_codec[1], output_codec[2], output_codec[3]), fps_vid_out, vis_track.size(), true);