将tensorflow模型转换为tensorflow lite时出错(从.h5到.tflite) = ValueError:错误的编组数据(未知类型代码)
将tensorflow模型转换为tensorflow lite时出错(从.h5到.tflite) = ValueError:错误的编组数据(未知类型代码)
提问于 2020-04-13 22:57:00
我正在使用带有yolov2架构的基于keras和tensorflow的预训练模型来检测凹坑，在将tensorflow模型转换为tensorflow lite时，我得到了一个错误
from tensorflow.contrib import lite
converter = lite.TFLiteConverter.from_keras_model_file( 'model.h5' )
model = converter.convert()
file = open( 'model.tflite' , 'wb' )
file.write( model )回答 1
Stack Overflow用户
发布于 2020-05-07 16:13:02
我建议走一条更长但经过测试的道路。
1)直接将模型转换为.pb冻结图形格式或使用How to export Keras .h5 to tensorflow .pb?从h5转换为.pb
2)按照我在这里描述的那样，将.pb转换为tflite：https://medium.com/@prashantdandriyal7/darknet-to-tensorflow-to-tensorflowlite-b783d55b106a
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/61190619
复制相关文章
相似问题
腾讯云开发者
