如何将从Google的可教机器保存的tensorflow.js模型转换为keras.hdf5或tflite模型?

时间:2020-03-16 06:29:56

标签: tensorflow tensorflow.js

保存的tensorflow.js模型由以下3个文件组成。

  1. metadata.json
  2. model.json
  3. weights.bin

1 个答案:

答案 0 :(得分:1)

您可以使用TensorFlow.js转换器库在格式之间进行转换:

https://github.com/tensorflow/tfjs/tree/master/tfjs-converter#getting-started

它不会转换为TFLite,但应允许您从TensorFlow.js to Keras进行转换。例如:

tensorflowjs_converter \
    --input_format tfjs_layers_model \
    --output_format keras \
    tfjs_model/model.json \
    keras_model/