tensorflow1.x python模型转换

保存为saved_model

主要用于python自己使用

saver = tf.train.Saver(max_to_keep = 1000)
saver.save(sess,path)

转为tflite

converter = tf.lite.TFLiteConverter.from_session(sess, [inputs],[outputs])
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)

参数说明:

• inputs: 输入张量
• outputs: 输出张量
• sess: sess

转为js

使用工具tfjs, 需要用到tensorflow2.x，所以最好在新的环境中使用

1: 安装

pip install tensorflowjs

2: 支持转换类型

• TensorFlow SavedModel

tensorflowjs_converter  --input_format=tf_saved_model  --output_format=tfjs_graph_model --signature_name=serving_default  --saved_model_tags=serve /saved_model /web_model

• tensorflow Hub module
• Keras HDF5 model
• tf.keras SavedModel

如果想要转sess生成的Graph，必须用tensorflowjs==0.8.6, 但是这个版本由于依赖问题无法安装。