我有一些Keras的经验,我正在学习TensorFlowSharp(https://github.com/migueldeicaza/TensorFlowSharp).使用ExampleInceptionInference项目,我想使用我在Keras python代码中训练的模型.
(我在这里用@jdehesa回复创建了一个.pb文件：How to export Keras .h5 to tensorflow .pb?)
当我在TensorFlowSharp中加载.pb模型文件时,此行失败：
runner.AddInput (graph ["input"] [0], tensor).Fetch (graph ["output"] [0]);
看来graph [“input”]和graph [“output”]都是null.
我是否需要在模型中命名图层？
我使用的Keras模型是VGG16的经过修改和重新训练的版本.
是否可以在TensorFlowSharp中输出命名图层列表？或者是否可以通过其他方式引用它们？ (按号码？)
解决方法:
TFGraph的GetEnumerator()方法允许您枚举所有操作.
List<TFOperation> op_list = new List<TFOperation>( graph.GetEnumerator() );
您可以检查TFOperation名称和OpType以查看图表中的内容.
版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 [email protected] 举报，一经查实，本站将立刻删除。