OnnxConfig.cpp

Go to the documentation of this file.

 
#include "SEImplementation/Plugin/Onnx/OnnxConfig.h"
#include <boost/program_options.hpp>
 
namespace po = boost::program_options;
using namespace Euclid::Configuration;
 
namespace SourceXtractor {
 
static const std::string ML_MEASUREMENT_MODEL{"ml-measurement-model"};
 
OnnxConfig::OnnxConfig(long manager_id) : Configuration(manager_id) {
}
 
auto OnnxConfig::getProgramOptions() -> std::map<std::string, OptionDescriptionList> {
  return {{"ONNX", {
    {ML_MEASUREMENT_MODEL.c_str(), po::value<std::vector<std::string>>()->multitoken(),
        "ONNX-format models for machine learning based measurements"}
  }}};
}
 
void OnnxConfig::initialize(const Configuration::Configuration::UserValues& args) {
  auto i = args.find(ML_MEASUREMENT_MODEL);
  if (i != args.end()) {
    m_onnx_model_paths = i->second.as<std::vector<std::string>>();
  }
}
 
} // end of namespace SourceXtractor