Toybrick

rknn多图推理遇到问题

自由的学习

注册会员

积分
174
楼主
发表于 2023-9-18 14:47:18    查看: 2185|回复: 1 | [复制链接]    打印 | 只看该作者
我的batch_size为2,推理采用了同一张图片进行拼接输入,最后推理出来的两个结果不一样,前一张是对的,后一张是错的,但是在onnx模型下两个结果都是正确的,转为rknn后就出现了这种情况,不知是什么原因造成的
回复

使用道具 举报

自由的学习

注册会员

积分
174
沙发
 楼主| 发表于 2023-9-18 17:54:20 | 只看该作者
又采用两张不同的图片分别跑了下onnx和rknn模型,onnx的结果都是对的且输入顺序为图1图2 输出结果的list中也是一致的顺序;而rknn还是一个结果对一个结果错,输出list中的结果顺序与输入图片顺序不一致,第一个结果是图二的,第二个结果都不确定是不是图一的,反正是错的。大概情况就是这样,就是不知道问题出在哪
回复

使用道具 举报

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

本版积分规则

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


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