Toybrick

tflite转换rknn模型失败

板蓝根

注册会员

积分
187
楼主
发表于 2019-8-15 17:37:37    查看: 8733|回复: 0 | [复制链接]    打印 | 只看该作者
软件环境
操作系统:Fedroa
rknn_tookit:0.9.9
rknn_api:0.9.5
npu的boot_img:0.9.5


官方文档说明是支持tflite模型转到rknn模型,因此我在训练检测器的时候采用了quantization-aware training。再转换成tflite model。在这儿我做了两种处理,均不能通过实验:
1.export_tflite_ssd_graph.py 生成pb文件时加入 --add_postprocessing_op=True,最后在转tflite模型时设置--output_arrays='TFLite_Detection_PostProcess','TFLite_Detection_PostProcess:1','TFLite_Detection_PostProcess:2','TFLite_Detection_PostProcess:3'。但是将该tflite模型转换为rknn模型时会报错,报错:AttributeError: 'NoneType' object has no attribute 'to_string', 具体log见附件build1.log
2.export_tflite_ssd_graph.py 生成pb文件时加入 --add_postprocessing_op=False,最后在转tflite模型时设置--output_arrays='concat','concat_1'。这样能够转换成rknn model。但是在调用inference时候又会出现:RKNN init failed. error code: RKNN_ERR_MODEL_INVALID,具体log见附件build2.log



本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

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

本版积分规则

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


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