模型转化lite格式报错Output array not found: MobilenetV1/Predictions/Reshape_1

最新推荐文章于 2024-07-26 15:48:33 发布

kkongyu

最新推荐文章于 2024-07-26 15:48:33 发布

阅读量1.2k

点赞数

CC 4.0 BY-SA版权

分类专栏： tensorflow android

本文链接：https://blog.youkuaiyun.com/ting_163/article/details/81297636

这篇博客主要记录了在将模型转化为Lite格式时遇到的两个问题。第一个问题是出现'Output array not found: MobilenetV1/Predictions/Reshape_1'的错误，作者通过检查并修改保存pb文件的代码，将--output_arrays=MobilenetV1/Predictions/Reshape_1改为--output_arrays=output成功解决了问题。第二个问题的报错信息和解决方案未详细给出。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题一

描述：

~/tensorflow-master$ bazel-bin/tensorflow/contrib/lite/toco/toco  
--input_file=/home/tclxa/下载/TensorFlowMnist-master/output/mnist-tf1.0.1.pb  
--input_format=TENSORFLOW_GRAPHDEF  
--output_format=TFLITE   
--output_file=/tmp/mobilenet_v1_1.0_224.lite 
--inference_type=FLOAT   
--input_type=FLOAT 
--input_arrays=input   
<