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

求大佬指教 发表于 2019-11-1 15:03
我就是用netron查看的，比如下面这个例子
一个输入：

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

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

你好！在pc机上安装rknn-toolkit 运行rknn_convert 将示例caffe模型转成rknn 。报错
E Catch exception when loading caffe model: ./models/caffe/mobilenet_v2/../../../../mobilenet_v2/mobilenet_v2.prototxt!
T Traceback (most recent call last):
T   File "rknn/api/rknn_base.py", line 283, in rknn.api.rknn_base.RKNNBase.load_caffe
T   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 72, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.__init__
T   File "rknn/base/RKNNlib/RKNN_session.py", line 19, in rknn.base.RKNNlib.RKNN_session.RKNNSession.__init__
T AttributeError: module 'tensorflow' has no attribute 'ConfigProto'
caffe end
done
--> Build RKNN model...
E Catch exception when building RKNN model!
T Traceback (most recent call last):
T   File "rknn/api/rknn_base.py", line 498, in rknn.api.rknn_base.RKNNBase.build
T   File "rknn/api/rknn_base.py", line 1056, in rknn.api.rknn_base.RKNNBase._quantize
T   File "rknn/base/RKNNlib/console/utils.py", line 7, in rknn.base.RKNNlib.console.utils.get_model_data_file
T   File "/home/xah/anaconda3/lib/python3.6/posixpath.py", line 122, in splitext
T     p = os.fspath(p)
T TypeError: expected str, bytes or os.PathLike object, not NoneType
done

jefferyzhang 发表于 2019-7-18 10:03
rk的模型是有改过的，不要跟源码去比对，没有意义

如果不要和源码对比的话，请问对于一个新的模型，在使用rknn.load_tensorflow的时候，输入和输出节点该如何确定呀。

奥古师弟 发表于 2020-6-10 16:07
如果不要和源码对比的话，请问对于一个新的模型，在使用rknn.load_tensorflow的时候，输入和输出节点该如 ...

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