ZF/OpenFace
1
0
Fork 0
mirror of https://gitcode.com/gh_mirrors/ope/OpenFace.git synced 2025-12-30 04:52:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

do not force a flush when writing a line to the CSV file (#911)

Browse Source 
* do not force a flush when writing a line to the CSV file

* remove spaces from CSV files
This commit is contained in:
Torsten Wörtwein
2020-10-14 06:31:19 -04:00
committed by GitHub
parent fc033e00e8
commit 369a92caa7
2 changed files with 47 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
lib/local/FaceAnalyser/src/FaceAnalyser.cpp
View File

@@ -1346,7 +1346,7 @@ void FaceAnalyser::PostprocessOutputFile(std::string output_file)
outfile << std::fixed;
outfile << std::noshowpoint;
outfile << output_file_contents[0].c_str() << std::endl;
outfile << output_file_contents[0].c_str() << "\n";
// Write the contents
for (int i = 1; i < (int)output_file_contents.size(); ++i)
@@ -1363,20 +1363,20 @@ void FaceAnalyser::PostprocessOutputFile(std::string output_file)
{
if (t - begin_ind < num_reg)
{
outfile << ", " << predictions_reg[inds_reg[t - begin_ind]].second[i - 1];
outfile << "," << predictions_reg[inds_reg[t - begin_ind]].second[i - 1];
}
else
{
outfile << ", " << predictions_class[inds_class[t - begin_ind - num_reg]].second[i - 1];
outfile << "," << predictions_class[inds_class[t - begin_ind - num_reg]].second[i - 1];
}
}
else
{
rtrim(tokens[t]);
outfile << ", " << tokens[t];
outfile << "," << tokens[t];
}
}
outfile << std::endl;
outfile << "\n";
}

84
lib/local/Utilities/src/RecorderCSV.cpp
View File

