sustaining_gazes/lib/local/LandmarkDetector/src/LandmarkDetectorParameters.cpp

274 lines
7.2 KiB
C++
Raw Normal View History

2016-04-28 21:40:36 +02:00
///////////////////////////////////////////////////////////////////////////////
// Copyright (C) 2017, Carnegie Mellon University and University of Cambridge,
2016-04-28 21:40:36 +02:00
// all rights reserved.
//
// ACADEMIC OR NON-PROFIT ORGANIZATION NONCOMMERCIAL RESEARCH USE ONLY
//
// BY USING OR DOWNLOADING THE SOFTWARE, YOU ARE AGREEING TO THE TERMS OF THIS LICENSE AGREEMENT.
// IF YOU DO NOT AGREE WITH THESE TERMS, YOU MAY NOT USE OR DOWNLOAD THE SOFTWARE.
//
// License can be found in OpenFace-license.txt
2016-04-28 21:40:36 +02:00
//
// * Any publications arising from the use of this software, including but
// not limited to academic journal and conference publications, technical
// reports and manuals, must cite at least one of the following works:
//
// OpenFace: an open source facial behavior analysis toolkit
// Tadas Baltru<72>aitis, Peter Robinson, and Louis-Philippe Morency
// in IEEE Winter Conference on Applications of Computer Vision, 2016
//
// Rendering of Eyes for Eye-Shape Registration and Gaze Estimation
// Erroll Wood, Tadas Baltru<72>aitis, Xucong Zhang, Yusuke Sugano, Peter Robinson, and Andreas Bulling
// in IEEE International. Conference on Computer Vision (ICCV), 2015
//
// Cross-dataset learning and person-speci?c normalisation for automatic Action Unit detection
// Tadas Baltru<72>aitis, Marwa Mahmoud, and Peter Robinson
// in Facial Expression Recognition and Analysis Challenge,
// IEEE International Conference on Automatic Face and Gesture Recognition, 2015
//
// Constrained Local Neural Fields for robust facial landmark detection in the wild.
// Tadas Baltru<72>aitis, Peter Robinson, and Louis-Philippe Morency.
// in IEEE Int. Conference on Computer Vision Workshops, 300 Faces in-the-Wild Challenge, 2013.
//
///////////////////////////////////////////////////////////////////////////////
#include "stdafx.h"
#include "LandmarkDetectorParameters.h"
// Boost includes
#include <filesystem.hpp>
#include <filesystem/fstream.hpp>
// System includes
#include <sstream>
#include <iostream>
#include <cstdlib>
2016-04-28 21:40:36 +02:00
#ifndef CONFIG_DIR
#define CONFIG_DIR "~"
#endif
2016-04-28 21:40:36 +02:00
using namespace std;
using namespace LandmarkDetector;
FaceModelParameters::FaceModelParameters()
{
// initialise the default values
init();
check_model_path();
2016-04-28 21:40:36 +02:00
}
FaceModelParameters::FaceModelParameters(vector<string> &arguments)
{
// initialise the default values
init();
// First element is reserved for the executable location (useful for finding relative model locs)
boost::filesystem::path root = boost::filesystem::path(arguments[0]).parent_path();
bool* valid = new bool[arguments.size()];
valid[0] = true;
for (size_t i = 1; i < arguments.size(); ++i)
{
valid[i] = true;
if (arguments[i].compare("-mloc") == 0)
{
string model_loc = arguments[i + 1];
model_location = model_loc;
valid[i] = false;
valid[i + 1] = false;
i++;
}
if (arguments[i].compare("-fdloc") ==0)
{
string face_detector_loc = arguments[i + 1];
face_detector_location = face_detector_loc;
curr_face_detector = HAAR_DETECTOR;
valid[i] = false;
valid[i + 1] = false;
i++;
}
2016-04-28 21:40:36 +02:00
if (arguments[i].compare("-sigma") == 0)
{
stringstream data(arguments[i + 1]);
data >> sigma;
valid[i] = false;
valid[i + 1] = false;
i++;
}
else if (arguments[i].compare("-w_reg") == 0)
{
stringstream data(arguments[i + 1]);
data >> weight_factor;
valid[i] = false;
valid[i + 1] = false;
i++;
}
else if (arguments[i].compare("-reg") == 0)
{
stringstream data(arguments[i + 1]);
data >> reg_factor;
valid[i] = false;
valid[i + 1] = false;
i++;
}
else if (arguments[i].compare("-multi_view") == 0)
{
stringstream data(arguments[i + 1]);
int m_view;
data >> m_view;
multi_view = (bool)(m_view != 0);
valid[i] = false;
valid[i + 1] = false;
i++;
}
else if (arguments[i].compare("-validate_detections") == 0)
{
stringstream data(arguments[i + 1]);
int v_det;
data >> v_det;
validate_detections = (bool)(v_det != 0);
valid[i] = false;
valid[i + 1] = false;
i++;
}
else if (arguments[i].compare("-n_iter") == 0)
{
stringstream data(arguments[i + 1]);
data >> num_optimisation_iteration;
valid[i] = false;
valid[i + 1] = false;
i++;
}
else if (arguments[i].compare("-q") == 0)
{
quiet_mode = true;
valid[i] = false;
}
else if (arguments[i].compare("-wild") == 0)
{
// For in the wild fitting these parameters are suitable
window_sizes_init = vector<int>(4);
window_sizes_init[0] = 15; window_sizes_init[1] = 13; window_sizes_init[2] = 11; window_sizes_init[3] = 9;
sigma = 1.25;
reg_factor = 35;
weight_factor = 2.5;
num_optimisation_iteration = 10;
valid[i] = false;
// For in-the-wild images use an in-the wild detector
curr_face_detector = HOG_SVM_DETECTOR;
}
}
2016-09-06 18:32:33 +02:00
for (int i = (int)arguments.size() - 1; i >= 0; --i)
2016-04-28 21:40:36 +02:00
{
if (!valid[i])
{
arguments.erase(arguments.begin() + i);
}
}
check_model_path(root.string());
}
void FaceModelParameters::check_model_path(const std::string& root)
{
2016-04-28 21:40:36 +02:00
// Make sure model_location is valid
// First check working directory, then the executable's directory, then the config path set by the build process.
boost::filesystem::path config_path = boost::filesystem::path(CONFIG_DIR);
boost::filesystem::path model_path = boost::filesystem::path(model_location);
boost::filesystem::path root_path = boost::filesystem::path(root);
if (boost::filesystem::exists(model_path))
2016-04-28 21:40:36 +02:00
{
model_location = model_path.string();
}
else if (boost::filesystem::exists(root_path/model_path))
{
model_location = (root_path/model_path).string();
}
else if (boost::filesystem::exists(config_path/model_path))
{
model_location = (config_path/model_path).string();
}
else
{
std::cout << "Could not find the landmark detection model to load" << std::endl;
2016-04-28 21:40:36 +02:00
}
}
void FaceModelParameters::init()
{
// number of iterations that will be performed at each scale
num_optimisation_iteration = 5;
// using an external face checker based on SVM
validate_detections = true;
// Using hierarchical refinement by default (can be turned off)
refine_hierarchical = true;
// Refining parameters by default
refine_parameters = true;
window_sizes_small = vector<int>(4);
window_sizes_init = vector<int>(4);
// For fast tracking
window_sizes_small[0] = 0;
window_sizes_small[1] = 9;
window_sizes_small[2] = 7;
window_sizes_small[3] = 5;
// Just for initialisation
window_sizes_init.at(0) = 11;
window_sizes_init.at(1) = 9;
window_sizes_init.at(2) = 7;
window_sizes_init.at(3) = 5;
face_template_scale = 0.3;
// Off by default (as it might lead to some slight inaccuracies in slowly moving faces)
use_face_template = false;
// For first frame use the initialisation
window_sizes_current = window_sizes_init;
model_location = "model/main_clnf_general.txt";
sigma = 1.5;
reg_factor = 25;
weight_factor = 0; // By default do not use NU-RLMS for videos as it does not work as well for them
validation_boundary = 0.725;
2016-04-28 21:40:36 +02:00
limit_pose = true;
multi_view = false;
reinit_video_every = 4;
// Face detection
face_detector_location = "classifiers/haarcascade_frontalface_alt.xml";
quiet_mode = false;
// By default use HOG SVM
curr_face_detector = HOG_SVM_DETECTOR;
}