Toybrick

tensorflow keras模型转rknn报错,如何解决?

jefferyzhang

版主

积分
13001
楼主
发表于 2019-10-30 14:42:11 | 显示全部楼层
W Not match node batch_normalization_121/cond/FusedBatchNormV3_1 FusedBatchNormV3

OP不支持
回复

使用道具 举报

jefferyzhang

版主

积分
13001
沙发
发表于 2019-10-30 15:36:19 | 显示全部楼层
farces 发表于 2019-10-30 15:30
是不支持batch_normalization批归一化吗?四个网络的这个操作都不支持?

FusedBatchNormV3 和 BatchNorm 根本不是一个东西吧。。。
回复

使用道具 举报

jefferyzhang

版主

积分
13001
板凳
发表于 2019-10-30 15:40:56 | 显示全部楼层
farces 发表于 2019-10-30 15:02
Keras支持 SqueezeNet,ResNet,InceptionV3 和 DenseNet等四个网络,上边的报错是DenseNet的,请问另外 ...

SqueezeNet、DenseNet 我没转过,你要试试
ResNet,InceptionV3 都是转过OK的。

我们现在支持自定义OP,转换有问题的OP要么换掉,要么自己写就可以了。
回复

使用道具 举报

jefferyzhang

版主

积分
13001
地板
发表于 2019-10-31 16:33:48 | 显示全部楼层
farces 发表于 2019-10-31 16:29
"ResNet,InceptionV3 都是转过OK的"
确认下这两个网络训练和转换用的tensorflow的版本是哪个?我们用的1 ...

跟版本无关的,只要pb文件格式没变,用哪个tf版本生成出来都一样。
倒是rknntoolkit版本最好升级到最新1.2.1的转,tf配合就装1.14.0的
回复

使用道具 举报

jefferyzhang

版主

积分
13001
5#
发表于 2019-10-31 17:35:07 | 显示全部楼层
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/framework/importer.py", line 426, in import_graph_def

你没发现这是tf报的错么。都还没到rknn转换那步。建议你们pb模型一定要先用tf读出来推理正常了再转rknn。
回复

使用道具 举报

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

本版积分规则

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


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