@@ -71,54 +71,54 @@ bool RecorderCSV::Open(std::string output_file_name, bool is_sequence, bool outp
// Different headers if we are writing out the results on a sequence or an individual image
if(this->is_sequence)
{
output_file << "frame, face_id, timestamp, confidence, success";
output_file << "frame,face_id,timestamp,confidence,success";
}
else
{
output_file << "face, confidence";
output_file << "face,confidence";
}
if (output_gaze)
{
output_file << ", gaze_0_x, gaze_0_y, gaze_0_z, gaze_1_x, gaze_1_y, gaze_1_z, gaze_angle_x, gaze_angle_y";
output_file << ",gaze_0_x,gaze_0_y,gaze_0_z,gaze_1_x,gaze_1_y,gaze_1_z,gaze_angle_x,gaze_angle_y";
for (int i = 0; i < num_eye_landmarks; ++i)
{
output_file << ", eye_lmk_x_" << i;
output_file << ",eye_lmk_x_" << i;
}
for (int i = 0; i < num_eye_landmarks; ++i)
{
output_file << ", eye_lmk_y_" << i;
output_file << ",eye_lmk_y_" << i;
}
for (int i = 0; i < num_eye_landmarks; ++i)
{
output_file << ", eye_lmk_X_" << i;
output_file << ",eye_lmk_X_" << i;
}
for (int i = 0; i < num_eye_landmarks; ++i)
{
output_file << ", eye_lmk_Y_" << i;
output_file << ",eye_lmk_Y_" << i;
}
for (int i = 0; i < num_eye_landmarks; ++i)
{
output_file << ", eye_lmk_Z_" << i;
output_file << ",eye_lmk_Z_" << i;
}
}
if (output_pose)
{
output_file << ", pose_Tx, pose_Ty, pose_Tz, pose_Rx, pose_Ry, pose_Rz";
output_file << ",pose_Tx,pose_Ty,pose_Tz,pose_Rx,pose_Ry,pose_Rz";
}
if (output_2D_landmarks)
{
for (int i = 0; i < num_face_landmarks; ++i)
{
output_file << ", x_" << i;
output_file << ",x_" << i;
}
for (int i = 0; i < num_face_landmarks; ++i)
{
output_file << ", y_" << i;
output_file << ",y_" << i;
}
}
@@ -126,25 +126,25 @@ bool RecorderCSV::Open(std::string output_file_name, bool is_sequence, bool outp
{
for (int i = 0; i < num_face_landmarks; ++i)
{
output_file << ", X_" << i;
output_file << ",X_" << i;
}
for (int i = 0; i < num_face_landmarks; ++i)
{
output_file << ", Y_" << i;
output_file << ",Y_" << i;
}
for (int i = 0; i < num_face_landmarks; ++i)
{
output_file << ", Z_" << i;
output_file << ",Z_" << i;
}
}
// Outputting model parameters (rigid and non-rigid), the first parameters are the 6 rigid shape parameters, they are followed by the non rigid shape parameters
if (output_model_params)
{
output_file << ", p_scale, p_rx, p_ry, p_rz, p_tx, p_ty";
output_file << ",p_scale,p_rx,p_ry,p_rz,p_tx,p_ty";
for (int i = 0; i < num_model_modes; ++i)
{
output_file << ", p_" << i;
output_file << ",p_" << i;
}
}
@@ -153,17 +153,17 @@ bool RecorderCSV::Open(std::string output_file_name, bool is_sequence, bool outp
std::sort(this->au_names_reg.begin(), this->au_names_reg.end());
for (std::string reg_name : this->au_names_reg)
{
output_file << ", " << reg_name << "_r";
output_file << "," << reg_name << "_r";
}
std::sort(this->au_names_class.begin(), this->au_names_class.end());
for (std::string class_name : this->au_names_class)
{
output_file << ", " << class_name << "_c";
output_file << "," << class_name << "_c";
}
}
output_file << std::endl;
output_file << "\n";
return true;
@@ -188,54 +188,54 @@ void RecorderCSV::WriteLine(int face_id, int frame_num, double time_stamp, bool
{
output_file << std::setprecision(3);
output_file << frame_num << ", " << face_id << ", " << time_stamp;
output_file << frame_num << "," << face_id << "," << time_stamp;
output_file << std::setprecision(2);
output_file << ", " << landmark_confidence;
output_file << "," << landmark_confidence;
output_file << std::setprecision(0);
output_file << ", " << landmark_detection_success;
output_file << "," << landmark_detection_success;
}
else
{
output_file << std::setprecision(3);
output_file << face_id << ", " << landmark_confidence;
output_file << face_id << "," << landmark_confidence;
}
// Output the estimated gaze
if (output_gaze)
{
output_file << std::setprecision(6);
output_file << ", " << gazeDirection0.x << ", " << gazeDirection0.y << ", " << gazeDirection0.z
<< ", " << gazeDirection1.x << ", " << gazeDirection1.y << ", " << gazeDirection1.z;
output_file << "," << gazeDirection0.x << "," << gazeDirection0.y << "," << gazeDirection0.z
<< "," << gazeDirection1.x << "," << gazeDirection1.y << "," << gazeDirection1.z;
// Output gaze angle (same format as head pose angle)
output_file << std::setprecision(3);
output_file << ", " << gaze_angle[0] << ", " << gaze_angle[1];
output_file << "," << gaze_angle[0] << "," << gaze_angle[1];
// Output the 2D eye landmarks
output_file << std::setprecision(1);
for (auto eye_lmk : eye_landmarks2d)
{
output_file << ", " << eye_lmk.x;
output_file << "," << eye_lmk.x;
}
for (auto eye_lmk : eye_landmarks2d)
{
output_file << ", " << eye_lmk.y;
output_file << "," << eye_lmk.y;
}
// Output the 3D eye landmarks
for (auto eye_lmk : eye_landmarks3d)
{
output_file << ", " << eye_lmk.x;
output_file << "," << eye_lmk.x;
}
for (auto eye_lmk : eye_landmarks3d)
{
output_file << ", " << eye_lmk.y;
output_file << "," << eye_lmk.y;
}
for (auto eye_lmk : eye_landmarks3d)
{
output_file << ", " << eye_lmk.z;
output_file << "," << eye_lmk.z;
}
}
@@ -243,9 +243,9 @@ void RecorderCSV::WriteLine(int face_id, int frame_num, double time_stamp, bool
if (output_pose)
{
output_file << std::setprecision(1);
output_file << ", " << pose_estimate[0] << ", " << pose_estimate[1] << ", " << pose_estimate[2];
output_file << "," << pose_estimate[0] << "," << pose_estimate[1] << "," << pose_estimate[2];
output_file << std::setprecision(3);
output_file << ", " << pose_estimate[3] << ", " << pose_estimate[4] << ", " << pose_estimate[5];
output_file << "," << pose_estimate[3] << "," << pose_estimate[4] << "," << pose_estimate[5];
}
// Output the detected 2D facial landmarks
@@ -255,7 +255,7 @@ void RecorderCSV::WriteLine(int face_id, int frame_num, double time_stamp, bool
// Output the 2D eye landmarks
for (auto lmk : landmarks_2D)
{
output_file << ", " << lmk;
output_file << "," << lmk;
}
}
@@ -266,7 +266,7 @@ void RecorderCSV::WriteLine(int face_id, int frame_num, double time_stamp, bool
// Output the 2D eye landmarks
for (auto lmk : landmarks_3D)
{
output_file << ", " << lmk;
output_file << "," << lmk;
}
}
@@ -275,12 +275,12 @@ void RecorderCSV::WriteLine(int face_id, int frame_num, double time_stamp, bool
output_file.precision(3);
for (int i = 0; i < 6; ++i)
{
output_file << ", " << rigid_shape_params[i];
output_file << "," << rigid_shape_params[i];
}
// Output the non_rigid shape parameters
for (auto lmk : pdm_model_params)
{
output_file << ", " << lmk;
output_file << "," << lmk;
}
}
@@ -295,7 +295,7 @@ void RecorderCSV::WriteLine(int face_id, int frame_num, double time_stamp, bool
{
if (au_name.compare(au_reg.first) == 0)
{
output_file << ", " << au_reg.second;
output_file << "," << au_reg.second;
break;
}
}
@@ -305,7 +305,7 @@ void RecorderCSV::WriteLine(int face_id, int frame_num, double time_stamp, bool
{
for (size_t p = 0; p < au_names_reg.size(); ++p)
{
output_file << ", 0";
output_file << ",0";
}
}
@@ -317,7 +317,7 @@ void RecorderCSV::WriteLine(int face_id, int frame_num, double time_stamp, bool
{
if (au_name.compare(au_class.first) == 0)
{
output_file << ", " << au_class.second;
output_file << "," << au_class.second;
break;
}
}
@@ -327,11 +327,11 @@ void RecorderCSV::WriteLine(int face_id, int frame_num, double time_stamp, bool
{
for (size_t p = 0; p < au_names_class.size(); ++p)
{
output_file << ", 0";
output_file << ",0";
}
}
}
output_file << std::endl;
output_file << "\n";
}
// Closing the file and cleaning up