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

Private Attributes
std::unique_ptr< tf::Graph >	g

std::string	fNNetModelFilePath

std::vector< std::string >	fNNetOutputPattern

bool	fUseBundle

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 26 of file WaveformRecogTf_tool.cc.

References tf::Graph::create(), fNNetModelFilePath, fNNetOutputPattern, fUseBundle, g, fhicl::ParameterSet::get(), and art::errors::Unknown.

   {
     fNNetModelFilePath = pset.get<std::string>("NNetModelFile", "mymodel.pb");
     fUseBundle = pset.get<bool>("UseSavedModelBundle", false);
     fNNetOutputPattern =
       pset.get<std::vector<std::string>>("NNetOutputPattern", {"cnn_output", "dense_3"});
     if ((fNNetModelFilePath.length() > 3) &&
         (fNNetModelFilePath.compare(fNNetModelFilePath.length() - 3, 3, ".pb") == 0) &&
         !fUseBundle) {
       g = tf::Graph::create(
         findFile(fNNetModelFilePath.c_str()).c_str(), fNNetOutputPattern, fUseBundle);
       if (!g) { throw art::Exception(art::errors::Unknown) << "TF model failed."; }
       mf::LogInfo("WaveformRecogTf") << "TF model loaded.";
     }
     else if ((fNNetModelFilePath.length() > 3) && fUseBundle) {
       g = tf::Graph::create(
         findFile(fNNetModelFilePath.c_str()).c_str(), fNNetOutputPattern, fUseBundle);
       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

override

Definition at line 54 of file WaveformRecogTf_tool.cc.

References DEFINE_ART_CLASS_TOOL, and g.

   {
     if (waveforms.empty() || waveforms.front().empty()) {
       return std::vector<std::vector<float>>();
     }
 
     long long int samples = waveforms.size(), numtcks = waveforms.front().size();
 
     //std::cout<<"Samples: "<<samples<<", Ticks: "<<numtcks<<std::endl;
     std::vector<tensorflow::Tensor> _x;
     _x.push_back(
       tensorflow::Tensor(tensorflow::DT_FLOAT, tensorflow::TensorShape({samples, numtcks, 1})));
     auto input_map = _x[0].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->runx(_x);
   }