keras_to_tensorflow:将经过训练的keras模型转换为推断张量流模型的通用代码

时间:2021-02-03 07:10:49
【文件属性】:
文件名称:keras_to_tensorflow:将经过训练的keras模型转换为推断张量流模型的通用代码
文件大小:12KB
文件格式:ZIP
更新时间:2021-02-03 07:10:49
python machine-learning deep-learning tensorflow keras Keras到TensorFlow keras_to_tensorflow是将经过训练的keras模型转换为可随时推断的TensorFlow模型的工具。 该工具不是为TensorFlow 2.0量身定制的,它看起来运行良好。 概要 按照默认行为,此工具将冻结节点(将所有TF变量转换为TF常量),并将推理图和权重保存到二进制protobuf(.pb)文件中。 在冻结期间,TensorFlow还应用节点修剪,该修剪将删除对输出张量无贡献的节点。 该工具支持多个输出网络,并使用户能够通过--output_nodes_prefix标志重命名输出张量。 如果设置了--output_meta_ckpt标志,还将导出TensorFlow的检查点和元文件,以后可以在tf.train.Saver类中使用该tf.train.Saver继续训练。 如何使用 Keras模型可以使用model.save()函数保存为单个[ .hdf5或h5 ]文件,该文件同时存储体系结构和权重。 然后可以通过如下调用此工具将该模型转换为TensorFlow模型: python keras_to_tensorflow.py
【文件预览】:
keras_to_tensorflow-master
----requirements.txt(34B)
----LICENSE(1KB)
----README.md(2KB)
----.gitignore(71B)
----legacy()
--------keras_to_tensorflow_via_freeze_graph.ipynb(6KB)
--------freeze_graph.py(7KB)
--------keras_to_tensorflow.ipynb(8KB)
----keras_to_tensorflow.py(8KB)

网友评论