转efficientnet.tflite到rknn时出现异常,'NoneType' object has no attribu...

[复制链接] · 发表于 2020-3-31 21:16:36

D Convert layer conv_2d
W Rename efficientnet-lite0/model/head/Relu6;efficientnet-lite0/model/head/tpu_batch_normalization/FusedBatchNormV3;efficientnet-lite0/head/conv2d/kernel;efficientnet-lite0/model/head/conv2d to rknn_efficientnet-lite0/model/head/Relu6;efficientnet-lite0/model/head/tpu_batch_normalization/FusedBatchNormV3;efficientnet-lite0/head/conv2d/kernel;efficientnet-lite0/model/head/conv2d
W Rename efficientnet-lite0/model/head/Relu6;efficientnet-lite0/model/head/tpu_batch_normalization/FusedBatchNormV3;efficientnet-lite0/head/conv2d/kernel;efficientnet-lite0/model/head/conv2d to rknn_efficientnet-lite0/model/head/Relu6;efficientnet-lite0/model/head/tpu_batch_normalization/FusedBatchNormV3;efficientnet-lite0/head/conv2d/kernel;efficientnet-lite0/model/head/conv2d
D Convert layer average_pool_2d
D Convert layer reshape
E Catch exception when loading tflite model: ./efficientnet.tflite!
E Traceback (most recent call last):
E File "rknn/api/rknn_base.py", line 365, in rknn.api.rknn_base.RKNNBase.load_tflite
E File "rknn/base/RKNNlib/converter/tflite_loader.py", line 75, in rknn.base.RKNNlib.converter.tflite_loader.TFLiteLoader.load
E File "rknn/base/RKNNlib/converter/tflite_loader.py", line 393, in rknn.base.RKNNlib.converter.tflite_loader.ModelParser.parse
E File "rknn/base/RKNNlib/converter/tflite_loader.py", line 339, in rknn.base.RKNNlib.converter.tflite_loader.ModelParser._build_layers
E File "rknn/base/RKNNlib/converter/tflite_loader.py", line 235, in rknn.base.RKNNlib.converter.tflite_loader.ModelParser._get_layer_RKNN_op
E File "rknn/base/RKNNlib/converter/lite/RKNN_map_op.py", line 505, in rknn.base.RKNNlib.converter.lite.RKNN_map_op.RKNNMapOp.map_RKNN_ops
E File "rknn/base/RKNNlib/converter/lite/RKNN_map_op.py", line 299, in rknn.base.RKNNlib.converter.lite.RKNN_map_op._reshape
E AttributeError: 'NoneType' object has no attribute 'NewShapeAsNumpy'

只看该作者 · 发表于 2020-3-31 23:39:34

请将原始模型和转换脚本一起上传到百度网盘，然后链接发给我们，我提给NPU部门调试。还有请将以下信息提供给我们：

1. 转换时候 rknn_toolkit 版本号
2. 转换时候 tensorflow版本号

只看该作者 · 发表于 2020-4-1 12:48:48

jefferyzhang 发表于 2020-3-31 23:39
请将原始模型和转换脚本一起上传到百度网盘，然后链接发给我们，我提给NPU部门调试。还有请将以下信息提供 ...

链接：https://pan.baidu.com/s/17ZOc9E4nfheQqTEWcE8VTQ
提取码：weh1

TensorFlow-GPU 1.14
rknn_toolkit 1.3.0

只看该作者 · 发表于 2020-4-1 14:25:19

好的，问题已经提交了。
你的tf可以降到1.13.x试下，1.14会遇到一些op问题

只看该作者 · 发表于 2020-4-1 20:40:29

jefferyzhang 发表于 2020-4-1 14:25
好的，问题已经提交了。
你的tf可以降到1.13.x试下，1.14会遇到一些op问题

试了tf 1.13.1的, 报了同样的错误

只看该作者 · 发表于 2020-4-3 15:33:41

hanrb 发表于 2020-4-1 20:40
试了tf 1.13.1的, 报了同样的错误

有更新了吗

只看该作者 · 发表于 2020-4-10 08:49:47

本帖最后由 jefferyzhang 于 2020-4-10 19:41 编辑

hanrb 发表于 2020-4-3 15:33
有更新了吗

NPU部门回复：

转换rknn失败的bug我们这边已经解决了，efficientnet-lite0-fp32.tflite模型在tflite runtime上的top5与rknn runtime的top5结果基本一致，efficientnet-lite0-int8.tflite模型还在验证中，验证成功后，会第一时间提供rknn-toolkit给您。
在调试过程发现,efficientnet-lite0-int8.tflite模型使用int8非对称量化,目前我们的toolkit暂时不支持.如果您还需要支持浮点模型的转换,我们可以提供一个toolkit-beta给您

rknn-toolkit-1.3.1-beta版本
链接: https://pan.baidu.com/s/1pUBknzDTlik1_9qQ2SS2uQ 密码: gd25

Update:
我们的toolkit目前支持从efficientnet-lite0-fp32.tflite模型的量化，模型转换等功能。验证过官方的panda.jpg图片top-1正确。

Update:
支持 efficientnet-lite0-fp32.tflite 量化的工具如下
链接: https://pan.baidu.com/s/1B7nJGC-qBnTH0WDju9wHgQ 密码: 3lkh

只看该作者 · 发表于 2020-4-20 15:03:56

jefferyzhang 发表于 2020-4-10 08:49
NPU部门回复：

转换rknn失败的bug我们这边已经解决了，efficientnet-lite0-fp32.tflite模型在tflite runt ...

好的,谢谢