EvaluateModel.h

Go to the documentation of this file.

// Dear emacs, this is -*- c++ -*-
// Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
#ifndef ATHEXONNXRUNTIME_EVALUATEMODEL_H
#define ATHEXONNXRUNTIME_EVALUATEMODEL_H
 
// Local include(s).
#include "AthOnnxInterfaces/IOnnxRuntimeInferenceTool.h"
 
// Framework include(s).
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "GaudiKernel/ServiceHandle.h"
 
// Onnx Runtime include(s).
#include <onnxruntime_cxx_api.h>
 
// System include(s).
#include <memory>
#include <string>
#include <vector>
 
namespace AthOnnx {
 
   class EvaluateModel: public AthReentrantAlgorithm {
 
   public:
      using AthReentrantAlgorithm::AthReentrantAlgorithm;
 
 
      virtual StatusCode initialize() override;
      virtual StatusCode execute( const EventContext& ctx ) const override;
      virtual StatusCode finalize() override;
 
 
   private:
 
      Gaudi::Property< std::string > m_pixelFileName{ this, "InputDataPixel",
         "dev/MLTest/2020-03-31/t10k-images-idx3-ubyte",
         "Name of the input pixel file to load" };
 
      Gaudi::Property<int> m_batchSize {this, "BatchSize", 1, "No. of elements/example in a batch"};
 
      ToolHandle< IOnnxRuntimeInferenceTool >  m_onnxTool{
         this, "ORTInferenceTool", "AthOnnx::OnnxRuntimeInferenceTool"
      };
 
      std::vector<std::vector<std::vector<float>>> m_input_tensor_values_notFlat;
 
   }; // class EvaluateModel
 
} // namespace AthOnnx
 
#endif // ATHEXONNXRUNTIME_EVALUATEMODEL_H