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

Constructor & Destructor Documentation

nnet::TfModelInterface::TfModelInterface

(

const char *

modelFileName

)

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

References art::errors::NotFound.

Referenced by nnet::ModelInterface::ModelInterface(), and 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 106 of file PointIdAlg.cxx.

References s, 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();

    tensorflow::Tensor _x(tensorflow::DT_FLOAT, tensorflow::TensorShape({ samples, rows, cols, 1 }));
    auto input_map = _x.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->run(_x);
}

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

overridevirtual

Implements nnet::ModelInterface.

Definition at line 131 of file PointIdAlg.cxx.

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

    tensorflow::Tensor _x(tensorflow::DT_FLOAT, tensorflow::TensorShape({ 1, rows, cols, 1 }));
    auto input_map = _x.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->run(_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 85 of file PointIdAlg.h.

---

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