phot::TFLoaderMLP Class Reference

#include "TFLoaderMLP.h"

Inheritance diagram for phot::TFLoaderMLP:

Public Member Functions
	TFLoaderMLP (fhicl::ParameterSet const &pset)
void	Initialization ()
void	CloseSession ()
void	Predict (std::vector< double > pars)
std::vector< double >	GetPrediction () const

Protected Attributes
std::vector< double >	prediction

Private Attributes
std::string	ModelName
std::vector< std::string >	InputsName
std::string	OutputName
tensorflow::SavedModelBundleLite *	modelbundle
tensorflow::Status	status

Detailed Description

Definition at line 16 of file TFLoaderMLP.h.

Constructor & Destructor Documentation

phot::TFLoaderMLP::TFLoaderMLP ( fhicl::ParameterSet const &  pset )

explicit

Definition at line 11 of file TFLoaderMLP_tool.cc.

References fhicl::ParameterSet::get(), InputsName, and OutputName.

    : ModelName{pset.get<std::string>("ModelName")}
    , InputsName{pset.get<std::vector<std::string>>("InputsName")}
    , OutputName{pset.get<std::string>("OutputName")}
  {}

Member Function Documentation

void phot::TFLoaderMLP::CloseSession ( )

virtual

Implements phot::TFLoader.

Definition at line 78 of file TFLoaderMLP_tool.cc.

References modelbundle, and status.

  {
    if (status.ok()) {
      std::cout << "Close TF session." << std::endl;
      //            session->Close();
    }

    delete modelbundle;

    //        delete session;
    return;
  }

std::vector<double> phot::TFLoader::GetPrediction ( ) const

inlineinherited

Definition at line 36 of file TFLoader.h.

References phot::TFLoader::prediction.

{ return prediction; }

void phot::TFLoaderMLP::Initialization ( )

virtual

Implements phot::TFLoader.

Definition at line 18 of file TFLoaderMLP_tool.cc.

References InputsName, modelbundle, ModelName, OutputName, and status.

  {
    int num_input = int(InputsName.size());
    if (num_input != 3) {
      std::cout << "Input name error! exit!" << std::endl;
      return;
    }
    std::string GraphFileWithPath;
    cet::search_path sp("FW_SEARCH_PATH");
    if (!sp.find_file(ModelName, GraphFileWithPath)) {
      throw cet::exception("TFLoaderMLP")
        << "In larrecodnn:phot::TFLoaderMLP: Failed to load SavedModel in : " << sp.to_string()
        << "\n";
    }
    std::cout << "larrecodnn:phot::TFLoaderMLP Loading TF Model from: " << GraphFileWithPath
              << ", Input Layer: ";
    for (int i = 0; i < num_input; ++i) {
      std::cout << InputsName[i] << " ";
    }
    std::cout << ", Output Layer: " << OutputName << "\n";

    //Load SavedModel
    modelbundle = new tensorflow::SavedModelBundleLite();

    status = tensorflow::LoadSavedModel(tensorflow::SessionOptions(),
                                        tensorflow::RunOptions(),
                                        GraphFileWithPath,
                                        {tensorflow::kSavedModelTagServe},
                                        modelbundle);

    //Initialize a tensorflow session
    //        status = tensorflow::NewSession(tensorflow::SessionOptions(), &session);
    if (!status.ok()) {
      throw cet::exception("TFLoaderMLP")
        << "In larrecodnn:phot::TFLoaderMLP: Failed to load SavedModel, status: "
        << status.ToString() << std::endl;
    }

    //Read in the protobuf graph
    //        tensorflow::GraphDef graph_def;
    //        status = tensorflow::ReadBinaryProto(tensorflow::Env::Default(), ModelName, &graph_def);
    //        if (!status.ok())
    //        {
    //            std::cout << status.ToString() << std::endl;
    //            return;
    //        }
    //
    //        //Add the graph to the session
    //        status = session->Create(graph_def);
    //        if (!status.ok())
    //        {
    //            std::cout << status.ToString() << std::endl;
    //            return;
    //        }

    std::cout << "TF SavedModel loaded successfully." << std::endl;
    return;
  }

