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

6
我想将一个从tensorflow.js训练出来的模型及其相应的已训练权重转换为标准tensorflow格式,但是无法确定如何操作,因为tensorflow.js的文档没有提到这个问题。我有一个manifest.json文件和许多其他文件,对应于每一层的权重和偏差,我想得到一个.bytes文件,用于标准版的tensorflow。我查看了与此问题有关的其他问题的答案如何从tensorflow.js下载模型和权重(我曾经问过这个问题),该回答只部分解决了问题,没有提供任何有关此问题的提示。

可能是如何从tensorflow.js下载模型和权重的重复问题。 - edkeveked
我是之前发布那个问题的同一个人,这部分还没有得到回答,如果你有这个特定问题的答案,请分享。 - Boris
1个回答

6

Tensorflow.js转换器可以使用以下参数将tfjs模型转换为Keras HDF5格式:

tensorflowjs_converter --input_format=tfjs_layers_model --output_format=keras path/to/tfjs_files path/to/save_keras_hdf5

您可以查看这里进行转换工具的安装。 一旦模型转换为Keras HDF5格式,您就可以将其转换为TensorFlow SavedModel格式。因此,这个主题可能会有所帮助。 注意: 这适用于“tfjs层模型”格式的模型,但不适用于tfjs图模型格式。

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接