13 #include "tensorflow/core/public/session.h" 14 #include "tensorflow/core/platform/env.h" 27 std::vector<Tensor>
Predict(std::vector<std::pair<std::string,Tensor>>
inputs,
28 std::vector<std::string> outputLabels);
TFHandler(std::string model, int CPUlimit=1)
Basic constructor, takes path to model pb.
void Initialize(int CPUlimit)
std::vector< Tensor > Predict(std::vector< std::pair< std::string, Tensor >> inputs, std::vector< std::string > outputLabels)
Wrapper for Tensorflow which handles construction and prediction.
const XML_Char XML_Content * model