RegCNNNumuHandler.h
Go to the documentation of this file.
1 ////////////////////////////////////////////////////////////////////////
2 /// \file RegCNNNumuHandler.h
3 /// \brief RegCNNNumuHandler for numu energy estimation
4 /// \author Wenjie Wu - wenjieww@uci.edu
5 ////////////////////////////////////////////////////////////////////////
6 
7 #ifndef REGCNN_NUMUHANDLER_H
8 #define REGCNN_NUMUHANDLER_H
9 
10 #include <vector>
11 #include <memory>
12 
13 #include "fhiclcpp/ParameterSet.h"
16 
17 namespace cnn
18 {
19 
20  /// Wrapper for caffe::Net which handles construction and prediction
22  {
23  public:
24 
25  /// Constructor which takes a pset with DeployProto and ModelFile fields
27 
28  /// Return prediction arrays for RegPixelMap
29  std::vector<float> Predict(const RegPixelMap& pm, bool fLongestTrackContained);
30 
31  private:
32 
35 
36  };
37 }
38 #endif // REGCNN_NUMUHANDLER_H
Wrapper for caffe::Net which handles construction and prediction.
RegPixelMap, basic input to CNN neural net.
Definition: RegPixelMap.h:22
RegCNNNumuHandler(const fhicl::ParameterSet &pset)
Constructor which takes a pset with DeployProto and ModelFile fields.
std::vector< float > Predict(const RegPixelMap &pm, bool fLongestTrackContained)
Return prediction arrays for RegPixelMap.
cnn::TFRegNetHandler fTFHandlerExiting
Wrapper for caffe::Net which handles construction and prediction.
Defines an enumeration for cellhit classification.
TFRegNetHandler for RegCNN modified from TFNetHandler.h.
RegPixelMap for RegCNN modified from PixelMap.h.
cnn::TFRegNetHandler fTFHandlerContained