获取输入数组和输出数组项以将模型转换为tflite格式
PS.请不要将我指向converting Keras model directly to tflite,因为我的.h5文件无法直接转换为.tflite.我以某种方式设法将我的.h5文件转换为.pb 我关注了this Jupyter笔记本,使用Keras进行面部识别.然后,我将模型保存到model.h5文件,然后使用this将其转换为冻结图model.pb. 现在我想在Android中使用我的tensorflow文件.为此,我需要Tensorflow Lite,这需要我将模型转换为.tflite格式. 为此,我尝试遵循其官方指南here.如您所见,它需要input_array和output_array数组.如何从我的model.pb文件中获得这些内容的详细信息? 最佳答案
输入数组和输出数组是分别存储输入和输出张量的数组.
对于Keras模型, 输入张量是第一层的占位符张量.
输出张量可以与激活函数有关.
对于冻结图,
我们得到节点的名称,
为了得到张量,
输入张量可以是占位符张量.输出张量是您使用session.run()运行的张量 对于转换,我们得到
(编辑:李大同) 【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容! |