加载训练好的模型graph 然后开始run,在GPU下一张图,在这个run中需要差不多300ms,而在python中只需要40ms左右。
//// 这是使用tensorflowsharp库,在c#中。
runner.AddInput(graph["image_batch"][0], tensor).Fetch(graph["prediction"][0]);
var output = runner.Run(); //一张图输入到模型中,run,得到输出,时间远大于python实现
相似问题