LArSoft  v09_90_00
Liquid Argon Software toolkit - https://larsoft.org/
ITFNetHandler.h
Go to the documentation of this file.
1 
7 #ifndef LCVN_TFNETHANDLER_H
8 #define LCVN_TFNETHANDLER_H
9 
10 #include <memory>
11 #include <vector>
12 
13 #include "larrecodnn/CVN/func/InteractionType.h"
14 #include "larrecodnn/CVN/func/PixelMap.h"
15 
16 namespace lcvn {
17 
19  class ITFNetHandler {
20  public:
21  virtual ~ITFNetHandler() noexcept = default;
23  virtual std::vector<std::vector<float>> Predict(const PixelMap& pm) const = 0;
24  };
25 
26 }
27 
28 #endif // CVN_TFNETHANDLER_H
lcvn::ITFNetHandler::Predict
virtual std::vector< std::vector< float > > Predict(const PixelMap &pm) const =0
Return prediction arrays for PixelMap.
lcvn::PixelMap
PixelMap, basic input to CVN neural net.
Definition: PixelMap.h:20
lcvn
Utility class for truth labels.
Definition: AssignLabels.cxx:16
PixelMap.h
PixelMap for CVN.
lcvn::ITFNetHandler::~ITFNetHandler
virtual ~ITFNetHandler() noexcept=default
lcvn::ITFNetHandler
Wrapper for caffe::Net which handles construction and prediction.
Definition: ITFNetHandler.h:19