void phot::TFLoaderMLP::Predict ( std::vector< double >  pars )

virtual

Implements phot::TFLoader.

Definition at line 92 of file TFLoaderMLP_tool.cc.

References DEFINE_ART_CLASS_TOOL, InputsName, modelbundle, OutputName, phot::TFLoader::prediction, status, and value.

  {
    //std::cout << "TFLoader MLP:: Predicting... " << std::endl;
    int num_input = int(pars.size());
    if (num_input != 3) {
      std::cout << "Input parameter error! exit!" << std::endl;
      return;
    }
    //Clean prediction
    std::vector<double>().swap(prediction);

    //Define inputs
    tensorflow::Tensor pos_x(tensorflow::DT_FLOAT, tensorflow::TensorShape({1, 1}));
    tensorflow::Tensor pos_y(tensorflow::DT_FLOAT, tensorflow::TensorShape({1, 1}));
    tensorflow::Tensor pos_z(tensorflow::DT_FLOAT, tensorflow::TensorShape({1, 1}));
    auto dst_x = pos_x.flat<float>().data();
    auto dst_y = pos_y.flat<float>().data();
    auto dst_z = pos_z.flat<float>().data();
    copy_n(pars.begin(), 1, dst_x);
    copy_n(pars.begin() + 1, 1, dst_y);
    copy_n(pars.begin() + 2, 1, dst_z);
    std::vector<std::pair<std::string, tensorflow::Tensor>> inputs = {
      {InputsName[0], pos_x}, {InputsName[1], pos_y}, {InputsName[2], pos_z}};
    //Define outps
    std::vector<tensorflow::Tensor> outputs;

    //Run the session
    status = modelbundle->GetSession()->Run(inputs, {OutputName}, {}, &outputs);
    //        status = session->Run(inputs, {OutputName}, {}, &outputs);
    if (!status.ok()) {
      std::cout << status.ToString() << std::endl;
      return;
    }

    //Grab the outputs
    unsigned int pdr = outputs[0].shape().dim_size(1);
    //std::cout << "TFLoader MLP::Num of optical channels: " << pdr << std::endl;

    for (unsigned int i = 0; i < pdr; i++) {
      double value = outputs[0].flat<float>()(i);
      //std::cout << value << ", ";
      prediction.push_back(value);
    }
    //std::cout << std::endl;
    return;
  }

Member Data Documentation

std::vector<std::string> phot::TFLoaderMLP::InputsName

private

Definition at line 25 of file TFLoaderMLP.h.

Referenced by Initialization(), Predict(), and TFLoaderMLP().

tensorflow::SavedModelBundleLite* phot::TFLoaderMLP::modelbundle

private

Definition at line 28 of file TFLoaderMLP.h.

Referenced by CloseSession(), Initialization(), and Predict().

std::string phot::TFLoaderMLP::ModelName

private

Definition at line 24 of file TFLoaderMLP.h.

Referenced by Initialization().

std::string phot::TFLoaderMLP::OutputName

private

Definition at line 26 of file TFLoaderMLP.h.

Referenced by Initialization(), Predict(), and TFLoaderMLP().

std::vector<double> phot::TFLoader::prediction

protectedinherited

Definition at line 39 of file TFLoader.h.

Referenced by phot::TFLoader::GetPrediction(), and Predict().

tensorflow::Status phot::TFLoaderMLP::status

private

Definition at line 31 of file TFLoaderMLP.h.

Referenced by CloseSession(), Initialization(), and Predict().

---

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

• larsimdnn/v09_06_05/source/larsimdnn/PhotonPropagation/TFLoaderTools/TFLoaderMLP.h
• larsimdnn/v09_06_05/source/larsimdnn/PhotonPropagation/TFLoaderTools/TFLoaderMLP_tool.cc