isSequence fix.

This commit is contained in:
Tadas Baltrusaitis 2017-11-07 07:46:25 +00:00
parent 4707b0a687
commit 71407c9067
3 changed files with 6 additions and 3 deletions

View file

@ -57,7 +57,7 @@ namespace Recorder
RecorderCSV(); RecorderCSV();
// Opening the file and preparing the header for it // Opening the file and preparing the header for it
bool Open(std::string output_file_name, bool output_2D_landmarks, bool output_3D_landmarks, bool output_model_params, bool output_pose, bool output_AUs, bool output_gaze, bool Open(std::string output_file_name, bool is_sequence, bool output_2D_landmarks, bool output_3D_landmarks, bool output_model_params, bool output_pose, bool output_AUs, bool output_gaze,
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); 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);
// Closing the file and cleaning up // Closing the file and cleaning up

View file

@ -47,14 +47,17 @@ RecorderCSV::RecorderCSV():output_file(){};
// TODO the other 4 constructors + destructors? // TODO the other 4 constructors + destructors?
// Opening the file and preparing the header for it // Opening the file and preparing the header for it
bool RecorderCSV::Open(std::string output_file_name, bool output_2D_landmarks, bool output_3D_landmarks, bool output_model_params, bool output_pose, bool output_AUs, bool output_gaze, bool RecorderCSV::Open(std::string output_file_name, bool is_sequence, bool output_2D_landmarks, bool output_3D_landmarks, bool output_model_params, bool output_pose, bool output_AUs, bool output_gaze,
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) 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)
{ {
output_file.open(output_file_name, std::ios_base::out); output_file.open(output_file_name, std::ios_base::out);
if (!output_file.is_open()) if (!output_file.is_open())
return false; return false;
this->is_sequence = is_sequence;
// Set up what we are recording // Set up what we are recording
this->output_2D_landmarks = output_2D_landmarks; this->output_2D_landmarks = output_2D_landmarks;
this->output_3D_landmarks = output_3D_landmarks; this->output_3D_landmarks = output_3D_landmarks;

View file

@ -172,7 +172,7 @@ void RecorderOpenFace::WriteObservation()
std::sort(au_names_reg.begin(), au_names_reg.end()); std::sort(au_names_reg.begin(), au_names_reg.end());
csv_recorder.Open(csv_filename, params.output2DLandmarks(), params.output3DLandmarks(), params.outputPDMParams(), params.outputPose(), csv_recorder.Open(csv_filename, params.isSequence(), params.output2DLandmarks(), params.output3DLandmarks(), params.outputPDMParams(), params.outputPose(),
params.outputAUs(), params.outputGaze(), num_face_landmarks, num_model_modes, num_eye_landmarks, au_names_class, au_names_reg); params.outputAUs(), params.outputGaze(), num_face_landmarks, num_model_modes, num_eye_landmarks, au_names_class, au_names_reg);
} }