如何读取tensorflow检查点文件中保存的权重？

Question

我想阅读权重并将它们视为图像.但我没有看到任何关于模型格式的文档以及如何阅读训练有素的权重.

Answer 1

这个实用程序有print_tensors_in_checkpoint_file方法http://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/tools/inspect_checkpoint.py

或者,您可以使用Saver恢复模型并session.run在变量张量上使用以获取值为numpy数组