
TensorFlow用于显示每一层的输出。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
通过在test.py文件中运用以下代码,可以便捷地创建包含weight的pb文件。此外,借助TensorFlow官方提供的freeze_graph.py脚本,还可以将ckpt文件转换为pb文件格式。随后,使用`graph_util.convert_variables_to_constants`函数,将会话及其图定义中的特定节点(具体为net_loss/inference/encode/conv_output/conv_output)转换为常量,并将其写入到名为net_model.pb的文件中。该过程涉及将这些常量序列化为字节流,并通过`f.write(constant_graph.SerializeT)`语句将其写入到打开的FastGFile对象中。
全部评论 (0)
还没有任何评论哟~


