Toybrick

标题: tflite转换rknn模型失败 [打印本页]

---

作者: 板蓝根    时间: 2019-8-15 17:37
标题: tflite转换rknn模型失败

软件环境

操作系统: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

---

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

Powered by Discuz! X3.3