Toybrick

标题: 加载模型错误 [打印本页]

作者: mailonghua 时间: 2019-10-30 13:56
标题: 加载模型错误
在加载yolov3模型时，出现报错，关于输入和输出节点也详细的检查了one
--> Loading model
E Catch exception when loading tensorflow model: ./yolo_v3_surgery.pb!
T Traceback (most recent call last):
T File "rknn/api/rknn_base.py", line 188, in rknn.api.rknn_base.RKNNBase.load_tensorflow
T File "rknn/base/RKNNlib/converter/convert_tf.py", line 639, in rknn.base.RKNNlib.converter.convert_tf.convert_tf.match_paragraph_and_param
T AttributeError: 'LeakyRelu' object has no attribute 'load_params_from_tf'

作者: mailonghua 时间: 2019-10-30 14:57

hisping 发表于 2019-10-30 14:32
论坛上有yolov3的教程，你这是自己用tensorflow写的yolov3模型，然后转rknn是吗？
这个错误有可能是你训练t ...

下载的预编译模型，这个有没有可能是当前的rknn不支持这个算子（LeakyRelu）的转换呢

作者: jefferyzhang 时间: 2019-10-30 16:59

mailonghua 发表于 2019-10-30 14:57
下载的预编译模型，这个有没有可能是当前的rknn不支持这个算子（LeakyRelu）的转换呢 ...

这个算子是支持的

作者: mailonghua 时间: 2019-10-30 18:09

jefferyzhang 发表于 2019-10-30 16:59
这个算子是支持的

有没有怀疑的原因呢？目前的tf版本是1.14
看了其他帖子，很多都是直接使用darknet
现在使用的yoloV3.pb

作者: jefferyzhang 时间: 2019-10-30 18:17
你有开verbose=True么，你这里截取的打印看不出原因

作者: mailonghua 时间: 2019-10-31 09:50

jefferyzhang 发表于 2019-10-30 18:17
你有开verbose=True么，你这里截取的打印看不出原因

1 D import clients finished
  2 W:tensorflow:From /home/longhuamai/.local/lib/python3.6/site-packages/rknn/api/rknn.py:68: extract _sub_graph (from tensorflow.python.framework.graph_util_impl) is deprecated and will be removed in    a future version.
  3 Instructions for updating:
  4 Use `tf.compat.v1.graph_util.extract_sub_graph`
  5 I Current TF Model producer version 0 min consumer version 0 bad consumer version []
  6 I Have 0 tensors convert to const tensor
  7 []
  8 I build output layer attach_yolov3/yolov3_head/feature_map_1

ut0
9 I build output layer attach_yolov3/yolov3_head/feature_map_2

ut0
10 I build output layer attach_yolov3/yolov3_head/feature_map_3

ut0
11 I build input layer Placeholder

ut0
12 D Try match Identity yolov3/yolov3_head/feature_map_1
13 I Match identity [['yolov3/yolov3_head/feature_map_1']] [['Identity']] to [['noop']]
14 D Try match Identity yolov3/yolov3_head/feature_map_2
15 I Match identity [['yolov3/yolov3_head/feature_map_2']] [['Identity']] to [['noop']]
16 D Try match Identity yolov3/yolov3_head/feature_map_3
17 I Match identity [['yolov3/yolov3_head/feature_map_3']] [['Identity']] to [['noop']]
18 D Try match BiasAdd yolov3/yolov3_head/Conv_6/BiasAdd
19 I Match convolution_biasadd [['yolov3/yolov3_head/Conv_6/BiasAdd', 'yolov3/yolov3_head/Conv_6/Conv 2D', 'yolov3/yolov3_head/Conv_6/biases/read/_341__cf__341', 'yolov3/yolov3_head/Conv_6/weights/rea d/_342__cf__342']] [['BiasAdd', 'Conv', 'C', 'C_1']] to [['convolution']]
20 D Try match BiasAdd yolov3/yolov3_head/Conv_14/BiasAdd
21 I Match convolution_biasadd [['yolov3/yolov3_head/Conv_14/BiasAdd', 'yolov3/yolov3_head/Conv_14/Co nv2D', 'yolov3/yolov3_head/Conv_14/biases/read/_284__cf__284', 'yolov3/yolov3_head/Conv_14/weights /read/_285__cf__285']] [['BiasAdd', 'Conv', 'C', 'C_1']] to [['convolution']]
22 D Try match BiasAdd yolov3/yolov3_head/Conv_22/BiasAdd
23 I Match convolution_biasadd [['yolov3/yolov3_head/Conv_22/BiasAdd', 'yolov3/yolov3_head/Conv_22/Co nv2D', 'yolov3/yolov3_head/Conv_22/biases/read/_326__cf__326', 'yolov3/yolov3_head/Conv_22/weights /read/_327__cf__327']] [['BiasAdd', 'Conv', 'C', 'C_1']] to [['convolution']]
24 D Try match LeakyRelu yolov3/yolov3_head/Conv_5/LeakyRelu
25 E Catch exception when loading tensorflow model: ./yolo_v3_surgery.pb!
26 T Traceback (most recent call last):
27 T File "rknn/api/rknn_base.py", line 188, in rknn.api.rknn_base.RKNNBase.load_tensorflow
28 T File "rknn/base/RKNNlib/converter/convert_tf.py", line 639, in rknn.base.RKNNlib.converter.con vert_tf.convert_tf.match_paragraph_and_param
29 T AttributeError: 'LeakyRelu' object has no attribute 'load_params_from_tf'

作者: jefferyzhang 时间: 2019-10-31 10:21
你的这个pb文件用tensorflow读取过能正常推理么？

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