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 64 of file PointIdAlg.h.

Constructor & Destructor Documentation

nnet::KerasModelInterface::KerasModelInterface

(

const char *

modelFileName

)

Definition at line 73 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 51 of file PointIdAlg.cxx.

References art::errors::NotFound.

Referenced by nnet::ModelInterface::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;
}

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 35 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 80 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 72 of file PointIdAlg.h.

Referenced by Run().

---

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

• /cvmfs/larsoft.opensciencegrid.org/products/larreco/v07_10_02/source/larreco/RecoAlg/ImagePatternAlgs/Tensorflow/PointIdAlg/PointIdAlg.h
• /cvmfs/larsoft.opensciencegrid.org/products/larreco/v07_10_02/source/larreco/RecoAlg/ImagePatternAlgs/Tensorflow/PointIdAlg/PointIdAlg.cxx