PointIdAlgTools::PointIdAlgKeras Class Reference

Inheritance diagram for PointIdAlgTools::PointIdAlgKeras:

Public Member Functions
	PointIdAlgKeras (const fhicl::ParameterSet &pset)
	PointIdAlgKeras (const Config &config)
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

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

Private Attributes
std::unique_ptr< keras::KerasModel >	m
std::string	fNNetModelFilePath

Detailed Description

Definition at line 19 of file PointIdAlgKeras_tool.cc.

Constructor & Destructor Documentation

PointIdAlgTools::PointIdAlgKeras::PointIdAlgKeras ( const fhicl::ParameterSet &  pset )

inlineexplicit

Definition at line 21 of file PointIdAlgKeras_tool.cc.

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

      : PointIdAlgKeras(fhicl::Table<Config>(pset, {})())
    {}

PointIdAlgTools::PointIdAlgKeras::PointIdAlgKeras ( const Config &  config )

explicit

Definition at line 37 of file PointIdAlgKeras_tool.cc.

References findFile(), fNNetModelFilePath, and m.

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

    // ... Get "optional" config vars specific to tf interface
    std::string s_cfgvr;
    if (config.NNetModelFile(s_cfgvr)) { fNNetModelFilePath = s_cfgvr; }
    else {
      fNNetModelFilePath = "mycnn";
    }

    if ((fNNetModelFilePath.length() > 5) &&
        (fNNetModelFilePath.compare(fNNetModelFilePath.length() - 5, 5, ".nnet") == 0)) {
      m = std::make_unique<keras::KerasModel>(findFile(fNNetModelFilePath.c_str()).c_str());
      mf::LogInfo("PointIdAlgKeras") << "Keras model loaded.";
    }
    else {
      mf::LogError("PointIdAlgKeras") << "File name extension not supported.";
    }

    resizePatch();
  }

Member Function Documentation

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

private

Definition at line 66 of file PointIdAlgKeras_tool.cc.

References art::errors::NotFound.

Referenced by PointIdAlgKeras().

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

override

Definition at line 81 of file PointIdAlgKeras_tool.cc.

References m, and keras::DataChunk2D::set_data().

Referenced by PointIdAlgKeras().

  {
    std::vector<std::vector<std::vector<float>>> inp3d;
    inp3d.push_back(inp2d); // lots of copy, should add 2D to keras...

    keras::DataChunk2D sample;
    sample.set_data(inp3d);
    return m->compute_output(&sample);
  }

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

override

Definition at line 92 of file PointIdAlgKeras_tool.cc.

References DEFINE_ART_CLASS_TOOL, m, and keras::DataChunk::set_data().

  {

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

    std::vector<std::vector<float>> out;

    for (long long int s = 0; s < samples; ++s) {
      std::vector<std::vector<std::vector<float>>> inp3d;
      inp3d.push_back(inps[s]); // lots of copy, should add 2D to keras...

      keras::DataChunk* sample = new keras::DataChunk2D();
      sample->set_data(inp3d); // and more copy...
      out.push_back(m->compute_output(sample));
      delete sample;
    }

    return out;
  }

Member Data Documentation

std::string PointIdAlgTools::PointIdAlgKeras::fNNetModelFilePath

private

Definition at line 32 of file PointIdAlgKeras_tool.cc.

Referenced by PointIdAlgKeras().

std::unique_ptr<keras::KerasModel> PointIdAlgTools::PointIdAlgKeras::m

private

Definition at line 31 of file PointIdAlgKeras_tool.cc.

Referenced by PointIdAlgKeras(), and Run().

---

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

• larrecodnn/v09_23_00/source/larrecodnn/ImagePatternAlgs/Keras/Tools/PointIdAlgKeras_tool.cc