12 #include "tensorflow/core/public/session.h" 13 #include "tensorflow/core/platform/env.h" 14 #include "tensorflow/core/public/session_options.h" 35 tensorflow::SessionOptions
options;
36 tensorflow::ConfigProto &
config = options.config;
38 config.set_inter_op_parallelism_threads(CPUlimit);
39 config.set_intra_op_parallelism_threads(CPUlimit);
40 config.set_use_per_session_threads(
false);
49 status = ReadBinaryProto(Env::Default(),
fModelPath, &graph_def);
55 status =
fSession->Create(graph_def);
65 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)