Toybrick

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

WuKing

注册会员

积分
67
楼主
发表于 2019-7-12 10:48:38    查看: 31881|回复: 13 | [复制链接]    打印 | 只看该作者
mobilenet_v1 例程中, 输入输出节点从网络中间截取的,这是为什么呢?我看模型原来的输入,输出应该是: inputs=['image_tensor'],
outputs=['detection_boxes','detection_scores','num_detections','detection_classes'],

是由于自定义的OP 不支持转化吗?望知情的朋友解惑下,谢谢



回复

使用道具 举报

jefferyzhang

版主

积分
13578
沙发
发表于 2019-7-12 16:37:16 | 只看该作者
没看懂你问什么,节点名字用Netron看会比较准确
回复

使用道具 举报

WuKing

注册会员

积分
67
板凳
 楼主| 发表于 2019-7-16 16:28:11 | 只看该作者
jefferyzhang 发表于 2019-7-12 16:37
没看懂你问什么,节点名字用Netron看会比较准确

    rknn.load_tensorflow(tf_pb='./ssd_mobilenet_v1_coco_2017_11_17.pb',
                         inputs=['FeatureExtractor/MobilenetV1/MobilenetV1/Conv2d_0/BatchNorm/batchnorm/mul_1'],
                         outputs=['concat', 'concat_1'],
                         input_size_list=[[INPUT_SIZE, INPUT_SIZE, 3]])

你好,我前面描述有误,其实是mobilenet-ssd的模型, 在load的时候,这个inputs 和outputs参数指定的节点和 我从ssd_mobilenet_v1_coco_2017_11_17.pb模型中取出的输入,输出节点不一致。所以有点困惑
回复

使用道具 举报

jefferyzhang

版主

积分
13578
地板
发表于 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的模型是有改过的,不要跟源码去比对,没有意义
回复

使用道具 举报

摇太阳

注册会员

积分
132
5#
发表于 2019-7-25 10:26:45 | 只看该作者
WuKing 发表于 2019-7-16 16:28
rknn.load_tensorflow(tf_pb='./ssd_mobilenet_v1_coco_2017_11_17.pb',
                         i ...

楼主你好 请问你解决了嘛
我也遇到了同样的问题了 请问怎么解决这个问题的呀
回复

使用道具 举报

WuKing

注册会员

积分
67
6#
 楼主| 发表于 2019-7-25 10:39:37 | 只看该作者
摇太阳 发表于 2019-7-25 10:26
楼主你好 请问你解决了嘛
我也遇到了同样的问题了 请问怎么解决这个问题的呀
...

我是直接按照例程的输入输出节点来的,如果预训练模型一样的话,输入输出也是一样的
回复

使用道具 举报

摇太阳

注册会员

积分
132
7#
发表于 2019-7-26 08:40:48 | 只看该作者
WuKing 发表于 2019-7-25 10:39
我是直接按照例程的输入输出节点来的,如果预训练模型一样的话,输入输出也是一样的 ...

你好 请问你的训练模型怎么训练出来的呀 可以给一个教程嘛?
谢谢啦
回复

使用道具 举报

求大佬指教

中级会员

积分
241
8#
发表于 2019-11-1 14:09:38 | 只看该作者
本帖最后由 求大佬指教 于 2019-11-1 14:19 编辑
jefferyzhang 发表于 2019-7-18 10:03
rk的模型是有改过的,不要跟源码去比对,没有意义

请问input和outputs的节点参数该怎么确定呢?
我看有人是下面这么说的:

那么这两个op.name在哪里看呢?

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

jefferyzhang

版主

积分
13578
9#
发表于 2019-11-1 14:27:47 | 只看该作者
求大佬指教 发表于 2019-11-1 14:09
请问input和outputs的节点参数该怎么确定呢?
我看有人是下面这么说的:

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

使用道具 举报

求大佬指教

中级会员

积分
241
10#
发表于 2019-11-1 15:03:43 | 只看该作者
jefferyzhang 发表于 2019-11-1 14:27
如果你不会用tf看的话,可以下一个netron来查看各个节点的名字

我就是用netron查看的,比如下面这个例子
一个输入:


三个输出:



我是这么写的

inputs = ['input/input_data'],
outputs = ['pred_sbbox/concat_2','pred_lbbox/concat_2','pred_mbbox/concat_2']
但是报错

求教!谢谢!

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

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

本版积分规则

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


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