0
点赞
收藏
分享

微信扫一扫

tensorflow1.x python模型转换

独孤凌雪 2022-01-26 阅读 69

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

举报

相关推荐

0 条评论