13 #include "tensorflow/core/public/session.h" 14 #include "tensorflow/core/platform/env.h" 30 Status
status = NewSession(SessionOptions(), &session);
37 if(fRHC) Model =
"/nova/app/users/vold/ART2/ShowerLID/RHC_Model.pb";
40 status = ReadBinaryProto(Env::Default(),Model, &graph_def);
47 status = session->Create(graph_def);
54 Tensor aux_input(DT_FLOAT, TensorShape({1,1}));
55 Tensor main_input(DT_FLOAT, TensorShape({1,7,19}));
56 auto Slc = aux_input.tensor<
float,2>();
57 auto Shw = main_input.tensor<
float,3>();
62 for(std::map<
int,std::vector<float>>::iterator
i = featuremap.begin();
i != featuremap.end();
i++){
63 std::vector<float>
vec =
i->second;
65 for(std::vector<float>::iterator
j = vec.begin();
j != vec.end();
j++){
66 Shw(0,num1,num2) = *
j;
78 std::vector<std::pair<string,Tensor>>
inputs = {
79 {
"main_input", main_input },
80 {
"aux_input", aux_input },
87 status = session->Run(inputs,{
"output_node0"},{},&
outputs);
94 auto output_c = outputs[0].scalar<
float>();
TensorflowHandler(fhicl::ParameterSet const &pset)
Basic constructor, takes path to deploy model pb.
rosenbrock_model_namespace::rosenbrock_model Model
TensorflowHandler for ShowerLID.
float Predict(float sliceE, std::map< int, std::vector< float >> featuremap, bool fRHC)