Toybrick

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

hanrb

新手上路

积分
46
发表于 2020-3-31 21:16:36    查看: 6623|回复: 7 | [复制链接]    打印 | 显示全部楼层
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'


回复

使用道具 举报

jefferyzhang

版主

积分
12844
发表于 2020-3-31 23:39:34 | 显示全部楼层
请将原始模型和转换脚本一起上传到百度网盘,然后链接发给我们,我提给NPU部门调试。还有请将以下信息提供给我们:

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

回复

使用道具 举报

hanrb

新手上路

积分
46
 楼主| 发表于 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
回复

使用道具 举报

jefferyzhang

版主

积分
12844
发表于 2020-4-1 14:25:19 | 显示全部楼层
好的,问题已经提交了。
你的tf可以降到1.13.x试下,1.14会遇到一些op问题
回复

使用道具 举报

hanrb

新手上路

积分
46
 楼主| 发表于 2020-4-1 20:40:29 | 显示全部楼层
jefferyzhang 发表于 2020-4-1 14:25
好的,问题已经提交了。
你的tf可以降到1.13.x试下,1.14会遇到一些op问题

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

使用道具 举报

hanrb

新手上路

积分
46
 楼主| 发表于 2020-4-3 15:33:41 | 显示全部楼层
hanrb 发表于 2020-4-1 20:40
试了tf 1.13.1的, 报了同样的错误

有更新了吗
回复

使用道具 举报

jefferyzhang

版主

积分
12844
发表于 2020-4-10 08:49:47 | 显示全部楼层
本帖最后由 jefferyzhang 于 2020-4-10 19:41 编辑

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
回复

使用道具 举报

hanrb

新手上路

积分
46
 楼主| 发表于 2020-4-20 15:03:56 | 显示全部楼层
jefferyzhang 发表于 2020-4-10 08:49
NPU部门回复:

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

好的,谢谢
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表