Toybrick

rknn转出后推理错误

wokafu

注册会员

积分
106
发表于 2022-4-19 10:20:10    查看: 2065|回复: 3 | [复制链接]    打印 | 显示全部楼层
1. onnx推理结果跟pytorch推理结果一致,onnx转rknn后,rknn推理完成错误,打开rknn的各个卷积的weights看起来又是对的,在运行调试的时候提示:W Inputs should be placed in a list, like [img1, img2], both the img1 and img2 are ndarray

麻烦各位老师指点迷津,谢谢了!

本帖子中包含更多资源

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

x
回复

使用道具 举报

jefferyzhang

版主

积分
12930
发表于 2022-4-19 10:34:54 | 显示全部楼层
warning不用在意
回复

使用道具 举报

wokafu

注册会员

积分
106
 楼主| 发表于 2022-4-19 10:49:08 | 显示全部楼层
onnx推理是正确的,rknn推理完全不正确,请问这个一般是哪个步骤错误导致的呢?大佬!
打开rknn文件,各个层的weight参数感觉又是对的
回复

使用道具 举报

wokafu

注册会员

积分
106
 楼主| 发表于 2022-4-19 13:23:00 | 显示全部楼层
结题了,输入的时候图片处理得不对,各种变换处理后,推理正确了!在inference的参数指定data_format="nchw"
回复

使用道具 举报

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

本版积分规则

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


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