TFFifoHandler.h
Go to the documentation of this file.
1 ///////////////////////////////////////////////////////////////////////
2 // TFFifoHander
3 //
4 // A module to evaluate a tensorflow model using FIFO connections to python
5 //
6 // \author $Author: Alex Himmel
7 ////////////////////////////////////////////////////////////////////////
8 
9 #ifndef TFFIFOHANDLER_H
10 #define TFFIFOHANDLER_H
11 
12 #include <vector>
13 #include "fhiclcpp/ParameterSet.h"
14 #include "CVN/func/PixelMap.h"
15 
16 
17 namespace tensorflow
18 {
19  /// Wrapper which handles communication with a TF script via FIFO pipes
21  {
22  public:
23  /// Basic constructor, takes path to model pb
26 
27  std::vector<std::vector<double> > Predict(const std::vector<cvn::PixelMap const*> pixellist) const;
28 
29  private:
32  };
33 }
34 
35 #endif
std::vector< std::vector< double > > Predict(const std::vector< cvn::PixelMap const * > pixellist) const
PixelMap for CVN.
Wrapper which handles communication with a TF script via FIFO pipes.
Definition: TFFifoHandler.h:20
TFFifoHandler(fhicl::ParameterSet const &pset)
Basic constructor, takes path to model pb.
enum BeamMode string