SourceXtractorPlusPlus
0.16
Please provide a description of the project.
SEImplementation
src
lib
Plugin
Onnx
OnnxConfig.cpp
Go to the documentation of this file.
1
18
#include "
SEImplementation/Plugin/Onnx/OnnxConfig.h
"
19
#include <boost/program_options.hpp>
20
21
namespace
po = boost::program_options;
22
using namespace
Euclid::Configuration
;
23
24
namespace
SourceXtractor
{
25
26
static
const
std::string
ML_MEASUREMENT_MODEL
{
"ml-measurement-model"
};
27
28
OnnxConfig::OnnxConfig(
long
manager_id) :
Configuration
(manager_id) {
29
}
30
31
auto
OnnxConfig::getProgramOptions
() ->
std::map<std::string, OptionDescriptionList>
{
32
return
{{
"ONNX"
, {
33
{
ML_MEASUREMENT_MODEL
.
c_str
(), po::value<std::vector<std::string>>()->multitoken(),
34
"ONNX-format models for machine learning based measurements"
}
35
}}};
36
}
37
38
void
OnnxConfig::initialize
(
const
Configuration::Configuration::UserValues& args) {
39
auto
i = args.find(
ML_MEASUREMENT_MODEL
);
40
if
(i != args.end()) {
41
m_onnx_model_paths
= i->second.as<
std::vector<std::string>
>();
42
}
43
}
44
45
}
// end of namespace SourceXtractor
OnnxConfig.h
std::string
std::string::c_str
T c_str(T... args)
Euclid::Configuration::Configuration
SourceXtractor::OnnxConfig::getProgramOptions
std::map< std::string, OptionDescriptionList > getProgramOptions() override
Definition:
OnnxConfig.cpp:31
SourceXtractor::OnnxConfig::initialize
void initialize(const UserValues &args) override
Definition:
OnnxConfig.cpp:38
SourceXtractor::OnnxConfig::m_onnx_model_paths
std::vector< std::string > m_onnx_model_paths
Definition:
OnnxConfig.h:49
std::map
Euclid::Configuration
SourceXtractor
Definition:
Aperture.h:30
SourceXtractor::ML_MEASUREMENT_MODEL
static const std::string ML_MEASUREMENT_MODEL
Definition:
OnnxConfig.cpp:26
std::vector< std::string >
Generated by
1.9.1