Toybrick

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

WuKing

注册会员

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

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



回复

使用道具 举报

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模型中取出的输入,输出节点不一致。所以有点困惑
回复

使用道具 举报

WuKing

注册会员

积分
67
板凳
 楼主| 发表于 2019-7-25 10:39:37 | 显示全部楼层
摇太阳 发表于 2019-7-25 10:26
楼主你好 请问你解决了嘛
我也遇到了同样的问题了 请问怎么解决这个问题的呀
...

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

使用道具 举报

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

本版积分规则

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


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