Toybrick

标题: ssd_mobilenet_v1_coco_2017_11_17.pb的输入节点问题 [打印本页]

作者: lucasling 时间: 2019-5-28 11:02
标题: ssd_mobilenet_v1_coco_2017_11_17.pb的输入节点问题
为什么文档上ssd_mobilenet_v1_coco_2017_11_17.pb的输入节点是'FeatureExtractor/MobilenetV1/MobilenetV1/Conv2d_0/BatchNorm/batchnorm/mul_1 输出节点是'concat', 'concat_1'，我用summarize_graph的到的结果和文档不一样

作者: 求大佬指教 时间: 2019-11-1 15:28
请问解决没有

作者: jefferyzhang 时间: 2019-11-1 16:59
用Netron查看就知道了。
以Netron看到的name为准。

作者: lucasling 时间: 2019-11-16 19:13

求大佬指教发表于 2019-11-1 15:28
请问解决没有

解决了，用tensorflow看节点。多看看

作者: wujialiang 时间: 2020-2-17 16:59

jefferyzhang 发表于 2019-11-1 16:59
用Netron查看就知道了。
以Netron看到的name为准。

请问如何确定模型的输入与输出节点？节点很多，不知道如何确定。

作者: jefferyzhang 时间: 2020-2-17 17:15

wujialiang 发表于 2020-2-17 16:59
请问如何确定模型的输入与输出节点？节点很多，不知道如何确定。

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

作者: wujialiang 时间: 2020-2-17 17:40

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 时间: 2020-2-17 17:50

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 时间: 2020-3-17 18:49

wujialiang 发表于 2020-2-17 17:40
转换的时候报下面的错误：

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

你好，我也遇到同样问题。请问最后是如何解决的？求教！

作者: SodrSnne 时间: 2020-3-27 15:00
我也是这个节点/TensorArrayGatherV3,请问解决了吗

欢迎光临 Toybrick (https://t.rock-chips.com/)