Toybrick

楼主: WuKing

SSD 模型 rknn.load_tensorflow 输入输出节点问题

jefferyzhang

版主

积分
12953
楼主
发表于 2019-7-12 16:37:16 | 显示全部楼层
没看懂你问什么,节点名字用Netron看会比较准确
回复

使用道具 举报

jefferyzhang

版主

积分
12953
沙发
发表于 2019-7-18 10:03:29 | 显示全部楼层
WuKing 发表于 2019-7-16 16:28
rknn.load_tensorflow(tf_pb='./ssd_mobilenet_v1_coco_2017_11_17.pb',
                         i ...

rk的模型是有改过的,不要跟源码去比对,没有意义
回复

使用道具 举报

jefferyzhang

版主

积分
12953
板凳
发表于 2019-11-1 14:27:47 | 显示全部楼层
求大佬指教 发表于 2019-11-1 14:09
请问input和outputs的节点参数该怎么确定呢?
我看有人是下面这么说的:

如果你不会用tf看的话,可以下一个netron来查看各个节点的名字
回复

使用道具 举报

jefferyzhang

版主

积分
12953
地板
发表于 2019-11-1 15:43:03 | 显示全部楼层
求大佬指教 发表于 2019-11-1 15:03
我就是用netron查看的,比如下面这个例子
一个输入:

你就不能单独开个帖子问么

首先你这里报的错并不是input output name问题。
其次你可以吧verbose=True打开来看节点匹配问题。
再次看你log应该是pred_sbbox/exp 这个节点不支持造成的
回复

使用道具 举报

jefferyzhang

版主

积分
12953
5#
发表于 2020-6-10 16:45:12 | 显示全部楼层
奥古师弟 发表于 2020-6-10 16:07
如果不要和源码对比的话,请问对于一个新的模型,在使用rknn.load_tensorflow的时候,输入和输出节点该如 ...

如果是你自己写的tensorflow模型,你是不可能不知道input output name的,因为你自己用tf加载pb时候也要填这几个参数。
如果是别人模型,那你就用netron或者tensorboard看了。一样的道理,你也要先保证当前tf版本能正常载入pb才能去转rknn,不然回头一堆问题都不好定位
回复

使用道具 举报

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

本版积分规则

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


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