tflite转rknn模型失败

[复制链接] · 发表于 2019-10-1 21:12:13

我用TF2.0的Keras训练出.h5格式的模型，将其转换为tflite后，再将其转换为rknn，转换代码用的是minist量化里提供的代码，报错信息如图

只看该作者 · 发表于 2019-10-1 21:14:27

图片报错1是转换代码

只看该作者 · 发表于 2019-10-2 01:33:19

你的模型里有些算子是rknn不支持的

只看该作者 · 发表于 2019-10-2 08:14:45

protossw512 发表于 2019-10-2 01:33
你的模型里有些算子是rknn不支持的

那是要换低版本的tensorflow重新训练吗

只看该作者 · 发表于 2019-10-2 15:38:00

protossw512 发表于 2019-10-2 01:33
你的模型里有些算子是rknn不支持的

用低版本的tensorflow训练的模型可以转换了，但转换后的推理输出（one-hot多分类）都是1，不知道为什么

只看该作者 · 发表于 2019-10-2 16:55:06

Samuel 发表于 2019-10-2 15:38
用低版本的tensorflow训练的模型可以转换了，但转换后的推理输出（one-hot多分类）都是1，不知道为什么 ...

你换的版本有多低？让我学习一下，避免以后踩坑。

只看该作者 · 发表于 2019-10-3 08:26:17

agogos 发表于 2019-10-2 16:55
你换的版本有多低？让我学习一下，避免以后踩坑。

用的tf1.10，开发板也装的tf1.10

只看该作者 · 发表于 2019-10-3 10:40:45

Samuel 发表于 2019-10-3 08:26
用的tf1.10，开发板也装的tf1.10

谢谢！:

。。。。。。