+关注
已关注

分类  

暂无分类

标签  

暂无标签

日期归档  

暂无数据

OSError:SavedModel文件不存在tflite

发布于2021-02-28 01:51     阅读(812)     评论(0)     点赞(19)     收藏(2)


0

1

2

3

4

5

6

7

8

9

我正在尝试将保存的模型转换为tflite模型,保存的模型保存在我的桌面上,但是当我尝试运行以下代码时:我产生了错误-

OSError: SavedModel file does not exist at: C:/Users/Omar/Desktop/model00000014.h5/{saved_model.pbtxt|saved_model.pb}. 

不确定是什么问题。


import tensorflow as tf

saved_model_dir = "r"C:/Users/Omar/Desktop/model00000014.h5""
converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)



解决方案


如果您尝试将.h5Keras模型转换为TFLite模型,请确保使用docs中TFLiteConverter.from_keras_model()所述的方法

model = tf.keras.models.load( "C:/Users/Omar/Desktop/model00000014.h5" )
converter = tf.lite.TFLiteConverter.from_keras_model( model )
open( 'model.tflite' , 'wb' ).write( converter.convert() )

如果是SavedModel,请使用TFLiteConverter.from_saved_model()并提及SavedModel目录的文件路径,

saved_model_dir = 'path/to/savedModelDir'
converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)

您正在为该TFLiteConverter.from_saved_model()方法提供Keras模型,这可能会导致错误。

0

1

2

3

4

5

6

7

8

9



所属网站分类: 技术文章 > 问答

作者:黑洞官方问答小能手

链接: https://www.pythonheidong.com/blog/article/861637/68fbc7f9de8fb80be5e7/

来源: python黑洞网

任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任

19 0
收藏该文
已收藏

评论内容:(最多支持255个字符)