如何将tensorflow.js模型和权重转换为标准tensorflow?

时间:2019-01-17 19:08:14

标签: tensorflow tensorflow.js

我想将一个模型及其受过训练的权重从tensorflow.js转换为标准tensorflow,但无法完全确定该怎么做,tensorflow.js的文档对此一无所知

我有一个manifest.json和很多其他文件,分别对应于每一层的权重和偏差,我想获得一个.bytes文件,用于标准版本的tensorflow

我检查了可能与轻度相关的其他问题的答案 How to download models and weights from tensorflow.js (我问了这个问题),只是部分回答了,没有给出关于这个问题的任何提示

1 个答案:

答案 0 :(得分:0)

Tensorflow.js转换器可以使用以下参数将tfjs模型转换为Keras HDF5格式: tensorflowjs_converter --input_format = tensorflowjs --output_format = keras路径/至/ tfjs_files路径/至/ save_keras_hdf5

您可以查看code here。 将模型转换为Keras HDF5后,您可以将其转换为TensorFlow SavedModel格式。这个SO thread可能会有所帮助。