PointIdAlgTools::PointIdAlgTf Class Reference

Inheritance diagram for PointIdAlgTools::PointIdAlgTf:

Public Member Functions
	PointIdAlgTf (fhicl::Table< Config > const &table)
std::vector< float >	Run (std::vector< std::vector< float >> const &inp2d) const override
std::vector< std::vector< float > >	Run (std::vector< std::vector< std::vector< float >>> const &inps, int samples=-1) const override

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

Private Attributes
std::unique_ptr< tf::Graph >	g
std::vector< std::string >	fNNetOutputPattern
std::string	fNNetModelFilePath

Detailed Description

Definition at line 20 of file PointIdAlgTf_tool.cc.

Constructor & Destructor Documentation

PointIdAlgTools::PointIdAlgTf::PointIdAlgTf ( fhicl::Table< Config > const &  table )

explicit

Definition at line 38 of file PointIdAlgTf_tool.cc.

References tf::Graph::create(), findFile(), fNNetModelFilePath, fNNetOutputPattern, g, and art::errors::Unknown.

                                                            : img::DataProviderAlg(table())
  {
    // ... Get common config vars
    fNNetOutputs = table().NNetOutputs();
    fPatchSizeW = table().PatchSizeW();
    fPatchSizeD = table().PatchSizeD();
    fCurrentWireIdx = 99999;
    fCurrentScaledDrift = 99999;

    // ... Get "optional" config vars specific to tf interface
    std::string s_cfgvr;
    if (table().NNetModelFile(s_cfgvr)) { fNNetModelFilePath = s_cfgvr; }
    else {
      fNNetModelFilePath = "mycnn";
    }
    std::vector<std::string> vs_cfgvr;
    if (table().NNetOutputPattern(vs_cfgvr)) { fNNetOutputPattern = vs_cfgvr; }
    else {
      fNNetOutputPattern = {"cnn_output", "_netout"};
    }

    if ((fNNetModelFilePath.length() > 3) &&
        (fNNetModelFilePath.compare(fNNetModelFilePath.length() - 3, 3, ".pb") == 0)) {
      g = tf::Graph::create(findFile(fNNetModelFilePath.c_str()).c_str(), fNNetOutputPattern);
      if (!g) { throw art::Exception(art::errors::Unknown) << "TF model failed."; }
      mf::LogInfo("PointIdAlgTf") << "TF model loaded.";
    }
    else {
      mf::LogError("PointIdAlgTf") << "File name extension not supported.";
    }

    resizePatch();
  }

Member Function Documentation

std::string PointIdAlgTools::PointIdAlgTf::findFile ( const char *  fileName ) const

protected

Definition at line 73 of file PointIdAlgTf_tool.cc.

References art::errors::NotFound.

Referenced by PointIdAlgTf().

  {
    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< float > PointIdAlgTools::PointIdAlgTf::Run ( std::vector< std::vector< float >> const &  inp2d ) const

override

Definition at line 88 of file PointIdAlgTf_tool.cc.

References g, and r.

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

    std::vector<tensorflow::Tensor> _x;
    _x.push_back(
      tensorflow::Tensor(tensorflow::DT_FLOAT, tensorflow::TensorShape({1, rows, cols, 1})));
    auto input_map = _x[0].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->runx(_x);
    if (!out.empty())
      return out.front();
    else
      return std::vector<float>();
  }

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

override

Definition at line 111 of file PointIdAlgTf_tool.cc.

References DEFINE_ART_CLASS_TOOL, g, and r.

  {

    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();

    std::vector<tensorflow::Tensor> _x;
    _x.push_back(
      tensorflow::Tensor(tensorflow::DT_FLOAT, tensorflow::TensorShape({samples, rows, cols, 1})));
    auto input_map = _x[0].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->runx(_x);
  }

Member Data Documentation

std::string PointIdAlgTools::PointIdAlgTf::fNNetModelFilePath

private

Definition at line 34 of file PointIdAlgTf_tool.cc.

Referenced by PointIdAlgTf().

std::vector<std::string> PointIdAlgTools::PointIdAlgTf::fNNetOutputPattern

private

Definition at line 33 of file PointIdAlgTf_tool.cc.

Referenced by PointIdAlgTf().

std::unique_ptr<tf::Graph> PointIdAlgTools::PointIdAlgTf::g

private

Definition at line 32 of file PointIdAlgTf_tool.cc.

Referenced by PointIdAlgTf(), and Run().

---

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

• larrecodnn/v09_23_00/source/larrecodnn/ImagePatternAlgs/Tensorflow/Tools/PointIdAlgTf_tool.cc