Toybrick

ssd_mobilenet_v1_coco_2017_11_17.pb的输入节点问题

lucasling

注册会员

积分
149
发表于 2019-5-28 11:02:36    查看: 7564|回复: 9 | [复制链接]    打印 | 显示全部楼层
为什么文档上ssd_mobilenet_v1_coco_2017_11_17.pb的输入节点是'FeatureExtractor/MobilenetV1/MobilenetV1/Conv2d_0/BatchNorm/batchnorm/mul_1 输出节点是'concat', 'concat_1',我用summarize_graph的到的结果和文档不一样
回复

使用道具 举报

求大佬指教

中级会员

积分
241
发表于 2019-11-1 15:28:10 | 显示全部楼层
请问解决没有
回复

使用道具 举报

jefferyzhang

版主

积分
12851
发表于 2019-11-1 16:59:24 | 显示全部楼层
用Netron查看就知道了。
以Netron看到的name为准。
回复

使用道具 举报

lucasling

注册会员

积分
149
 楼主| 发表于 2019-11-16 19:13:06 | 显示全部楼层

解决了,用tensorflow看节点。多看看
回复

使用道具 举报

wujialiang

注册会员

积分
120
发表于 2020-2-17 16:59:42 | 显示全部楼层
jefferyzhang 发表于 2019-11-1 16:59
用Netron查看就知道了。
以Netron看到的name为准。

请问如何确定模型的输入与输出节点?节点很多,不知道如何确定。
回复

使用道具 举报

jefferyzhang

版主

积分
12851
发表于 2020-2-17 17:15:48 | 显示全部楼层
wujialiang 发表于 2020-2-17 16:59
请问如何确定模型的输入与输出节点?节点很多,不知道如何确定。

你如果输入输出节点都不知道的话,原模型是如何训练、导出、导入、推理的呢?
回复

使用道具 举报

wujialiang

注册会员

积分
120
发表于 2020-2-17 17:40:36 | 显示全部楼层
jefferyzhang 发表于 2020-2-17 17:15
你如果输入输出节点都不知道的话,原模型是如何训练、导出、导入、推理的呢?
...

转换的时候报下面的错误:

2020-02-17 17:32:19.894557: W tensorflow/core/framework/cpu_allocator_impl.cc:81] Allocation of 276480000 exceeds 10% of system memory.
E Unknow layer "tensorarraygatherv3"
E Try match TensorArrayGatherV3 BatchMultiClassNonMaxSuppression_1/map/TensorArrayStack/TensorArrayGatherV3 failed, catch exception!
E Catch exception when loading tensorflow model: ./frozen_inference_graph.pb!
E Traceback (most recent call last):
E   File "rknn\base\RKNNlib\converter\convert_tf.py", line 643, in rknn.base.RKNNlib.converter.convert_tf.convert_tf.match_paragraph_and_param
E   File "rknn\base\RKNNlib\RKNNnet.py", line 189, in rknn.base.RKNNlib.RKNNnet.RKNNNet.new_layer
E   File "rknn\base\RKNNlib\RKNNlog.py", line 327, in rknn.base.RKNNlib.RKNNlog.RKNNLog.e
E ValueError: Unknow layer "tensorarraygatherv3"
E During handling of the above exception, another exception occurred:
E Traceback (most recent call last):
E   File "rknn\api\rknn_base.py", line 303, in rknn.api.rknn_base.RKNNBase.load_tensorflow
E   File "rknn\base\RKNNlib\converter\convert_tf.py", line 742, in rknn.base.RKNNlib.converter.convert_tf.convert_tf.match_paragraph_and_param
E   File "rknn\base\RKNNlib\RKNNlog.py", line 327, in rknn.base.RKNNlib.RKNNlog.RKNNLog.e
E ValueError: Try match TensorArrayGatherV3 BatchMultiClassNonMaxSuppression_1/map/TensorArrayStack/TensorArrayGatherV3 failed, catch exception!
done

回复

使用道具 举报

jefferyzhang

版主

积分
12851
发表于 2020-2-17 17:50:52 | 显示全部楼层
wujialiang 发表于 2020-2-17 17:40
转换的时候报下面的错误:

2020-02-17 17:32:19.894557: W tensorflow/core/framework/cpu_allocator_im ...

这里的意思是: TensorArrayGatherV3 这个op不支持。

解决方法有几种:
1. 修改原模型,替换这个OP用其他方法
2. 修改rknn转换代码,使用自定义OP去实现该OP转换。
3. 换其他模型框架尝试下(例如caffe)
4. 如果这个op在结尾处,可以尝试把output节点往前挪,NPU计算返回后使用cpu来实现该OP
回复

使用道具 举报

AnnieLavigne

新手上路

积分
14
发表于 2020-3-17 18:49:00 | 显示全部楼层
wujialiang 发表于 2020-2-17 17:40
转换的时候报下面的错误:

2020-02-17 17:32:19.894557: W tensorflow/core/framework/cpu_allocator_im ...

你好,我也遇到同样问题。请问最后是如何解决的?求教!
回复

使用道具 举报

SodrSnne

中级会员

积分
249
发表于 2020-3-27 15:00:56 | 显示全部楼层
我也是这个节点/TensorArrayGatherV3,请问解决了吗
回复

使用道具 举报

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

本版积分规则

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


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