nnet::ModelInterface Class Reference

#include "PointIdAlg.h"

Inheritance diagram for nnet::ModelInterface:

Public Member Functions
virtual	~ModelInterface (void)
virtual std::vector< float >	Run (std::vector< std::vector< float > > const &inp2d)=0
virtual std::vector< std::vector< float > >	Run (std::vector< std::vector< std::vector< float > > > const &inps, int samples=-1)

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

Detailed Description

Interface class for various classifier models. Now MLP (NetMaker) and CNN (Keras with simple cpp interface) are supported. Will add interface to Protobuf as soon as Tensorflow may be used from UPS.

Definition at line 49 of file PointIdAlg.h.

Constructor & Destructor Documentation

virtual nnet::ModelInterface::~ModelInterface ( void  )

inlinevirtual

Definition at line 52 of file PointIdAlg.h.

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

{ }

nnet::ModelInterface::ModelInterface ( void  )

inlineprotected

Definition at line 58 of file PointIdAlg.h.

References findFile().

{ }

Member Function Documentation

std::string nnet::ModelInterface::findFile ( const char *  fileName ) const

protected

Definition at line 51 of file PointIdAlg.cxx.

References art::errors::NotFound.

Referenced by ModelInterface(), and 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;
}

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

pure virtual

Implemented in nnet::TfModelInterface, and nnet::KerasModelInterface.

Referenced by nnet::PointIdAlg::predictIdValue(), nnet::PointIdAlg::predictIdVector(), nnet::PointIdAlg::predictIdVectors(), Run(), and ~ModelInterface().

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

virtual

Reimplemented in nnet::TfModelInterface.

Definition at line 35 of file PointIdAlg.cxx.

References 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;
}

---

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

• larreco/v06_64_02/source/larreco/RecoAlg/ImagePatternAlgs/Tensorflow/PointIdAlg/PointIdAlg.h
• larreco/v06_64_02/source/larreco/RecoAlg/ImagePatternAlgs/Tensorflow/PointIdAlg/PointIdAlg.cxx