met::METNetSigHandler Class Reference

#include <METNetSigHandler.h>

Collaboration diagram for met::METNetSigHandler:

Public Member Functions
	METNetSigHandler (const std::string &modelName)
virtual	~METNetSigHandler ()
int	initialize ()
int	getReqSize () const
std::vector< float >	predict (std::vector< float > inputs) const

Private Member Functions
	METNetSigHandler ()

Private Attributes
std::string	m_modelName
std::string	m_modelPath
size_t	m_numInputs
size_t	m_numOutputs
std::vector< int64_t >	m_inputDims
std::vector< int64_t >	m_outputDims
std::vector< const char * >	m_graphInputNames
std::vector< const char * >	m_graphOutputNames
Ort::Env	m_onnxEnv
Ort::SessionOptions	m_onnxSessionOptions
Ort::AllocatorWithDefaultOptions	m_onnxAllocator
std::unique_ptr< Ort::Session >	m_onnxSession

Detailed Description

Definition at line 24 of file METNetSigHandler.h.

Constructor & Destructor Documentation

METNetSigHandler() [1/2]

met::METNetSigHandler::METNetSigHandler

(

const std::string &

modelName

)

Definition at line 20 of file METNetSigHandler.cxx.

                                                               :
    m_modelName(modelName),
    m_numInputs(1),
    m_numOutputs(4),
    m_inputDims({1,77}),
    m_outputDims({1,1}), // m_outputDims({1,2}),
    m_graphInputNames({"input"}),
    m_graphOutputNames({"MET_x", "MET_y", "Sigma_x", "Sigma_y"}){}

~METNetSigHandler()

met::METNetSigHandler::~METNetSigHandler ( )

virtual

Definition at line 29 of file METNetSigHandler.cxx.

{ }

METNetSigHandler() [2/2]

met::METNetSigHandler::METNetSigHandler ( )

private

Member Function Documentation

getReqSize()

int met::METNetSigHandler::getReqSize

(

)

const

Definition at line 50 of file METNetSigHandler.cxx.

                                         {
    // Returns the required size of the inputs for the network
    return static_cast<int>(m_inputDims[1]);
  }

initialize()

int met::METNetSigHandler::initialize

(

)

Definition at line 31 of file METNetSigHandler.cxx.

                                   {
 
    // Use the path resolver to find the location of the network .onnx file
    // m_modelPath = PathResolverFindCalibFile(m_modelName);
    // m_modelPath = "ggHyydAnalysis/model_mc23.onnx";
    m_modelPath = PathResolverFindCalibFile(m_modelName);
    // m_modelPath = "../source/ggHyydAnalysis/share/model_mc23.onnx";
    if (m_modelPath == "") return 1;
 
    // Use the default ONNX session settings for 1 CPU thread
    m_onnxSessionOptions.SetIntraOpNumThreads(1);
    m_onnxSessionOptions.SetGraphOptimizationLevel(ORT_ENABLE_BASIC);
 
    // Initialise the ONNX environment and session using the above options and the model name
    m_onnxSession = std::make_unique<Ort::Session>(m_onnxEnv, m_modelPath.c_str(), m_onnxSessionOptions);
 
    return 0;
  }

predict()

std::vector< float > met::METNetSigHandler::predict

(

std::vector< float >

inputs

)

const

Definition at line 55 of file METNetSigHandler.cxx.

                                                                          {
    // This method passes a input vector through the neural network and returns its estimate
    // It requires conversions to onnx type tensors and back
 
    // Create a CPU tensor to be used as input
    auto memory_info = Ort::MemoryInfo::CreateCpu(OrtArenaAllocator, OrtMemTypeDefault);
    Ort::Value input_tensor = Ort::Value::CreateTensor<float>( memory_info,
                                                               inputs.data(),
                                                               inputs.size(),
                                                               m_inputDims.data(),
                                                               m_inputDims.size() );
                                                               
 
    std::vector<Ort::Value> ort_outputs = m_onnxSession->Run(
      Ort::RunOptions{nullptr},
      m_graphInputNames.data(), &input_tensor, m_numInputs,
      m_graphOutputNames.data(), m_numOutputs
    );
 
    // Extract output values and convert from GeV to MeV
    std::vector<float> outputs;
    for (size_t i = 0; i < ort_outputs.size(); ++i) {
      const float* data = ort_outputs[i].GetTensorData<float>();
      outputs.push_back(data[0] * 1000.0f);
    }
    return outputs;
 
  }

Member Data Documentation

m_graphInputNames

std::vector<const char *> met::METNetSigHandler::m_graphInputNames

private

Definition at line 53 of file METNetSigHandler.h.

m_graphOutputNames

std::vector<const char *> met::METNetSigHandler::m_graphOutputNames

private

Definition at line 54 of file METNetSigHandler.h.

m_inputDims

std::vector<int64_t> met::METNetSigHandler::m_inputDims

private

Definition at line 51 of file METNetSigHandler.h.

m_modelName

std::string met::METNetSigHandler::m_modelName

private

Definition at line 45 of file METNetSigHandler.h.

m_modelPath

std::string met::METNetSigHandler::m_modelPath

private

Definition at line 46 of file METNetSigHandler.h.

m_numInputs

size_t met::METNetSigHandler::m_numInputs

private

Definition at line 49 of file METNetSigHandler.h.

m_numOutputs

size_t met::METNetSigHandler::m_numOutputs

private

Definition at line 50 of file METNetSigHandler.h.

m_onnxAllocator

Ort::AllocatorWithDefaultOptions met::METNetSigHandler::m_onnxAllocator

private

Definition at line 59 of file METNetSigHandler.h.

m_onnxEnv

Ort::Env met::METNetSigHandler::m_onnxEnv

private

Definition at line 57 of file METNetSigHandler.h.

m_onnxSession

std::unique_ptr<Ort::Session> met::METNetSigHandler::m_onnxSession

private

Definition at line 60 of file METNetSigHandler.h.

m_onnxSessionOptions

Ort::SessionOptions met::METNetSigHandler::m_onnxSessionOptions

private

Definition at line 58 of file METNetSigHandler.h.

m_outputDims

std::vector<int64_t> met::METNetSigHandler::m_outputDims

private

Definition at line 52 of file METNetSigHandler.h.

---

The documentation for this class was generated from the following files:

• METNetSigHandler.h
• METNetSigHandler.cxx