Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- TFGraph graph = new TFGraph();
- graph.Import(File.ReadAllBytes(Application.dataPath + "/frozen_model.pb"));
- TFTensor input = ImageUtil.CreateTensorFromImageFile("C:/Users/Lopes/Desktop/longhand.jpg", TFDataType.Float);
- float[][][][] outfloats;
- using (TFSession sess = new TFSession(graph))
- {
- var runner = sess.GetRunner();
- runner.AddInput(graph[input_tensorname][0], input);
- runner.Fetch(graph[output_tensorname][0]);
- var output = runner.Run();
- outfloats = (float[][][][])output[0].GetValue();
- }
Add Comment
Please, Sign In to add comment