Toybrick

标题: tflite转rknn模型失败 [打印本页]

作者: Samuel 时间: 2019-10-1 21:12
标题: tflite转rknn模型失败
我用TF2.0的Keras训练出.h5格式的模型，将其转换为tflite后，再将其转换为rknn，转换代码用的是minist量化里提供的代码，报错信息如图

作者: Samuel 时间: 2019-10-1 21:14
图片报错1是转换代码

作者: protossw512 时间: 2019-10-2 01:33
你的模型里有些算子是rknn不支持的

作者: Samuel 时间: 2019-10-2 08:14

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

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

作者: Samuel 时间: 2019-10-2 15:38

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

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

作者: agogos 时间: 2019-10-2 16:55

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

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

作者: Samuel 时间: 2019-10-3 08:26

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

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

作者: agogos 时间: 2019-10-3 10:40

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

谢谢！:

。。。。。。

欢迎光临 Toybrick (https://t.rock-chips.com/)