Toybrick

tensorflow pb转化rknn报错

jefferyzhang

版主

积分
12996
楼主
发表于 2019-11-10 21:50:27 | 显示全部楼层
1. 生成该pb格式模型的tf是什么版本?
2. 读取pb文件用的tf是什么版本,是否可用?
3. 转换rknn使用的tf是什么版本?

我们目前rknn支持1.14以及1.14以下版本的tf,但是生成pb和转换rknn的tf必须版本一致。
并且你需要先用tf去尝试读下这个pb文件,目前这个错很可能是tf无法读取出这个模型
回复

使用道具 举报

jefferyzhang

版主

积分
12996
沙发
发表于 2019-11-11 14:45:19 | 显示全部楼层
wangshuyi 发表于 2019-11-11 14:25
# Build model
    print('--> Building model')
    ret = rknn.build(do_quantization=True, dataset=' ...

什么叫转换时候用keras2.2.4? keras是tf自带的 tf.keras,而不是单独下的keras包。
量化报错的log发出来。
也可以先关闭量化来看下rknn是否可以用,最后再考虑量化。
回复

使用道具 举报

jefferyzhang

版主

积分
12996
板凳
发表于 2019-11-12 08:19:11 | 显示全部楼层
wangshuyi 发表于 2019-11-11 21:03
各个python依赖包的版本,关闭量化转化成功了

嗯,那就没问题,先测试不量化结果是否是对的。最后再搞量化。
量化请参看文档
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表