Inheritance diagram for wavdenoise_tool::WaveformDenoiseTf:

Public Member Functions
	WaveformDenoiseTf (const fhicl::ParameterSet &pset)

std::vector< std::vector< float > >	applyDenoisingAE (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 WaveformDenoiseTf_tool.cc.

Constructor & Destructor Documentation

wavdenoise_tool::WaveformDenoiseTf::WaveformDenoiseTf ( const fhicl::ParameterSet & pset )

explicit

Definition at line 26 of file WaveformDenoiseTf_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("WaveformDenoiseTf") << "TF model loaded (pb format).";
     }
     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("WaveformDenoiseTf") << "TF model loaded (SavedModel format).";
     }
     else {
       mf::LogError("WaveformDenoiseTf") << "File name extension not supported.";
     }
 
     setupWaveDenoiseParams(pset);
   }

Member Function Documentation

std::vector< std::vector< float > > wavdenoise_tool::WaveformDenoiseTf::applyDenoisingAE ( const std::vector< std::vector< float >> & waveforms ) const

override

Definition at line 54 of file WaveformDenoiseTf_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->runae(_x);
   }