nnet::KerasModelInterface Class Reference

#include "PointIdAlg.h"

Inheritance diagram for nnet::KerasModelInterface:

Public Member Functions
	KerasModelInterface (const char *modelFileName)
std::vector< float >	Run (std::vector< std::vector< float >> const &inp2d) override
virtual std::vector< std::vector< float > >	Run (std::vector< std::vector< std::vector< float >>> const &inps, int samples=-1)

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

Private Attributes
keras::KerasModel	m

Detailed Description

Definition at line 74 of file PointIdAlg.h.

Constructor & Destructor Documentation

nnet::KerasModelInterface::KerasModelInterface

(

const char *

modelFileName

)

Definition at line 85 of file PointIdAlg.cxx.

  : m(nnet::ModelInterface::findFile(modelFileName).c_str())
{
  mf::LogInfo("KerasModelInterface") << "Keras 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 nnet::TfModelInterface::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::ModelInterface::Run ( std::vector< std::vector< std::vector< float >>> const &  inps, int  samples = -1  )

virtualinherited

Reimplemented in nnet::TfModelInterface.

Definition at line 51 of file PointIdAlg.cxx.

References nnet::ModelInterface::Run(), 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 > (int)inps.size())) { samples = inps.size(); }

  std::vector<std::vector<float>> results;
  for (int i = 0; i < samples; ++i) {
    results.push_back(Run(inps[i]));
  }
  return results;
}

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

overridevirtual

Implements nnet::ModelInterface.

Definition at line 92 of file PointIdAlg.cxx.

References keras::KerasModel::compute_output(), m, and keras::DataChunk::set_data().

{
  std::vector<std::vector<std::vector<float>>> inp3d;
  inp3d.push_back(inp2d); // lots of copy, should add 2D to keras...

  keras::DataChunk* sample = new keras::DataChunk2D();
  sample->set_data(inp3d); // and more copy...
  std::vector<float> out = m.compute_output(sample);
  delete sample;
  return out;
}

Member Data Documentation

keras::KerasModel nnet::KerasModelInterface::m

private

Definition at line 81 of file PointIdAlg.h.

Referenced by Run().

---

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