【rknn】onnx转rknn load_onnx函数报错: KeyError ‘output‘

阿颖&阿伟

已于 2024-02-22 14:30:21 修改

阅读量1.3k

点赞数 2

分类专栏：【11-1】rknn开发板文章标签：深度学习人工智能

于 2022-11-24 14:52:21 首次发布

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接：https://blog.csdn.net/sazass/article/details/128018409

版权

【11-1】rknn开发板专栏收录该内容

5 篇文章

订阅专栏

博客内容涉及ONNX模型的输出层问题及其解决方法。通过使用Netron工具查看模型的输出层名称，特别是Reshape层的OUTPUTS，确认并修正了在rknn.load_onnx函数中outputs参数的设置，以确保正确加载模型。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

报错：
在这里插入图片描述

原因：

输出层的设置不对

解决方式

1. 查看onnx模型的输出层名称：

网页输入netron.app：
https://netron.app/

在网站内打开自己的onnx模型：
在这里插入图片描述
然后找到模型最后的Reshape层:

分别单击Reshape模块，查看OUTPUTS的name

然后name自左往右分别是：338, 376, 414

2. 修改rknn.load_onnx函数的参数

修改load_onnx函数终端outputs的参数
在这里插入图片描述

评论 3

被折叠的条评论为什么被折叠?

到【灌水乐园】发言

查看更多评论

添加红包

成就一亿技术人!

hope_wisdom

发出的红包

实付元

使用余额支付

点击重新获取

扫码支付

钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。