wavrec_tool::WaveformRecogTf Class Reference

Inheritance diagram for wavrec_tool::WaveformRecogTf:

Public Member Functions
	WaveformRecogTf (const fhicl::ParameterSet &pset)
std::vector< std::vector< float > >	predictWaveformType (const std::vector< std::vector< float >> &) const override
Public Member Functions inherited from wavrec_tool::IWaveformRecog
virtual	~IWaveformRecog () noexcept=default
std::vector< bool >	findROI (const std::vector< float > &adcin) const
std::vector< float >	predROI (const std::vector< float > &adcin) const

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

Additional Inherited Members
Protected Member Functions inherited from wavrec_tool::IWaveformRecog
std::string	findFile (const char *fileName) const
void	setupWaveRecRoiParams (const fhicl::ParameterSet &pset)

Detailed Description

Definition at line 11 of file WaveformRecogTf_tool.cc.

Constructor & Destructor Documentation

wavrec_tool::WaveformRecogTf::WaveformRecogTf ( const fhicl::ParameterSet &  pset )

explicit

Definition at line 25 of file WaveformRecogTf_tool.cc.

  {
    fNNetModelFilePath = pset.get<std::string>("NNetModelFile", "mymodel.pb");
    fNNetOutputPattern =
      pset.get<std::vector<std::string>>("NNetOutputPattern", {"cnn_output", "dense_3"});
    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("WaveformRecogTf") << "TF model loaded.";
    }
    else {
      mf::LogError("WaveformRecogTf") << "File name extension not supported.";
    }

    setupWaveRecRoiParams(pset);
  }

Member Function Documentation

std::vector< std::vector< float > > wavrec_tool::WaveformRecogTf::predictWaveformType ( const std::vector< std::vector< float >> &  waveforms ) const

overridevirtual

Implements wavrec_tool::IWaveformRecog.

Definition at line 45 of file WaveformRecogTf_tool.cc.

  {
    if (waveforms.empty() || waveforms.front().empty()) {
      return std::vector<std::vector<float>>();
    }

    long long int samples = waveforms.size(), numtcks = waveforms.front().size();

    tensorflow::Tensor _x(tensorflow::DT_FLOAT, tensorflow::TensorShape({samples, numtcks, 1}));
    auto input_map = _x.tensor<float, 3>();
    for (long long int s = 0; s < samples; ++s) {
      const auto& wvfrm = waveforms[s];
      for (long long int t = 0; t < numtcks; ++t) {
        input_map(s, t, 0) = wvfrm[t];
      }
    }

    return g->run(_x);
  }

Member Data Documentation

std::string wavrec_tool::WaveformRecogTf::fNNetModelFilePath

private

Definition at line 20 of file WaveformRecogTf_tool.cc.

std::vector<std::string> wavrec_tool::WaveformRecogTf::fNNetOutputPattern

private

Definition at line 21 of file WaveformRecogTf_tool.cc.

std::unique_ptr<tf::Graph> wavrec_tool::WaveformRecogTf::g

private

Definition at line 19 of file WaveformRecogTf_tool.cc.

---

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

• larrecodnn/larrecodnn/ImagePatternAlgs/Tensorflow/Tools/WaveformRecogTf_tool.cc