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

Protected Member Functions
std::string	findFile (const char *fileName) const

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

Detailed Description

Definition at line 85 of file PointIdAlg.h.

Constructor & Destructor Documentation

nnet::TfModelInterface::TfModelInterface

(

const char *

modelFileName

)

Definition at line 108 of file PointIdAlg.cxx.

References tf::Graph::create(), nnet::ModelInterface::findFile(), and art::errors::Unknown.

{
  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::string nnet::ModelInterface::findFile ( const char *  fileName ) const

protectedinherited

Definition at line 67 of file PointIdAlg.cxx.

References art::errors::NotFound.

Referenced by TfModelInterface().

{
  std::string fname_out;
  cet::search_path sp("FW_SEARCH_PATH");
  if (!sp.find_file(fileName, fname_out)) {
    struct stat buffer;
    if (stat(fileName, &buffer) == 0) { fname_out = fileName; }
    else {
      throw art::Exception(art::errors::NotFound) << "Could not find the model file " << fileName;
    }
  }
  return fname_out;
}

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 118 of file PointIdAlg.cxx.

References r, and lar::dump::vector().

{
  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();

  std::vector<tensorflow::Tensor> _x;
  _x.push_back(
    tensorflow::Tensor(tensorflow::DT_FLOAT, tensorflow::TensorShape({samples, rows, cols, 1})));
  auto input_map = _x[0].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->runx(_x);
}

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

overridevirtual

Implements nnet::ModelInterface.

Definition at line 147 of file PointIdAlg.cxx.

References r.

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

  std::vector<tensorflow::Tensor> _x;
  _x.push_back(
    tensorflow::Tensor(tensorflow::DT_FLOAT, tensorflow::TensorShape({1, rows, cols, 1})));
  auto input_map = _x[0].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->runx(_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 94 of file PointIdAlg.h.

---

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

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