保存为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, 但是这个版本由于依赖问题无法安装。