OnnxConfig.h

Go to the documentation of this file.

 
#ifndef _SEIMPLEMENTATION_PLUGIN_ONNXCONFIG_H_
#define _SEIMPLEMENTATION_PLUGIN_ONNXCONFIG_H_
 
#include <Configuration/Configuration.h>
 
namespace SourceXtractor {

class OnnxConfig: public Euclid::Configuration::Configuration {
public:

  explicit OnnxConfig(long manager_id);

  ~OnnxConfig() override = default;

  std::map<std::string, OptionDescriptionList> getProgramOptions() override;

  void initialize(const UserValues& args) override;

  const std::vector<std::string>& getModels() const {
    return m_onnx_model_paths;
  }
 
private:
  std::vector<std::string> m_onnx_model_paths;
};
 
} // end of namespace SourceXtractor
 
#endif // _SEIMPLEMENTATION_PLUGIN_ONNXCONFIG_H_