Toybrick

标题: rknn转出后推理错误 [打印本页]

作者: wokafu 时间: 2022-4-19 10:20
标题: rknn转出后推理错误
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

麻烦各位老师指点迷津，谢谢了！

作者: jefferyzhang 时间: 2022-4-19 10:34
warning不用在意

作者: wokafu 时间: 2022-4-19 10:49
onnx推理是正确的，rknn推理完全不正确，请问这个一般是哪个步骤错误导致的呢？大佬！
打开rknn文件，各个层的weight参数感觉又是对的

作者: wokafu 时间: 2022-4-19 13:23
结题了，输入的时候图片处理得不对，各种变换处理后，推理正确了！在inference的参数指定data_format="nchw"

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