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