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
Public Member Functions inherited from nnet::ModelInterface
virtual	~ModelInterface ()
virtual std::vector< std::vector< float > >	Run (std::vector< std::vector< std::vector< float >>> const &inps, int samples=-1)

Private Attributes
keras::KerasModel	m

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

Detailed Description

Definition at line 64 of file PointIdAlg.h.

Constructor & Destructor Documentation

nnet::KerasModelInterface::KerasModelInterface

(

const char *

modelFileName

)

Definition at line 67 of file PointIdAlg.cxx.

  : m(nnet::ModelInterface::findFile(modelFileName).c_str())
{
  mf::LogInfo("KerasModelInterface") << "Keras model loaded.";
}

Member Function Documentation

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

overridevirtual

Implements nnet::ModelInterface.

Definition at line 75 of file PointIdAlg.cxx.

{
  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 71 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