Toybrick

标题: RKNN 转换 mobile V1 ssd 报错，请问这个是为什么？已经使用. [打印本页]

作者: linuxsky 时间: 2019-2-18 11:22
标题: RKNN 转换 mobile V1 ssd 报错，请问这个是为什么？已经使用.
本帖最后由 linuxsky 于 2019-2-18 11:51 编辑

$ python3 ssd.py
W verbose file path is invalid, debug info will not dump to file.
--> Loading model
D import clients finished
E Catch exception when loading tensorflow model: ./frozen_inference_graph.pb!
T Traceback (most recent call last):
T File "rknn/api/rknn_base.py", line 143, in rknn.api.rknn_base.RKNNBase.load_tensorflow
T File "rknn/base/rknnlib/converter/convert_tf.py", line 93, in rknn.base.rknnlib.converter.convert_tf.convert_tf.__init__
T File "rknn/base/rknnlib/converter/tensorflowloader.py", line 53, in rknn.base.rknnlib.converter.tensorflowloader.TF_Graph_Preprocess.__init__
T AttributeError: 'NoneType' object has no attribute 'op'
done
--> Building model
2019-02-18 10:59:07.223537: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA
E Catch exception when building RKNN model!
T Traceback (most recent call last):
T File "rknn/api/rknn_base.py", line 470, in rknn.api.rknn_base.RKNNBase.build
T File "rknn/api/rknn_base.py", line 874, in rknn.api.rknn_base.RKNNBase._quantize
T File "rknn/base/tensorzonex.py", line 213, in rknn.base.tensorzonex.get_data_file
T File "/usr/lib/python3.5/posixpath.py", line 122, in splitext
T    return genericpath._splitext(p, sep, None, extsep)
T File "/usr/lib/python3.5/genericpath.py", line 118, in _splitext
T    sepIndex = p.rfind(sep)
T AttributeError: 'NoneType' object has no attribute 'rfind'
done
--> Init runtime environment
done
--> Running model
E RKNN model is None
done
inference result:  None
Traceback (most recent call last):
  File "ssd.py", line 108, in <module>
predictions = outputs[1].reshape((1, NUM_RESULTS, 4))
TypeError: 'NoneType' object is not subscriptable

作者: yhc 时间: 2019-2-18 12:05
确认是ssd_mobilenet_v1？
如果是ssd_mobilenet_v2的input节点有点不一样

    rknn.load_tensorflow(tf_pb='./ssd_mobilenet_v2_coco_2018_03_29/frozen_inference_graph.pb',

                         inputs=['FeatureExtractor/MobilenetV2/MobilenetV2/input'],

                         outputs=['concat', 'concat_1'],

                         input_size_list=[[INPUT_SIZE, INPUT_SIZE, 3]])
复制代码

作者: linuxsky 时间: 2019-2-18 13:47

yhc 发表于 2019-2-18 12:05
确认是ssd_mobilenet_v1？
如果是ssd_mobilenet_v2的input节点有点不一样

果然是训练的有问题。。。V2 确实改了 iput 就可以了。
顺便问下，怎么确定模型的 input output 呢？

作者: yhc 时间: 2019-2-18 15:53
可以用netron打开pb模型看网络结构，取这个节点是有去掉了一些前处理

作者: chuyee 时间: 2019-2-19 15:24

yhc 发表于 2019-2-18 15:53
可以用netron打开pb模型看网络结构，取这个节点是有去掉了一些前处理

作者: linuxsky 时间: 2019-2-19 15:53

yhc 发表于 2019-2-18 15:53
可以用netron打开pb模型看网络结构，取这个节点是有去掉了一些前处理

我刚才安装了这个工具，打开后看到大网络，但是一般看到输入不就是 image 么？为什么要选择 conv2D 呢?
是自己随便选择输入输出的节点么？
做嵌入式开发的，小白求大神不吝赐教下。

作者: yhc 时间: 2019-2-19 16:24
一般的话要自己熟悉网络的结构，就ssd_mobilenet_v2网络来说，前处理的话都是Preprocessor开头的，所以就跳过这些Preprocessor的节点，取到
FeatureExtractor/MobilenetV2/MobilenetV2/input，因为这些前处理节点如果用NPU来跑并不是效率最高的，可以用cpu来替代处理

作者: linuxsky 时间: 2019-2-19 16:47

yhc 发表于 2019-2-19 16:24
一般的话要自己熟悉网络的结构，就ssd_mobilenet_v2网络来说，前处理的话都是Preprocessor开头的，所以就跳 ...

好的多谢，搞嵌入式的不懂这些。。。。
能加您个微信不，有啥不懂的跟请教学学
我的微信号 linuxsky

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