nnet::TfModelInterface Class Reference

#include <PointIdAlg.h>

Inheritance diagram for nnet::TfModelInterface:

Public Member Functions
	TfModelInterface (const char *modelFileName)
std::vector< std::vector< float > >	Run (std::vector< std::vector< std::vector< float >>> const &inps, int samples=-1) override
std::vector< float >	Run (std::vector< std::vector< float >> const &inp2d) override
Public Member Functions inherited from nnet::ModelInterface
virtual	~ModelInterface ()

Private Attributes
std::unique_ptr< tf::Graph >	g

Additional Inherited Members
Protected Member Functions inherited from nnet::ModelInterface
std::string	findFile (const char *fileName) const

Detailed Description

Definition at line 75 of file PointIdAlg.h.

Constructor & Destructor Documentation

nnet::TfModelInterface::TfModelInterface

(

const char *

modelFileName

)

Definition at line 91 of file PointIdAlg.cxx.

{
  g = tf::Graph::create(nnet::ModelInterface::findFile(modelFileName).c_str(),
                        {"cnn_output", "_netout"});
  if (!g) { throw art::Exception(art::errors::Unknown) << "TF model failed."; }

  mf::LogInfo("TfModelInterface") << "TF model loaded.";
}

Member Function Documentation

std::vector< std::vector< float > > nnet::TfModelInterface::Run ( std::vector< std::vector< std::vector< float >>> const &  inps, int  samples = -1  )

overridevirtual

Reimplemented from nnet::ModelInterface.

Definition at line 102 of file PointIdAlg.cxx.

{
  if ((samples == 0) || inps.empty() || inps.front().empty() || inps.front().front().empty())
    return std::vector<std::vector<float>>();

  if ((samples == -1) || (samples > (long long int)inps.size())) { samples = inps.size(); }

  long long int rows = inps.front().size(), cols = inps.front().front().size();

  tensorflow::Tensor _x(tensorflow::DT_FLOAT, tensorflow::TensorShape({samples, rows, cols, 1}));
  auto input_map = _x.tensor<float, 4>();
  for (long long int s = 0; s < samples; ++s) {
    const auto& sample = inps[s];
    for (long long int r = 0; r < rows; ++r) {
      const auto& row = sample[r];
      for (long long int c = 0; c < cols; ++c) {
        input_map(s, r, c, 0) = row[c];
      }
    }
  }

  return g->run(_x);
}

std::vector< float > nnet::TfModelInterface::Run ( std::vector< std::vector< float >> const &  inp2d )

overridevirtual

Implements nnet::ModelInterface.

Definition at line 128 of file PointIdAlg.cxx.

{
  long long int rows = inp2d.size(), cols = inp2d.front().size();

  tensorflow::Tensor _x(tensorflow::DT_FLOAT, tensorflow::TensorShape({1, rows, cols, 1}));
  auto input_map = _x.tensor<float, 4>();
  for (long long int r = 0; r < rows; ++r) {
    const auto& row = inp2d[r];
    for (long long int c = 0; c < cols; ++c) {
      input_map(0, r, c, 0) = row[c];
    }
  }

  auto out = g->run(_x);
  if (!out.empty())
    return out.front();
  else
    return std::vector<float>();
}

Member Data Documentation

std::unique_ptr<tf::Graph> nnet::TfModelInterface::g

private

Definition at line 84 of file PointIdAlg.h.

---

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

• larrecodnn/larrecodnn/ImagePatternAlgs/Tensorflow/PointIdAlg/PointIdAlg.h
• larrecodnn/larrecodnn/ImagePatternAlgs/Tensorflow/PointIdAlg/PointIdAlg.cxx