SourceXtractorPlusPlus
0.12
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
ONNX_MODEL
{
"onnx-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
{
ONNX_MODEL
.
c_str
(), po::value<std::vector<std::string>>()->multitoken(),
"ONNX models"
}
34
}}};
35
}
36
37
void
OnnxConfig::initialize
(
const
Configuration::Configuration::UserValues& args) {
38
auto
i = args.find(
ONNX_MODEL
);
39
if
(i != args.end()) {
40
m_onnx_model_paths
= i->second.as<
std::vector<std::string>
>();
41
}
42
}
43
44
}
// end of namespace SourceXtractor
Euclid::Configuration
Euclid::Configuration::Configuration
SourceXtractor::OnnxConfig::initialize
void initialize(const UserValues &args) override
Definition:
OnnxConfig.cpp:37
SourceXtractor::ONNX_MODEL
static const std::string ONNX_MODEL
Definition:
OnnxConfig.cpp:26
std::map
STL class.
SourceXtractor::OnnxConfig::m_onnx_model_paths
std::vector< std::string > m_onnx_model_paths
Definition:
OnnxConfig.h:49
std::string
STL class.
SourceXtractor::OnnxConfig::getProgramOptions
std::map< std::string, OptionDescriptionList > getProgramOptions() override
Definition:
OnnxConfig.cpp:31
SourceXtractor
Definition:
Aperture.h:30
std::vector< std::string >
std::string::c_str
T c_str(T... args)
OnnxConfig.h
Generated by
1.8.14