parameter_optimization_tool.h

Go to the documentation of this file.

#ifndef PARAMETER_OPTIMIZATION_TOOL_H
#define PARAMETER_OPTIMIZATION_TOOL_H

#include <tuple>
#include "evaluation_summary.h"

class ParameterOptimizationTool {
public:
    
    static const int FLOAT_PARAMETER = 1;
    static const int INTEGER_PARAMETER = 2;
    
    ParameterOptimizationTool(boost::filesystem::path img_directory, 
            boost::filesystem::path gt_directory, boost::filesystem::path base_directory,
            std::string command_line, std::string command_line_parameters);
    
    void addPostProcessingCommandLine(std::string command_line);
    
    void addSuperpixelTolerance(int superpixels, int tolerance);
    
    void addFloatParameter(std::string name, std::string parameter, 
            std::vector<float> values);
    
    void addIntegerParameter(std::string name, std::string parameter, 
            std::vector<int> values);
    
    void useDepth(boost::filesystem::path depth_directory);
    
    void useIntrinsics(boost::filesystem::path intrinsics_directory);
    
    void setVerbose(std::ostream &stream = std::cout);
    
    int numCombinations();
    
    void optimize(float weigh = 0.5, float weight_ue = 0.33, float weight_co = 0.33);
    
protected:
    
    void cleanUp(const boost::filesystem::path &sp_directory);
    
    std::string command_line;
    std::string command_line_parameters;
    std::string post_processing_command_line;
    
    boost::filesystem::path img_directory;
    boost::filesystem::path gt_directory;
    boost::filesystem::path depth_directory;
    boost::filesystem::path intrinsics_directory;
    boost::filesystem::path base_directory;
    
    EvaluationSummary::EvaluationMetrics evaluation_metrics;
    EvaluationSummary::EvaluationStatistics evaluation_statistics;
    
    std::vector< std::tuple<std::string, std::string, int, int> > parameters;
    std::vector< std::tuple<std::vector<int>, int, int, int> > integer_parameters;
    std::vector< std::tuple<std::vector<float>, int, float, float> > float_parameters;
    
    int superpixels_min;
    int superpixels_max;
    
};

#endif  /* PARAMETER_OPTIMIZATION_TOOL_H */